1 files changed, 29 insertions, 0 deletions
diff --git a/assembler/src/parser/code_gen.rs b/assembler/src/parser/code_gen.rs
index 2eaa34c..d183145 100644
--- a/assembler/src/parser/code_gen.rs
+++ b/assembler/src/parser/code_gen.rs
@@ -88,3 +88,32 @@ impl Program {
             .collect()
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::lexer::Token;
+
+    #[test]
+    fn after_two_collapses_only_subleq_remains() {
+        let mut prog = Program::new(vec![
+            Token::Operator("rts".to_string()),
+            Token::Operator("jsr".to_string()),
+            Token::Address(1000),
+            Token::Operator("mov".to_string()),
+            Token::Address(1001),
+            Token::Address(1002),
+            Token::Operator("subleq".to_string()),
+            Token::Address(1001),
+            Token::Address(1002),
+            Token::Address(1001),
+
+        ], 10);
+        prog.collapse_instructions();
+        prog.collapse_instructions();
+
+        for i in prog.instructions {
+            assert_eq!(i.name, "subleq");
+        }
+    }
+}