ST: cast storage type correctly in OpenShiftTemplatesST (strimzi#1576)

carlobongiovanni · Apr 25, 2019 · c001e72 · c001e72
1 parent 40202af
commit c001e72
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/systemtest/src/test/java/io/strimzi/systemtest/OpenShiftTemplatesST.java b/systemtest/src/test/java/io/strimzi/systemtest/OpenShiftTemplatesST.java
@@ -13,6 +13,7 @@
 import io.strimzi.api.kafka.model.DoneableKafkaConnect;
 import io.strimzi.api.kafka.model.DoneableKafkaConnectS2I;
 import io.strimzi.api.kafka.model.DoneableKafkaTopic;
+import io.strimzi.api.kafka.model.JbodStorage;
 import io.strimzi.api.kafka.model.Kafka;
 import io.strimzi.api.kafka.model.KafkaConnect;
 import io.strimzi.api.kafka.model.KafkaConnectS2I;
@@ -96,7 +97,7 @@ void testStrimziPersistent() {
         assertNotNull(kafka);
         assertEquals(1, kafka.getSpec().getKafka().getReplicas());
         assertEquals(1, kafka.getSpec().getZookeeper().getReplicas());
-        assertEquals("persistent-claim", kafka.getSpec().getKafka().getStorage().getType());
+        assertEquals("jbod", kafka.getSpec().getKafka().getStorage().getType());
         assertEquals("persistent-claim", kafka.getSpec().getZookeeper().getStorage().getType());
     }
 
@@ -160,7 +161,7 @@ void testStrimziPersistentWithCustomParameters() {
         assertEquals("2", kafka.getSpec().getKafka().getConfig().get("default.replication.factor"));
         assertEquals("5", kafka.getSpec().getKafka().getConfig().get("offsets.topic.replication.factor"));
         assertEquals("5", kafka.getSpec().getKafka().getConfig().get("transaction.state.log.replication.factor"));
-        assertEquals("2Gi", ((PersistentClaimStorage) kafka.getSpec().getKafka().getStorage()).getSize());
+        assertEquals("2Gi", ((PersistentClaimStorage) ((JbodStorage) kafka.getSpec().getKafka().getStorage()).getVolumes().get(0)).getSize());
         assertEquals("2Gi", ((PersistentClaimStorage) kafka.getSpec().getZookeeper().getStorage()).getSize());
     }