feat(model): Add modelData support + fix unit test warnings (#123)

* feat(model): Add modelData support + fix unit test warnings

* Fix lint

Author: Nicolas ERNY

package.json

@@ -1,6 +1,6 @@
 {
     "name": "react-gojs",
-    "version": "4.8.0",
+    "version": "4.9.0",
     "description": "GoJS React integration",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",

src/GojsDiagram.test.tsx

@@ -99,6 +99,7 @@ describe('<GojsDiagram />', () => {
     };
 
     const myDiagramId = 'myDiagramId';
+    const modelData = { test: 'modeldatata' };
 
     const defaultSelectedNodeKey = 'Beta';
 
@@ -107,6 +108,10 @@ describe('<GojsDiagram />', () => {
     let modelChangeCallback;
     let keyIndex = 0;
 
+    const dom = document.body;
+    const rootDiv = document.createElement('div');
+    dom.appendChild(rootDiv);
+
     beforeEach(() => {
         Element.prototype.getBoundingClientRect = function() {
             return {
@@ -116,20 +121,18 @@ describe('<GojsDiagram />', () => {
                 left: 0,
                 x: 0,
                 y: 0,
-                toJSON: () => {
-                    console.log('test');
-                },
+                toJSON: () => '',
                 bottom: 0,
                 right: 0
             };
         };
         keyIndex = 0;
-        const dom = document.body;
         modelChangeCallback = jest.fn();
         wrapper = mount(
             <GojsDiagram
                 diagramId={myDiagramId}
                 model={model}
+                initialModelData={modelData}
                 createDiagram={id => {
                     diagram = createDiagram(myDiagramId);
                     return diagram;
@@ -158,7 +161,7 @@ describe('<GojsDiagram />', () => {
                 }}
                 defaultSelectedNode={defaultSelectedNodeKey}
             />,
-            { attachTo: dom }
+            { attachTo: rootDiv }
         );
         jest.runAllTimers();
     });
@@ -167,6 +170,10 @@ describe('<GojsDiagram />', () => {
         expect(diagram.model.nodeCategoryProperty === 'category').toBeTruthy();
     });
 
+    it('should initialize the modelData', () => {
+        expect(diagram.model.modelData).toBe(modelData);
+    });
+
     it('should render links and nodes in the diagram based on the model provided as prop', () => {
         checkIfDiagramRendersModel(model, diagram);
     });

src/GojsDiagram.tsx

@@ -15,6 +15,7 @@ import {
 
 export interface GojsDiagramProps<N extends BaseNodeModel, L extends LinkModel> {
     model: DiagramModel<N, L>;
+    initialModelData?: ObjectData;
     createDiagram: (id: string) => Diagram;
     diagramId: string;
     className: string;
@@ -131,7 +132,8 @@ class GojsDiagram<N extends BaseNodeModel, L extends LinkModel> extends React.Pu
             nodeCategoryProperty: this.props.nodeCategoryProperty || 'category',
             linkKeyProperty: this.props.linkKeyProperty || '',
             makeUniqueLinkKeyFunction: this.props.makeUniqueLinkKeyFunction || null,
-            copyNodeDataFunction: this.props.copyNodeDataFunction || null
+            copyNodeDataFunction: this.props.copyNodeDataFunction || null,
+            ...(this.props.initialModelData && { modelData: this.props.initialModelData })
         });
 
         if (defaultSelectedNode) {