Use builder for Rust structs

1 files changed, 8 insertions, 8 deletions

diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustWriterTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustWriterTests.java
index 8062fb9..c41bb9e 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustWriterTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/output/lang/rust/RustWriterTests.java
@@ -25,8 +25,8 @@ public class RustWriterTests {
     @Test
     public void module() throws IOException {
         var rust = new RustModule(List.of(new RustIdentifier("shapes")), List.of(
-            new RustStructStruct(new RustIdentifier("Rectangle"), List.of()),
-            new RustStructStruct(new RustIdentifier("Square"), List.of())
+            RustStructStruct.arbitrary().withName(new RustIdentifier("Rectangle")).build(),
+            RustStructStruct.arbitrary().withName(new RustIdentifier("Square")).build()
         ));
 
         var string = write(writer -> writer.writeModule(rust));
@@ -42,7 +42,7 @@ public class RustWriterTests {
 
     @Test
     public void emptyStruct() throws IOException {
-        var rust = new RustStructStruct(new RustIdentifier("Rectangle"), List.of());
+        var rust = RustStructStruct.arbitrary().withName(new RustIdentifier("Rectangle")).build();
 
         var string = write(writer -> writer.writeItem(rust));
 
@@ -53,13 +53,13 @@ public class RustWriterTests {
 
     @Test
     public void structWithFields() throws IOException {
-        var rust = new RustStructStruct(
-            new RustIdentifier("Rectangle"),
-            List.of(
+        var rust = RustStructStruct.arbitrary()
+            .withName(new RustIdentifier("Rectangle"))
+            .withFields(List.of(
                 new RustStructField(new RustIdentifier("width"), RustTypePath.primitive("i32")),
                 new RustStructField(new RustIdentifier("height"), RustTypePath.primitive("i32"))
-            )
-        );
+            ))
+            .build();
 
         var string = write(writer -> writer.writeItem(rust));