update test

Azure · Feb 27, 2024 · e0a5ce3 · e0a5ce3
1 parent e6e1fda
commit e0a5ce3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/typespec-tests/src/test/java/com/type/union/UnionsClientTest.java b/typespec-tests/src/test/java/com/type/union/UnionsClientTest.java
@@ -7,7 +7,8 @@
 import com.azure.core.util.serializer.TypeReference;
 import com.type.union.models.Cat;
 import com.type.union.models.EnumsOnlyCases;
-import com.type.union.models.LR;
+import com.type.union.models.EnumsOnlyCasesLr;
+import com.type.union.models.EnumsOnlyCasesUd;
 import com.type.union.models.MixedLiteralsCases;
 import com.type.union.models.MixedTypesCases;
 import com.type.union.models.Prop;
@@ -16,7 +17,6 @@
 import com.type.union.models.Prop3;
 import com.type.union.models.StringAndArrayCases;
 import com.type.union.models.StringExtensibleNamedUnion;
-import com.type.union.models.UD;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 
@@ -81,8 +81,8 @@ public void testModelsOnlyClient() {
     @Test
     public void testEnumsOnlyClient() {
         EnumsOnlyCases prop = client7.get().getProp();
-        Assertions.assertEquals(LR.RIGHT, LR.fromString(prop.getLr().toObject(String.class)));
-        Assertions.assertEquals(UD.UP, UD.fromString(prop.getUd().toObject(String.class)));
+        Assertions.assertEquals(EnumsOnlyCasesLr.RIGHT, prop.getLr());
+        Assertions.assertEquals(EnumsOnlyCasesUd.UP, prop.getUd());
         client7.send(prop);
     }