From c7ebc2b0d6cc7ddd33f3b6b0de429e1813bdbad0 Mon Sep 17 00:00:00 2001
From: Eric Barroca <ebarroca@vertesiahq.com>
Date: Fri, 17 Jul 2026 03:16:12 -0300
Subject: [PATCH] fix: test a current generated union

---
 .../io/vertesia/GeneratedCompatibilityTest.java  | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/test/java/io/vertesia/GeneratedCompatibilityTest.java b/src/test/java/io/vertesia/GeneratedCompatibilityTest.java
index c611e60..a3223cb 100644
--- a/src/test/java/io/vertesia/GeneratedCompatibilityTest.java
+++ b/src/test/java/io/vertesia/GeneratedCompatibilityTest.java
@@ -9,7 +9,8 @@
 import io.vertesia.model.AccountType;
 import io.vertesia.model.ContentObjectStatus;
 import io.vertesia.model.ContentObjectType;
-import io.vertesia.model.DocTableResponse;
+import io.vertesia.model.DataStoreFullSchemaResponse;
+import io.vertesia.model.DataStoreSchemaResponse;
 import org.junit.jupiter.api.Test;
 
 class GeneratedCompatibilityTest {
@@ -76,16 +77,19 @@ void generatedClientRejectsDisabledTlsVerification() {
 
     @Test
     void generatedUnionsHandleKnownShapeWithUnknownFields() {
-        DocTableResponse response =
+        DataStoreSchemaResponse response =
                 JSON.deserialize(
                         "{"
-                                + "\"format\":\"csv\","
-                                + "\"data\":\"name,value\\nalpha,1\","
+                                + "\"schema_format\":\"full\","
+                                + "\"version\":\"1.0.0\","
+                                + "\"tables\":[],"
+                                + "\"relationships\":[],"
+                                + "\"updated_at\":\"2026-01-01T00:00:00.000Z\","
                                 + "\"server_added_field\":\"ignored\""
                                 + "}",
-                        DocTableResponse.class);
+                        DataStoreSchemaResponse.class);
 
         assertNotNull(response);
-        assertNotNull(response.getActualInstance());
+        assertTrue(response.getActualInstance() instanceof DataStoreFullSchemaResponse);
     }
 }