feat: added check for root model

Author: jake-bassett

src/dashboard/dashboard-manager.test.ts

@@ -188,6 +188,6 @@ describe('Dashboard manager', () => {
     mockModelManager.getModelInstances = jest.fn().mockReturnValue([new mockDataSource()]);
 
     expect(dashboard.getModelInstances(mockDataSource).length).toBe(1);
-    expect(mockModelManager.getModelInstances).toHaveBeenCalledWith(mockDataSource);
+    expect(mockModelManager.getModelInstances).toHaveBeenCalledWith(mockDataSource, {});
   });
 });

src/dashboard/default-dashboard.ts

@@ -91,6 +91,6 @@ export class DefaultDashboard<TRoot extends object> implements Dashboard<TRoot>
    * @inheritdoc
    */
   public getModelInstances<T extends object>(modelClass: Constructable<T>): object[] {
-    return this.modelManager.getModelInstances<T>(modelClass);
+    return this.modelManager.getModelInstances<T>(modelClass, this.root);
   }
 }

src/model/manager/model-manager.test.ts

@@ -74,6 +74,7 @@ describe('Model manager', () => {
     expect(manager.getModelInstances(testClass).length).toBe(2);
     expect(manager.getModelInstances(testClass1).length).toBe(1);
     expect(manager.getModelInstances(testClass2).length).toBe(0);
+    expect(manager.getModelInstances(testClass, {}).length).toBe(0);
   });
 
   test('allows constructing new models', () => {

src/model/manager/model-manager.ts

@@ -27,8 +27,10 @@ export class ModelManager {
   /**
    * Returns a shallow copy array of model instances that match the argument model class
    */
-  public getModelInstances<T extends object>(modelClass: Constructable<T>): object[] {
-    return Array.from(this.modelInstanceMap.keys()).filter(modelInstance => modelInstance instanceof modelClass);
+  public getModelInstances<T extends object>(modelClass: Constructable<T>, root?: object): object[] {
+    return Array.from(this.modelInstanceMap.keys())
+      .filter(modelInstance => root === undefined || this.getRoot(modelInstance) === root)
+      .filter(modelInstance => modelInstance instanceof modelClass);
   }
 
   /**