From b732b40d665c9548081f63d0db05193cdc0888c6 Mon Sep 17 00:00:00 2001
From: Dennis Collaris <d.collaris@me.com>
Date: Mon, 17 Mar 2025 12:24:16 +0100
Subject: [PATCH] fix: storybook by updating imports and types

---
 src/App.tsx                                        |  2 +-
 src/lib/data-access/store/schemaSlice.ts           | 10 ++--------
 src/lib/mock-data/schema/big2ndChamberSchemaRaw.ts | 10 +++++++---
 src/lib/mock-data/schema/fincenWithAttributes.ts   | 10 +++++++---
 .../schema/marieBoucherSampleSchemaRaw.ts          |  8 ++++++--
 src/lib/mock-data/schema/moviesSchemaRaw.ts        |  8 ++++++--
 .../mock-data/schema/moviesSchemaWithAttributes.ts | 10 +++++++---
 src/lib/mock-data/schema/northwindSchemaRaw.ts     |  8 ++++++--
 .../mock-data/schema/northwindWithAttributes.ts    | 10 +++++++---
 .../schema/recommendationsWithAttributes.ts        | 10 +++++++---
 src/lib/mock-data/schema/simpleAirportRaw.ts       |  8 ++++++--
 src/lib/mock-data/schema/simpleRaw.ts              |  8 ++++++--
 src/lib/mock-data/schema/slackWithAttributes.ts    | 10 +++++++---
 src/lib/mock-data/schema/twitterSchemaRaw.ts       |  8 ++++++--
 src/lib/mock-data/schema/typesMockSchemaRaw.ts     | 10 +++++++---
 .../panel/schemaquerybuilder.stories.tsx           |  6 +++---
 .../panel/stories/querybuilder-simple.stories.tsx  |  8 ++++++--
 .../stories/querybuilder-single-entity.stories.tsx |  4 ++--
 .../querybuilder-single-relationship.stories.tsx   |  4 ++--
 src/lib/schema/panel/schema.spec.ts                | 14 +++++++++-----
 src/lib/schema/panel/schema.stories.tsx            | 10 +++++++++-
 src/lib/schema/schema-utils/schemaUsecases.spec.ts |  6 +++---
 22 files changed, 122 insertions(+), 60 deletions(-)

diff --git a/src/App.tsx b/src/App.tsx
index f5488f9f5..719242c41 100644
--- a/src/App.tsx
+++ b/src/App.tsx
@@ -1,4 +1,4 @@
-import React, { useEffect, useState } from 'react';
+import { useEffect, useState } from 'react';
 import { useActiveQuery, useAppDispatch, useAuthCache, useML, useSessionCache } from '@/lib/data-access/store';
 import { addError, setCurrentTheme } from '@/lib/data-access/store/configSlice';
 import { resetGraphQueryResults, queryingBackend } from '@/lib/data-access/store/graphQueryResultSlice';
diff --git a/src/lib/data-access/store/schemaSlice.ts b/src/lib/data-access/store/schemaSlice.ts
index bfdbccffe..1d194f6d2 100644
--- a/src/lib/data-access/store/schemaSlice.ts
+++ b/src/lib/data-access/store/schemaSlice.ts
@@ -20,14 +20,8 @@ export type SchemaSliceI = {
 export const initialState: SchemaSliceI = {
   graph: new SchemaGraphology().export(),
   graphStats: {
-    nodes: {
-      count: 0,
-      stats: {},
-    },
-    edges: {
-      count: 0,
-      stats: {},
-    },
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
   },
   loading: false,
   error: false,
diff --git a/src/lib/mock-data/schema/big2ndChamberSchemaRaw.ts b/src/lib/mock-data/schema/big2ndChamberSchemaRaw.ts
index 535dd220a..a0d8bf1f4 100644
--- a/src/lib/mock-data/schema/big2ndChamberSchemaRaw.ts
+++ b/src/lib/mock-data/schema/big2ndChamberSchemaRaw.ts
@@ -1,7 +1,7 @@
-import { SchemaUtils } from '@/lib/schema/schema-utils';
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
+import { SchemaReturnFormat } from 'ts-common';
 
-export const big2ndChamberSchemaRaw: SchemaFromBackend = {
+export const big2ndChamberSchemaRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'kamerleden',
@@ -68,6 +68,10 @@ export const big2ndChamberSchemaRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const big2ndChamberSchema = SchemaUtils.schemaBackend2Graphology(big2ndChamberSchemaRaw);
diff --git a/src/lib/mock-data/schema/fincenWithAttributes.ts b/src/lib/mock-data/schema/fincenWithAttributes.ts
index 982aaf157..1a78f4818 100644
--- a/src/lib/mock-data/schema/fincenWithAttributes.ts
+++ b/src/lib/mock-data/schema/fincenWithAttributes.ts
@@ -1,7 +1,7 @@
-import { SchemaUtils } from '@/lib/schema/schema-utils';
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
+import { SchemaReturnFormat } from 'ts-common';
 
-export const fincenWithAttributesRaw: SchemaFromBackend = {
+export const fincenWithAttributesRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Entity',
@@ -173,6 +173,10 @@ export const fincenWithAttributesRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const fincenWithAttributes = SchemaUtils.schemaBackend2Graphology(fincenWithAttributesRaw);
diff --git a/src/lib/mock-data/schema/marieBoucherSampleSchemaRaw.ts b/src/lib/mock-data/schema/marieBoucherSampleSchemaRaw.ts
index 899d9c54c..c5f8901c8 100644
--- a/src/lib/mock-data/schema/marieBoucherSampleSchemaRaw.ts
+++ b/src/lib/mock-data/schema/marieBoucherSampleSchemaRaw.ts
@@ -1,7 +1,7 @@
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaReturnFormat } from 'ts-common';
 import { SchemaUtils } from '../../schema/schema-utils';
 
-export const marieBoucherSampleSchemaRaw: SchemaFromBackend = {
+export const marieBoucherSampleSchemaRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'merchant',
@@ -21,6 +21,10 @@ export const marieBoucherSampleSchemaRaw: SchemaFromBackend = {
       attributes: [{ name: 'time', type: 'int' }],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const marieBoucherSampleSchema = SchemaUtils.schemaBackend2Graphology(marieBoucherSampleSchemaRaw);
diff --git a/src/lib/mock-data/schema/moviesSchemaRaw.ts b/src/lib/mock-data/schema/moviesSchemaRaw.ts
index e99b7f9a2..1081a161b 100644
--- a/src/lib/mock-data/schema/moviesSchemaRaw.ts
+++ b/src/lib/mock-data/schema/moviesSchemaRaw.ts
@@ -1,7 +1,7 @@
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaReturnFormat } from 'ts-common';
 import { SchemaUtils } from '../../schema/schema-utils/schemaUtils';
 
-export const movieSchemaRaw: SchemaFromBackend = {
+export const movieSchemaRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Movie',
@@ -102,6 +102,10 @@ export const movieSchemaRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const movieSchema = SchemaUtils.schemaBackend2Graphology(movieSchemaRaw);
diff --git a/src/lib/mock-data/schema/moviesSchemaWithAttributes.ts b/src/lib/mock-data/schema/moviesSchemaWithAttributes.ts
index d369048f3..78bcd6fd1 100644
--- a/src/lib/mock-data/schema/moviesSchemaWithAttributes.ts
+++ b/src/lib/mock-data/schema/moviesSchemaWithAttributes.ts
@@ -1,7 +1,7 @@
-import { SchemaUtils } from '@/lib/schema/schema-utils/schemaUtils';
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
+import { SchemaReturnFormat } from 'ts-common';
 
-export const movieSchemaWithAttributesRaw: SchemaFromBackend = {
+export const movieSchemaWithAttributesRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Movie',
@@ -102,6 +102,10 @@ export const movieSchemaWithAttributesRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const movieSchemaWithAttributes = SchemaUtils.schemaBackend2Graphology(movieSchemaWithAttributesRaw);
diff --git a/src/lib/mock-data/schema/northwindSchemaRaw.ts b/src/lib/mock-data/schema/northwindSchemaRaw.ts
index e6abb6d04..eb3aad664 100644
--- a/src/lib/mock-data/schema/northwindSchemaRaw.ts
+++ b/src/lib/mock-data/schema/northwindSchemaRaw.ts
@@ -1,7 +1,7 @@
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaReturnFormat } from 'ts-common';
 import { SchemaUtils } from '../../schema/schema-utils/schemaUtils';
 
-export const northwindSchemaRaw: SchemaFromBackend = {
+export const northwindSchemaRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Order',
@@ -288,6 +288,10 @@ export const northwindSchemaRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const northWindSchema = SchemaUtils.schemaBackend2Graphology(northwindSchemaRaw);
diff --git a/src/lib/mock-data/schema/northwindWithAttributes.ts b/src/lib/mock-data/schema/northwindWithAttributes.ts
index 5bbeb2551..5a9ac1d77 100644
--- a/src/lib/mock-data/schema/northwindWithAttributes.ts
+++ b/src/lib/mock-data/schema/northwindWithAttributes.ts
@@ -1,7 +1,7 @@
-import { SchemaUtils } from '@/lib/schema/schema-utils';
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
+import { SchemaReturnFormat } from 'ts-common';
 
-export const northWindWithAttributesRaw: SchemaFromBackend = {
+export const northWindWithAttributesRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Order',
@@ -288,6 +288,10 @@ export const northWindWithAttributesRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const northWindWithAttributes = SchemaUtils.schemaBackend2Graphology(northWindWithAttributesRaw);
diff --git a/src/lib/mock-data/schema/recommendationsWithAttributes.ts b/src/lib/mock-data/schema/recommendationsWithAttributes.ts
index 6e33a0e91..b3755ceb8 100644
--- a/src/lib/mock-data/schema/recommendationsWithAttributes.ts
+++ b/src/lib/mock-data/schema/recommendationsWithAttributes.ts
@@ -1,7 +1,7 @@
-import { SchemaUtils } from '@/lib/schema/schema-utils';
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
+import { SchemaReturnFormat } from 'ts-common';
 
-export const recommendationsWithAttributesRaw: SchemaFromBackend = {
+export const recommendationsWithAttributesRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Movie',
@@ -467,6 +467,10 @@ export const recommendationsWithAttributesRaw: SchemaFromBackend = {
       ],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const recommendationsSchemaWithAttributes = SchemaUtils.schemaBackend2Graphology(recommendationsWithAttributesRaw);
diff --git a/src/lib/mock-data/schema/simpleAirportRaw.ts b/src/lib/mock-data/schema/simpleAirportRaw.ts
index c77808b0a..1c3286e86 100644
--- a/src/lib/mock-data/schema/simpleAirportRaw.ts
+++ b/src/lib/mock-data/schema/simpleAirportRaw.ts
@@ -1,7 +1,7 @@
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaReturnFormat } from 'ts-common';
 import { SchemaUtils } from '../../schema/schema-utils';
 
-export const simpleSchemaAirportRaw: SchemaFromBackend = {
+export const simpleSchemaAirportRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'airports',
@@ -39,6 +39,10 @@ export const simpleSchemaAirportRaw: SchemaFromBackend = {
       ],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const simpleSchemaAirport = SchemaUtils.schemaBackend2Graphology(simpleSchemaAirportRaw);
diff --git a/src/lib/mock-data/schema/simpleRaw.ts b/src/lib/mock-data/schema/simpleRaw.ts
index 86948c0ed..551b512cc 100644
--- a/src/lib/mock-data/schema/simpleRaw.ts
+++ b/src/lib/mock-data/schema/simpleRaw.ts
@@ -1,7 +1,7 @@
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaReturnFormat } from 'ts-common';
 import { SchemaUtils } from '../../schema/schema-utils/schemaUtils';
 
-export const simpleSchemaRaw: SchemaFromBackend = {
+export const simpleSchemaRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Thijs',
@@ -109,6 +109,10 @@ export const simpleSchemaRaw: SchemaFromBackend = {
       attributes: [{ name: 'test', type: 'string' }],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const simpleSchema = SchemaUtils.schemaBackend2Graphology(simpleSchemaRaw);
diff --git a/src/lib/mock-data/schema/slackWithAttributes.ts b/src/lib/mock-data/schema/slackWithAttributes.ts
index 6dd9da425..65285703f 100644
--- a/src/lib/mock-data/schema/slackWithAttributes.ts
+++ b/src/lib/mock-data/schema/slackWithAttributes.ts
@@ -1,7 +1,7 @@
-import { SchemaUtils } from '@/lib/schema/schema-utils';
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
+import { SchemaReturnFormat } from 'ts-common';
 
-export const slackWithAttributesRaw: SchemaFromBackend = {
+export const slackWithAttributesRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Owner',
@@ -2774,6 +2774,10 @@ export const slackWithAttributesRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const slackWithAttributes = SchemaUtils.schemaBackend2Graphology(slackWithAttributesRaw);
diff --git a/src/lib/mock-data/schema/twitterSchemaRaw.ts b/src/lib/mock-data/schema/twitterSchemaRaw.ts
index 06026e614..a0062c3b3 100644
--- a/src/lib/mock-data/schema/twitterSchemaRaw.ts
+++ b/src/lib/mock-data/schema/twitterSchemaRaw.ts
@@ -1,7 +1,7 @@
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaReturnFormat } from 'ts-common';
 import { SchemaUtils } from '../../schema/schema-utils/schemaUtils';
 
-export const twitterSchemaRaw: SchemaFromBackend = {
+export const twitterSchemaRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'Me',
@@ -334,6 +334,10 @@ export const twitterSchemaRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const twitterSchema = SchemaUtils.schemaBackend2Graphology(twitterSchemaRaw);
diff --git a/src/lib/mock-data/schema/typesMockSchemaRaw.ts b/src/lib/mock-data/schema/typesMockSchemaRaw.ts
index fbce7c3b8..8aa3ba305 100644
--- a/src/lib/mock-data/schema/typesMockSchemaRaw.ts
+++ b/src/lib/mock-data/schema/typesMockSchemaRaw.ts
@@ -1,7 +1,7 @@
-import { SchemaUtils } from '@/lib/schema/schema-utils';
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
+import { SchemaReturnFormat } from 'ts-common';
 
-export const typesMockSchemaRaw: SchemaFromBackend = {
+export const typesMockSchemaRaw: SchemaReturnFormat = {
   nodes: [
     {
       name: 'worker',
@@ -51,6 +51,10 @@ export const typesMockSchemaRaw: SchemaFromBackend = {
       attributes: [],
     },
   ],
+  graphStats: {
+    nodes: { count: 0, stats: {} },
+    edges: { count: 0, stats: {} },
+  },
 };
 
 export const typesMockSchema = SchemaUtils.schemaBackend2Graphology(typesMockSchemaRaw);
diff --git a/src/lib/querybuilder/panel/schemaquerybuilder.stories.tsx b/src/lib/querybuilder/panel/schemaquerybuilder.stories.tsx
index d68798c0c..96a023d28 100644
--- a/src/lib/querybuilder/panel/schemaquerybuilder.stories.tsx
+++ b/src/lib/querybuilder/panel/schemaquerybuilder.stories.tsx
@@ -1,8 +1,8 @@
-import React from 'react';
 import { Meta } from '@storybook/react';
 import { Provider } from 'react-redux';
-import { setQuerybuilderGraph, setSchema, store } from '@/lib/data-access/store';
-import { SchemaUtils } from '@/lib/schema/schema-utils';
+import { setSchema, store } from '@/lib/data-access/store';
+import { setQuerybuilderGraph } from '@/lib/data-access/store/sessionSlice';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
 import { Schema } from '@/lib/schema/panel';
 import { movieSchemaRaw } from '@/lib/mock-data';
 import { QueryBuilder } from '@/lib/querybuilder';
diff --git a/src/lib/querybuilder/panel/stories/querybuilder-simple.stories.tsx b/src/lib/querybuilder/panel/stories/querybuilder-simple.stories.tsx
index 42bd1e8f9..1a1b040d2 100644
--- a/src/lib/querybuilder/panel/stories/querybuilder-simple.stories.tsx
+++ b/src/lib/querybuilder/panel/stories/querybuilder-simple.stories.tsx
@@ -1,5 +1,5 @@
-import React from 'react';
-import { setQuerybuilderGraph, setSchema, store } from '@/lib/data-access/store';
+import { setSchema, store } from '@/lib/data-access/store';
+import { setQuerybuilderGraph } from '@/lib/data-access/store/sessionSlice';
 import { Meta } from '@storybook/react';
 import { Provider } from 'react-redux';
 import QueryBuilder from '../QueryBuilder';
@@ -55,6 +55,10 @@ export const Simple = {
           ],
         },
       ],
+      graphStats: {
+        nodes: { count: 0, stats: {} },
+        edges: { count: 0, stats: {} },
+      },
     });
 
     store.dispatch(setSchema(schema.export()));
diff --git a/src/lib/querybuilder/panel/stories/querybuilder-single-entity.stories.tsx b/src/lib/querybuilder/panel/stories/querybuilder-single-entity.stories.tsx
index 132405410..3b382ac7b 100644
--- a/src/lib/querybuilder/panel/stories/querybuilder-single-entity.stories.tsx
+++ b/src/lib/querybuilder/panel/stories/querybuilder-single-entity.stories.tsx
@@ -1,5 +1,5 @@
-import React from 'react';
-import { setQuerybuilderGraph, store } from '@/lib/data-access/store';
+import { store } from '@/lib/data-access/store';
+import { setQuerybuilderGraph } from '@/lib/data-access/store/sessionSlice';
 import { Meta } from '@storybook/react';
 import { Provider } from 'react-redux';
 import QueryBuilder from '../QueryBuilder';
diff --git a/src/lib/querybuilder/panel/stories/querybuilder-single-relationship.stories.tsx b/src/lib/querybuilder/panel/stories/querybuilder-single-relationship.stories.tsx
index 4907c643f..a24430b3a 100644
--- a/src/lib/querybuilder/panel/stories/querybuilder-single-relationship.stories.tsx
+++ b/src/lib/querybuilder/panel/stories/querybuilder-single-relationship.stories.tsx
@@ -1,5 +1,5 @@
-import React from 'react';
-import { setQuerybuilderGraph, store } from '@/lib/data-access/store';
+import { store } from '@/lib/data-access/store';
+import { setQuerybuilderGraph } from '@/lib/data-access/store/sessionSlice';
 import { Meta } from '@storybook/react';
 import { Provider } from 'react-redux';
 import QueryBuilder from '../QueryBuilder';
diff --git a/src/lib/schema/panel/schema.spec.ts b/src/lib/schema/panel/schema.spec.ts
index 548ec907b..3760edfaf 100644
--- a/src/lib/schema/panel/schema.spec.ts
+++ b/src/lib/schema/panel/schema.spec.ts
@@ -1,9 +1,9 @@
-import { assert, describe, expect, it } from 'vitest';
+import { describe, it } from 'vitest';
 
 // import { SchemaUtils } from '@graphpolaris/schema-utils';
-// import { SchemaFromBackend } from '@graphpolaris/models';
+// import { SchemaReturnFormat } from '@graphpolaris/models';
 
-// const twitterSchemaRaw: SchemaFromBackend = {
+// const twitterSchemaRaw: SchemaReturnFormat = {
 //   nodes: [
 //     {
 //       name: 'Me',
@@ -313,8 +313,12 @@ import { assert, describe, expect, it } from 'vitest';
 //       from: 'Me',
 //       to: 'Tweet',
 //       attributes: [],
-//     },
-//   ],
+//     //},
+//   //],
+//    graphStats: {
+//      nodes: { count: 0, stats: {} },
+//      edges: { count: 0, stats: {} },
+//    },
 // };
 
 // const twitterSchema = SchemaUtils.ParseSchemaFromBackend(twitterSchemaRaw);
diff --git a/src/lib/schema/panel/schema.stories.tsx b/src/lib/schema/panel/schema.stories.tsx
index ac7dae2b0..d0ef51cd7 100644
--- a/src/lib/schema/panel/schema.stories.tsx
+++ b/src/lib/schema/panel/schema.stories.tsx
@@ -1,6 +1,6 @@
 import { schemaSlice, setSchema } from '@/lib/data-access/store';
 import { movieSchemaRaw, northwindSchemaRaw, twitterSchemaRaw } from '@/lib/mock-data';
-import { SchemaUtils } from '@/lib/schema/schema-utils';
+import { SchemaUtils } from '../../../lib/schema/schema-utils';
 import { configureStore } from '@reduxjs/toolkit';
 import { Meta } from '@storybook/react';
 import { Provider } from 'react-redux';
@@ -72,6 +72,10 @@ export const TestSimple = {
           ],
         },
       ],
+      graphStats: {
+        nodes: { count: 0, stats: {} },
+        edges: { count: 0, stats: {} },
+      },
     });
 
     dispatch(setSchema(schema.export()));
@@ -93,6 +97,10 @@ export const TestTooltip = {
         },
       ],
       edges: [],
+      graphStats: {
+        nodes: { count: 0, stats: {} },
+        edges: { count: 0, stats: {} },
+      },
     });
 
     dispatch(setSchema(schema.export()));
diff --git a/src/lib/schema/schema-utils/schemaUsecases.spec.ts b/src/lib/schema/schema-utils/schemaUsecases.spec.ts
index 37dc7c726..2e4957141 100644
--- a/src/lib/schema/schema-utils/schemaUsecases.spec.ts
+++ b/src/lib/schema/schema-utils/schemaUsecases.spec.ts
@@ -2,19 +2,19 @@ import { describe, expect, it, test } from 'vitest';
 import { movieSchemaRaw, northwindSchemaRaw, simpleSchemaRaw, twitterSchemaRaw } from '../../mock-data/schema';
 import { SchemaGraphology } from '../model';
 import { SchemaUtils } from './schemaUtils';
-import { SchemaFromBackend } from 'ts-common';
+import { SchemaReturnFormat } from 'ts-common';
 
 describe('SchemaUsecases', () => {
   test.each([{ data: simpleSchemaRaw }, { data: movieSchemaRaw }, { data: northwindSchemaRaw }, { data: twitterSchemaRaw }])(
     'parseSchemaFromBackend parsing should work',
     ({ data }: any) => {
-      const parsed = SchemaUtils.schemaBackend2Graphology(data as SchemaFromBackend);
+      const parsed = SchemaUtils.schemaBackend2Graphology(data as SchemaReturnFormat);
       expect(parsed).toBeDefined();
     },
   );
 
   it('should export and reimport', () => {
-    const parsed = SchemaUtils.schemaBackend2Graphology(simpleSchemaRaw as SchemaFromBackend);
+    const parsed = SchemaGraphology.from(SchemaUtils.schemaBackend2Graphology(simpleSchemaRaw as SchemaReturnFormat));
     const reload = SchemaGraphology.from(parsed.export());
 
     expect(parsed).toEqual(reload);
-- 
GitLab