From b36e4eba7c6fe648db8f50ab8f591e7cb7b03057 Mon Sep 17 00:00:00 2001
From: Bernardo Sunderhus <bernardo.sunderhus@gmail.com>
Date: Fri, 14 Apr 2023 10:55:22 +0000
Subject: [PATCH] chore(react-utilities): add test to ensure slot render
 functions

---
 .../src/compose/getSlots.test.tsx             | 19 ++++++++++++++++++
 .../src/compose/getSlotsNext.test.tsx         | 20 +++++++++++++++++++
 2 files changed, 39 insertions(+)

diff --git a/packages/react-components/react-utilities/src/compose/getSlots.test.tsx b/packages/react-components/react-utilities/src/compose/getSlots.test.tsx
index bb4ca39b46eae..4318a1a0db689 100644
--- a/packages/react-components/react-utilities/src/compose/getSlots.test.tsx
+++ b/packages/react-components/react-utilities/src/compose/getSlots.test.tsx
@@ -1,6 +1,7 @@
 import * as React from 'react';
 import { getSlots } from './getSlots';
 import type { Slot } from './types';
+import { resolveShorthand } from './resolveShorthand';
 
 describe('getSlots', () => {
   type FooProps = { id?: string; children?: React.ReactNode };
@@ -126,6 +127,24 @@ describe('getSlots', () => {
     });
   });
 
+  it('can use slot children functions from resolveShorthand to replace default slot rendering', () => {
+    type Slots = {
+      root: Slot<'div'>;
+      icon: Slot<'a'>;
+    };
+    const renderFunction = (C: React.ElementType, p: {}) => <C {...p} />;
+    expect(
+      getSlots<Slots>({
+        components: { root: 'div', icon: Foo },
+        root: resolveShorthand({ as: 'div' }, { required: true }),
+        icon: resolveShorthand({ id: 'bar', children: renderFunction }),
+      }),
+    ).toEqual({
+      slots: { root: 'div', icon: React.Fragment },
+      slotProps: { root: {}, icon: { children: <Foo id="bar" /> } },
+    });
+  });
+
   it('can render a primitive input with no children', () => {
     type Slots = {
       root: Slot<'div'>;
diff --git a/packages/react-components/react-utilities/src/compose/getSlotsNext.test.tsx b/packages/react-components/react-utilities/src/compose/getSlotsNext.test.tsx
index fc824df268541..a66534b278698 100644
--- a/packages/react-components/react-utilities/src/compose/getSlotsNext.test.tsx
+++ b/packages/react-components/react-utilities/src/compose/getSlotsNext.test.tsx
@@ -1,6 +1,8 @@
 import * as React from 'react';
 import { getSlotsNext } from './getSlotsNext';
 import type { Slot } from './types';
+import { resolveShorthand } from './resolveShorthand';
+import { SLOT_RENDER_FUNCTION_SYMBOL } from './constants';
 
 describe('getSlotsNext', () => {
   type FooProps = { id?: string; children?: React.ReactNode };
@@ -127,6 +129,24 @@ describe('getSlotsNext', () => {
     });
   });
 
+  it('can use slot children functions from resolveShorthand to replace default slot rendering', () => {
+    type Slots = {
+      root: Slot<'div'>;
+      icon: Slot<'a'>;
+    };
+    const renderFunction = (C: React.ElementType, p: {}) => <C {...p} />;
+    expect(
+      getSlotsNext<Slots>({
+        components: { root: 'div', icon: Foo },
+        root: resolveShorthand({ as: 'div' }, { required: true }),
+        icon: resolveShorthand({ id: 'bar', children: renderFunction }),
+      }),
+    ).toEqual({
+      slots: { root: 'div', icon: Foo },
+      slotProps: { root: {}, icon: { children: undefined, id: 'bar', [SLOT_RENDER_FUNCTION_SYMBOL]: renderFunction } },
+    });
+  });
+
   it('can render a primitive input with no children', () => {
     type Slots = {
       root: Slot<'div'>;