From 4955f66ad63c6156cee478afd59a2d1cc976e356 Mon Sep 17 00:00:00 2001
From: Sasha Koshka <sashakoshka@tebibyte.media>
Date: Thu, 17 Jul 2025 14:06:05 -0400
Subject: [PATCH] generate: testEncode uses snakes now

---
 generate/generate_test.go | 2 +-
 generate/misc_test.go     | 9 ++++-----
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/generate/generate_test.go b/generate/generate_test.go
index a25b807..5f060e8 100644
--- a/generate/generate_test.go
+++ b/generate/generate_test.go
@@ -231,6 +231,6 @@ func TestGenerateRun(test *testing.T) {
 		}
 		testEncode(
 			&messageConnect,
-			0x0) // TODO
+			tu.S(0x0)) // TODO
 	`)
 }
diff --git a/generate/misc_test.go b/generate/misc_test.go
index eabb00b..44f32ef 100644
--- a/generate/misc_test.go
+++ b/generate/misc_test.go
@@ -34,26 +34,25 @@ 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 ...byte) {
+		func testEncode(message Message, correct tu.Snake) {
 			buffer := bytes.Buffer { }
 			encoder := tape.NewEncoder(&buffer)
 			n, err := message.Encode(encoder)
 			if err != nil { log.Fatalf("at %d: %v\n", n, err) }
 			got := buffer.Bytes()
 			log.Println("got:    ", tu.HexBytes(got))
-			log.Println("correct:", tu.HexBytes(correct))
+			log.Println("correct:", correct)
 			if n != len(got) {
 				log.Fatalf("n incorrect: %d != %d\n", n, len(got))
 			}
-			if !slices.Equal(got, correct) {
-				log.Fatalln("not equal")
+			if ok, n := correct.Check(got); !ok {
+				log.Fatalln("not equal at", n)
 			}
 		}
 		`