fix(knative-icons): show icons for event sources

openshift · Nov 28, 2019 · bda91a4 · bda91a4
1 parent 2c9d044
commit bda91a4
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/frontend/packages/knative-plugin/src/utils/get-knative-icon.ts b/frontend/packages/knative-plugin/src/utils/get-knative-icon.ts
@@ -4,18 +4,26 @@ import * as camelSourceImg from '../imgs/logos/camelsource.svg';
 import * as containerSourceImg from '../imgs/logos/containersource.png';
 import * as cronJobSourceImg from '../imgs/logos/cronjobsource.png';
 import * as kafkaSourceImg from '../imgs/logos/kafkasource.svg';
+import { referenceForModel } from '@console/internal/module/k8s';
+import {
+  EventSourceCronJobModel,
+  EventSourceContainerModel,
+  EventSourceApiServerModel,
+  EventSourceCamelModel,
+  EventSourceKafkaModel,
+} from '../models';
 
 export const getKnativeEventSourceIcon = (kind: string): string => {
   switch (kind) {
-    case 'ApiServerSource':
+    case referenceForModel(EventSourceApiServerModel):
       return apiServerSourceImg;
-    case 'CamelSource':
+    case referenceForModel(EventSourceCamelModel):
       return camelSourceImg;
-    case 'ContainerSource':
+    case referenceForModel(EventSourceContainerModel):
       return containerSourceImg;
-    case 'CronJobSource':
+    case referenceForModel(EventSourceCronJobModel):
       return cronJobSourceImg;
-    case 'KafkaSource':
+    case referenceForModel(EventSourceKafkaModel):
       return kafkaSourceImg;
     default:
       return openshiftImg;