Fix unit tests

cloudfoundry · Dec 4, 2020 · b1d4c01 · b1d4c01
1 parent 33e675c
commit b1d4c01
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/frontend/packages/kubernetes/src/kubernetes/kubernetes.testing.module.ts b/src/frontend/packages/kubernetes/src/kubernetes/kubernetes.testing.module.ts
@@ -15,7 +15,7 @@ import {
 import { generateStratosEntities } from '../../../store/src/stratos-entity-generator';
 import { createBasicStoreModule } from '../../../store/testing/public-api';
 import { HelmReleaseActivatedRouteMock, HelmReleaseGuidMock } from '../helm/helm-testing.module';
-import { generateKubernetesEntities } from './kubernetes-entity-generator';
+import { kubeEntityCatalog } from './kubernetes-entity-generator';
 import { BaseKubeGuid } from './kubernetes-page.types';
 import { HelmReleaseHelperService } from './workloads/release/tabs/helm-release-helper.service';
 
@@ -29,7 +29,7 @@ import { HelmReleaseHelperService } from './workloads/release/tabs/helm-release-
           testEntityCatalog.clear();
           return [
             ...generateStratosEntities(),
-            ...generateKubernetesEntities(),
+            ...kubeEntityCatalog.allKubeEntities(),
           ];
         }
       }

diff --git a/src/frontend/packages/kubernetes/src/kubernetes/workloads/workloads.testing.module.ts b/src/frontend/packages/kubernetes/src/kubernetes/workloads/workloads.testing.module.ts
@@ -15,7 +15,7 @@ import { generateStratosEntities } from '../../../../store/src/stratos-entity-ge
 import { createBasicStoreModule } from '../../../../store/testing/public-api';
 import { generateHelmEntities } from '../../helm/helm-entity-generator';
 import { HelmTestingModule } from '../../helm/helm-testing.module';
-import { generateKubernetesEntities } from '../kubernetes-entity-generator';
+import { kubeEntityCatalog } from '../kubernetes-entity-generator';
 
 @NgModule({
   imports: [{
@@ -27,7 +27,7 @@ import { generateKubernetesEntities } from '../kubernetes-entity-generator';
           testEntityCatalog.clear();
           return [
             ...generateStratosEntities(),
-            ...generateKubernetesEntities(),
+            ...kubeEntityCatalog.allKubeEntities(),
             ...generateHelmEntities(),
           ];
         }