Fix unit tests

GMOD · Jul 1, 2021 · e24c2af · e24c2af
1 parent b03d68c
commit e24c2af
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 22 deletions.
diff --git a/plugins/breakpoint-split-view/src/BreakpointSplitView.test.tsx b/plugins/breakpoint-split-view/src/BreakpointSplitView.test.tsx
@@ -23,25 +23,29 @@ const getView = () => {
       type: 'FakeTrack',
       configuration: types.frozen(),
       displays: types.array(
-        types.model('FakeDisplay', {
-          type: 'FakeDisplay',
-          displayId: 'FakeDisplay',
-          configuration: types.frozen(),
-          features: types.frozen(),
-          layoutFeatures: types.frozen(),
-        }),
+        types
+          .model('FakeDisplay', {
+            type: 'FakeDisplay',
+            displayId: 'FakeDisplay',
+            configuration: types.frozen(),
+            layoutFeatures: types.frozen(),
+            features: types.frozen(),
+          })
+          .views(self => ({
+            getFeatureByID(id: string) {
+              return self.layoutFeatures[id]
+            },
+          })),
       ),
     })
     .actions(self => ({
       afterCreate() {
         self.displays[0].features = new Map(
           Object.entries(self.displays[0].features),
         )
-        self.displays[0].layoutFeatures = new Map(
-          Object.entries(self.displays[0].layoutFeatures),
-        )
       },
     }))
+
   stubManager.addViewType(
     () =>
       new ViewType({

diff --git a/plugins/breakpoint-split-view/src/model.test.tsx b/plugins/breakpoint-split-view/src/model.test.tsx
@@ -23,23 +23,26 @@ const getView = () => {
       type: 'FakeTrack',
       configuration: types.frozen(),
       displays: types.array(
-        types.model('FakeDisplay', {
-          type: 'FakeDisplay',
-          displayId: 'FakeDisplay',
-          configuration: types.frozen(),
-          features: types.frozen(),
-          layoutFeatures: types.frozen(),
-        }),
+        types
+          .model('FakeDisplay', {
+            type: 'FakeDisplay',
+            displayId: 'FakeDisplay',
+            configuration: types.frozen(),
+            features: types.frozen(),
+            layoutFeatures: types.frozen(),
+          })
+          .views(self => ({
+            getFeatureByID(id: string) {
+              return self.layoutFeatures[id]
+            },
+          })),
       ),
     })
     .actions(self => ({
       afterCreate() {
         self.displays[0].features = new Map(
           Object.entries(self.displays[0].features),
         )
-        self.displays[0].layoutFeatures = new Map(
-          Object.entries(self.displays[0].layoutFeatures),
-        )
       },
     }))
   stubManager.addViewType(

diff --git a/plugins/breakpoint-split-view/src/model.ts b/plugins/breakpoint-split-view/src/model.ts
@@ -229,8 +229,9 @@ export default function stateModelFactory(pluginManager: any) {
         // use reverse to search the second track first
         const tracks = this.getMatchedTracks(trackConfigId)
 
-        const calc = (track: any, feat: Feature) =>
-          track.displays[0].getFeatureByID(feat.id())
+        const calc = (track: any, feat: Feature) => {
+          return track.displays[0].getFeatureByID(feat.id())
+        }
 
         return features.map(c =>
           c.map(feature => {