fix test

apache · May 5, 2021 · 01eb6dd · 01eb6dd
1 parent b7fa114
commit 01eb6dd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/schema/SchemaTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/schema/SchemaTest.java
@@ -460,9 +460,12 @@ private void testUseAutoConsumeWithSchemalessTopic(SchemaType schema) throws Exc
         if (schema == SchemaType.BYTES) {
             assertEquals(schema, message.getSchema().get().getSchemaInfo().getType());
             assertEquals(schema, message2.getSchema().get().getSchemaInfo().getType());
+        } else if (schema == SchemaType.NONE) {
+            // schema NONE is always reported as BYTES
+            assertEquals(SchemaType.BYTES, message.getSchema().get().getSchemaInfo().getType());
+            assertEquals(SchemaType.BYTES, message2.getSchema().get().getSchemaInfo().getType());
         } else {
-            assertFalse(message.getSchema().isPresent());
-            assertEquals(schema, message2.getSchema().get().getSchemaInfo().getType());
+            fail();
         }
 
         assertEquals("foo".getBytes(StandardCharsets.UTF_8), message.getValue());