From 4e90af0de903c74cebb16210f3ffb68f956b19ff Mon Sep 17 00:00:00 2001
From: Eduard Urbach <eduard@urbach.dev>
Date: Mon, 30 Jun 2025 16:03:16 +0200
Subject: [PATCH] Improved asm test coverage

---
 src/asm/Assembler_test.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/asm/Assembler_test.go b/src/asm/Assembler_test.go
index 15d4df8..3a3383d 100644
--- a/src/asm/Assembler_test.go
+++ b/src/asm/Assembler_test.go
@@ -11,7 +11,13 @@ import (
 func TestAssembler(t *testing.T) {
 	a := &asm.Assembler{}
 	a.Append(&asm.Label{Name: "a"})
+	a.Append(&asm.FunctionStart{})
 	a.Append(&asm.Call{Label: "b"})
+	a.Append(&asm.Call{Label: "c"})
+	a.Append(&asm.MoveRegisterLabel{Label: "b"})
+	a.Append(&asm.MoveRegisterNumber{Destination: 0, Number: 123})
+	a.Append(&asm.MoveRegisterRegister{Destination: 0, Source: 1})
+	a.Append(&asm.FunctionEnd{})
 	a.Append(&asm.Return{})
 
 	b := &asm.Assembler{}
@@ -19,9 +25,17 @@ func TestAssembler(t *testing.T) {
 	b.Append(&asm.Syscall{})
 	b.Append(&asm.Return{})
 
+	c := &asm.Assembler{}
+	c.Append(&asm.Label{Name: "c"})
+	c.Append(&asm.Jump{Label: "branch2"})
+	c.Append(&asm.Label{Name: "branch1"})
+	c.Append(&asm.Label{Name: "branch2"})
+	c.Append(&asm.Return{})
+
 	final := asm.Assembler{}
 	final.Merge(a)
 	final.Merge(b)
+	final.Merge(c)
 
 	code, _ := final.Compile(&build.Build{Arch: build.ARM})
 	assert.NotNil(t, code)