Add more tests on catalog

amazon-ion · Apr 8, 2024 · 5dd9664 · 5dd9664
1 parent 6374884
commit 5dd9664
Showing 1 changed file with 24 additions and 9 deletions.
diff --git a/src/test/java/com/amazon/ion/system/IonSystemBuilderTest.java b/src/test/java/com/amazon/ion/system/IonSystemBuilderTest.java
@@ -138,28 +138,43 @@ public void testStreamCopyOptimizedImmutability()
     @Test
     public void testIonTextWriterBuilder()
     {
-        IonTextWriterBuilder textWriterBuilder = IonTextWriterBuilder.standard();
+        IonCatalog textWriterCatalog = new SimpleCatalog(); // should be ignored
+        IonTextWriterBuilder textWriterBuilder = IonTextWriterBuilder.standard().withCatalog(textWriterCatalog);
 
-        IonSystemBuilder b = IonSystemBuilder.standard().withIonTextWriterBuilder(textWriterBuilder);
-        assertSame(textWriterBuilder, b.getIonTextWriterBuilder());
+        IonSystemBuilder systemBuilder = IonSystemBuilder.standard().withIonTextWriterBuilder(textWriterBuilder);
+        assertSame(textWriterBuilder, systemBuilder.getIonTextWriterBuilder());
+
+        IonCatalog systemCatalog = new SimpleCatalog();
+        IonSystem system = systemBuilder.withCatalog(systemCatalog).build();
+        assertSame(systemCatalog, system.getCatalog());
     }
 
     @Test
     public void testIonBinaryWriterBuilder()
     {
-        IonBinaryWriterBuilder binaryWriterBuilder = IonBinaryWriterBuilder.standard();
+        IonCatalog binaryWriterCatalog = new SimpleCatalog(); // should be ignored
+        IonBinaryWriterBuilder binaryWriterBuilder = IonBinaryWriterBuilder.standard().withCatalog(binaryWriterCatalog);
+
+        IonSystemBuilder systemBuilder = IonSystemBuilder.standard().withIonBinaryWriterBuilder(binaryWriterBuilder);
+        assertSame(binaryWriterBuilder, systemBuilder.getIonBinaryWriterBuilder());
 
-        IonSystemBuilder b = IonSystemBuilder.standard().withIonBinaryWriterBuilder(binaryWriterBuilder);
-        assertSame(binaryWriterBuilder, b.getIonBinaryWriterBuilder());
+        IonCatalog systemCatalog = new SimpleCatalog();
+        IonSystem system = systemBuilder.build();
+        assertSame(systemCatalog, system.getCatalog());
     }
 
     @Test
     public void testIonReaderBuilder()
     {
-        IonReaderBuilder readerBuilder = IonReaderBuilder.standard();
+        IonCatalog readerCatalog = new SimpleCatalog(); // should be ignored
+        IonReaderBuilder readerBuilder = IonReaderBuilder.standard().withCatalog(readerCatalog);
+
+        IonSystemBuilder systemBuilder = IonSystemBuilder.standard().withReaderBuilder(readerBuilder);
+        assertSame(readerBuilder, systemBuilder.getReaderBuilder());
 
-        IonSystemBuilder b = IonSystemBuilder.standard().withReaderBuilder(readerBuilder);
-        assertSame(readerBuilder, b.getReaderBuilder());
+        IonCatalog systemCatalog = new SimpleCatalog();
+        IonSystem system = systemBuilder.build();
+        assertSame(systemCatalog, system.getCatalog());
     }
 
     //-------------------------------------------------------------------------