2 files changed, 8 insertions, 29 deletions
diff --git a/azalea-brigadier/tests/builder/argument_builder_test.rs b/azalea-brigadier/tests/builder/argument_builder_test.rs
index ee44f5e6..d5f940dd 100755
--- a/azalea-brigadier/tests/builder/argument_builder_test.rs
+++ b/azalea-brigadier/tests/builder/argument_builder_test.rs
@@ -1,41 +1,17 @@
 use std::rc::Rc;
 
-use crate::{
-    arguments::integer_argument_type::integer,
-    builder::{literal_argument_builder::literal, required_argument_builder::argument},
-};
-
-use super::ArgumentBuilder;
-
-// public class ArgumentBuilderTest {
-//     private TestableArgumentBuilder<Object> builder;
-
-//     @Before
-//     public void setUp() throws Exception {
-//         builder = new TestableArgumentBuilder<>();
-//     }
-
-//     @Test
-//     public void testArguments() throws Exception {
-//         final RequiredArgumentBuilder<Object, ?> argument = argument("bar",
-// integer());
-
-//         builder.then(argument);
-
-//         assertThat(builder.getArguments(), hasSize(1));
-//         assertThat(builder.getArguments(), hasItem((CommandNode<Object>)
-// argument.build()));     }
+use azalea_brigadier::{builder::argument_builder::ArgumentBuilder, prelude::*};
 
 #[test]
 fn test_arguments() {
-    let mut builder: ArgumentBuilder<()> = literal("foo");
+    let builder: ArgumentBuilder<()> = literal("foo");
 
     let argument: ArgumentBuilder<()> = argument("bar", integer());
-    builder.then(argument.clone());
-    assert_eq!(builder.arguments.children.len(), 1);
+    let builder = builder.then(argument.clone());
+    assert_eq!(builder.arguments().children.len(), 1);
     let built_argument = Rc::new(argument.build());
     assert!(builder
-        .arguments
+        .arguments()
         .children
         .values()
         .any(|e| *e.read() == *built_argument));
diff --git a/azalea-brigadier/tests/builder/mod.rs b/azalea-brigadier/tests/builder/mod.rs
new file mode 100644
index 00000000..21944c68
--- /dev/null
+++ b/azalea-brigadier/tests/builder/mod.rs
@@ -0,0 +1,3 @@
+mod argument_builder_test;
+mod literal_argument_builder_test;
+mod required_argument_builder_test;