update test

justeattakeaway · Nov 29, 2024 · e955c28 · e955c28
1 parent 6ff9eb6
commit e955c28
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/apps/pie-storybook/test/utilities/index.test.ts b/apps/pie-storybook/test/utilities/index.test.ts
@@ -1,5 +1,5 @@
 import { describe, it, expect } from 'vitest';
-import { html, render } from 'lit';
+import { html, render, type TemplateResult} from 'lit';
 import { createStory, createVariantStory } from '../../utilities/index';
 import { type StoryOptions } from '../../types/StoryOptions';
 
@@ -19,7 +19,17 @@ describe('createStory', () => {
 
         const result = story();
         expect(result.args).toEqual(defaultArgs);
-        expect(result.render(defaultArgs)).toHaveProperty('strings');
+
+        const actualTemplateResult = result.render(defaultArgs);
+
+        const expectedTemplateResult: Partial<TemplateResult> = {
+          strings: Object.assign(['\n  <pie-component size=', ' variant=', '></pie-component>\n']),
+          values: ['medium', 'primary']
+        };
+
+        // Compare the key properties of the TemplateResult
+        expect(actualTemplateResult.strings).toEqual(expectedTemplateResult.strings);
+        expect(actualTemplateResult.values).toEqual(expectedTemplateResult.values);
     });
 
     it('should override default args with prop overrides', () => {