generate/generate_test.go

@@ -231,6 +231,6 @@ func TestGenerateRun(test *testing.T) {
 		}
 		testEncode(
 			&messageConnect,
-			tu.S(0x0)) // TODO
+			0x0) // TODO
 	`)
 }

generate/misc_test.go

@@ -34,26 +34,26 @@ func testGenerateRun(test *testing.T, protocol *Protocol, imports string, testCa
 	imports = `
 		import "log"
 		import "bytes"
+		import "slices"
 		import "git.tebibyte.media/sashakoshka/hopp/tape"
 		import tu "git.tebibyte.media/sashakoshka/hopp/internal/testutil"
 		` + imports
 	setup := `log.Println("*** BEGIN TEST CASE OUTPUT ***")`
 	teardown := `log.Println("--- END TEST CASE OUTPUT ---")`
 	static := `
-		func testEncode(message Message, correct tu.Snake) {
+		func testEncode(message Message, correct ...byte) {
 			buffer := bytes.Buffer { }
 			encoder := tape.NewEncoder(&buffer)
 			n, err := message.Encode(encoder)
 			if err != nil { log.Fatalf("at %d: %v\n", n, err) }
-			encoder.Flush()
 			got := buffer.Bytes()
 			log.Println("got:    ", tu.HexBytes(got))
-			log.Println("correct:", correct)
+			log.Println("correct:", tu.HexBytes(correct))
 			if n != len(got) {
 				log.Fatalf("n incorrect: %d != %d\n", n, len(got))
 			}
-			if ok, n := correct.Check(got); !ok {
-				log.Fatalln("not equal at", n)
+			if !slices.Equal(got, correct) {
+				log.Fatalln("not equal")
 			}
 		}
 		`