Add go test

Makefile

@@ -5,7 +5,11 @@ goas: $(GO_SOURCES)
 
 .PHONY: test
 # Check /etc/os-release to prevent non-linux from running this
-test: test1 test2 test3
+test: gotest test1 test2 test3
+
+.PHONY: gotest
+gotest:
+	go test .
 
 T1_SOURCES = $(wildcard t1/*.s)
 T1_GNU_OBJS = $(T1_SOURCES:t1/%.s=out/g/1/%.o)

main.go

@@ -619,6 +619,19 @@ func showVersion() {
 	fmt.Println("goas assembler version " + Version)
 }
 
+// For unit tests
+func EncodeString(source string) ([]byte, []byte) {
+	sc := &symbolCollection{
+		symbolsAppeared: make(map[string]bool),
+	}
+	stmts, symbolsInLexicalOrder := ParseString(source, sc)
+	textStmts, dataStmts, labeledSymbols := analyzeStatements(stmts)
+	allText := encodeAllText(textStmts, labeledSymbols)
+	allData := encodeAllData(dataStmts, labeledSymbols)
+	_ = symbolsInLexicalOrder
+	return allText, allData
+}
+
 func main() {
 	flag.Parse()
 

main_test.go

@@ -0,0 +1,30 @@
+package main
+
+import (
+	"reflect"
+	"testing"
+)
+
+func TestEncodeStringAsText(t *testing.T) {
+	tests := []struct {
+		name    string
+		source  string
+		encoded []byte
+	}{
+		// Simple instructions
+		{"ret", "ret", []byte{0xc3}},
+		{"nop", "nop", []byte{0x90}},
+		{"syscall", "syscall", []byte{0x0f, 0x05}},
+		{"leave", "leave", []byte{0xc9}},
+		{"movq 64", "movq $1, %rax", []byte{0x48, 0xc7, 0xc0, 0x01, 0x00, 0x00, 0x00}},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			encoded, data := EncodeString(tt.source)
+			_ = data
+			if !reflect.DeepEqual(encoded, tt.encoded) {
+				t.Errorf("EncodeString() got = %v, want %v", encoded, tt.encoded)
+			}
+		})
+	}
+}

parser.go

@@ -54,7 +54,7 @@ func (p *parser) readCh() byte {
 
 func (p *parser) peekCh() byte {
 	if p.idx == len(p.source) {
-		panic("Sudden EOF")
+		panic(fmt.Sprintf("Sudden EOF at idx=%d", p.idx))
 	}
 	return p.source[p.idx]
 }
@@ -580,3 +580,17 @@ func ParseFile(path string, sc *symbolCollection) []*Stmt {
 	stmts := p.parse()
 	return stmts
 }
+
+// For unit tests
+func ParseString(src string, sc *symbolCollection) ([]*Stmt, []string) {
+	p := &parser{
+		path:   "string",
+		lineno: 1,
+		source: []byte(src + "\n"),
+		idx:    0,
+		sc:     sc,
+	}
+	stmts := p.parse()
+	return stmts, sc.symbolsInLexicalOrder
+
+}

t1/05.s

@@ -3,7 +3,7 @@ nop
 syscall
 leave
 ret
-
+movq $1, %rax
 movq %rcx,0x8(%rsi)
 movq %rax, 0(%rsi)
 movq 8(%rax), %rdx