Add test for getControlSchema

xlate · May 11, 2020 · ea09b66 · ea09b66
1 parent 088edd2
commit ea09b66
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/test/java/io/xlate/edi/internal/schema/StaEDISchemaFactoryTest.java b/src/test/java/io/xlate/edi/internal/schema/StaEDISchemaFactoryTest.java
@@ -27,6 +27,7 @@
 import org.junit.jupiter.api.Test;
 
 import io.xlate.edi.schema.EDISchemaException;
+import io.xlate.edi.schema.EDIType;
 import io.xlate.edi.schema.Schema;
 import io.xlate.edi.schema.SchemaFactory;
 import io.xlate.edi.stream.EDIStreamConstants.Standards;
@@ -253,4 +254,14 @@ public void testDuplicateElementTypeNames() {
         assertNotNull(cause);
         assertEquals("duplicate name: DE0004", cause.getMessage());
     }
+
+    @Test
+    public void testGetControlSchema() throws EDISchemaException {
+        SchemaFactory factory = SchemaFactory.newFactory();
+        Schema schema = factory.getControlSchema(Standards.X12, new String[] { "00501" });
+        assertNotNull(schema);
+        assertEquals(EDIType.Type.SEGMENT, schema.getType("ISA").getType());
+        assertEquals(EDIType.Type.SEGMENT, schema.getType("GS").getType());
+        assertEquals(EDIType.Type.SEGMENT, schema.getType("ST").getType());
+    }
 }