From cf9da6dafbe6c2fd038d9b8176a2f28816b76e10 Mon Sep 17 00:00:00 2001
From: blam <ben@blam.sh>
Date: Mon, 22 Jul 2024 16:10:05 +0200
Subject: [PATCH] chore: added test for dataRef exporting

Signed-off-by: blam <ben@blam.sh>
---
 .../wiring/createExtensionBlueprint.test.tsx  | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/packages/frontend-plugin-api/src/wiring/createExtensionBlueprint.test.tsx b/packages/frontend-plugin-api/src/wiring/createExtensionBlueprint.test.tsx
index c7b077bdbd..87b3cec152 100644
--- a/packages/frontend-plugin-api/src/wiring/createExtensionBlueprint.test.tsx
+++ b/packages/frontend-plugin-api/src/wiring/createExtensionBlueprint.test.tsx
@@ -18,6 +18,7 @@ import React from 'react';
 import { coreExtensionData } from './coreExtensionData';
 import { createExtensionBlueprint } from './createExtensionBlueprint';
 import { createExtensionTester } from '@backstage/frontend-test-utils';
+import { createExtensionDataRef } from './createExtensionDataRef';
 
 describe('createExtensionBlueprint', () => {
   it('should allow creation of extension blueprints', () => {
@@ -102,4 +103,28 @@ describe('createExtensionBlueprint', () => {
     const { container } = createExtensionTester(extension).render();
     expect(container.querySelector('h2')).toHaveTextContent('Hello, world!');
   });
+
+  it('should allow exporting the dataRefs from the extension blueprint', () => {
+    const dataRef = createExtensionDataRef<string>().with({ id: 'test.data' });
+
+    const TestExtensionBlueprint = createExtensionBlueprint({
+      kind: 'test-extension',
+      attachTo: { id: 'test', input: 'default' },
+      output: {
+        element: coreExtensionData.reactElement,
+      },
+      dataRefs: {
+        data: dataRef,
+      },
+      factory(params: { text: string }) {
+        return {
+          element: <h1>{params.text}</h1>,
+        };
+      },
+    });
+
+    expect(TestExtensionBlueprint.dataRefs).toEqual({
+      data: dataRef,
+    });
+  });
 });