sashakoshka/hopp
sashakoshka/hopp
1
0
Fork 0
Code Issues 8 Pull Requests Actions Packages Projects Releases Wiki Activity

generate: Add support for recursive snake in test infrastructure

Browse Source
This commit is contained in:
Sasha Koshka 2025-10-29 15:19:27 -04:00
parent 67881a455a
commit 10d84c2184
1 changed files with 25 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
generate/misc_test.go
View File

@ -37,6 +37,7 @@ func testGenerateRun(test *testing.T, protocol *Protocol, title, imports, testCa
import "reflect" import "reflect"
import "git.tebibyte.media/sashakoshka/hopp/tape" import "git.tebibyte.media/sashakoshka/hopp/tape"
import tu "git.tebibyte.media/sashakoshka/hopp/internal/testutil" import tu "git.tebibyte.media/sashakoshka/hopp/internal/testutil"
import "git.tebibyte.media/sashakoshka/hopp/internal/testutil/snake"
` + imports ` + imports
setup := `log.Println("*** BEGIN TEST CASE OUTPUT ***")` setup := `log.Println("*** BEGIN TEST CASE OUTPUT ***")`
teardown := `log.Println("--- END TEST CASE OUTPUT ---")` teardown := `log.Println("--- END TEST CASE OUTPUT ---")`
@ -61,8 +62,9 @@ func testGenerateRun(test *testing.T, protocol *Protocol, title, imports, testCa
func testDecode(correct Message, data any) { func testDecode(correct Message, data any) {
var flat []byte var flat []byte
switch data := data.(type) { switch data := data.(type) {
case []byte: flat = data case []byte: flat = data
case tu.Snake: flat = data.Flatten() case tu.Snake: flat = data.Flatten()
case snake.Snake: flat = data.Flatten()
} }
message := reflect.New(reflect.ValueOf(correct).Elem().Type()).Interface().(Message) message := reflect.New(reflect.ValueOf(correct).Elem().Type()).Interface().(Message)
log.Println("before: ", message) log.Println("before: ", message)
@ -79,9 +81,7 @@ func testGenerateRun(test *testing.T, protocol *Protocol, title, imports, testCa
} }
} }
// TODO: possibly combine the two above functions into this one, func testEncodeDecode(message Message, data any) {buffer := bytes.Buffer { }
// also take a data parameter here (snake)
func testEncodeDecode(message Message, data tu.Snake) {buffer := bytes.Buffer { }
log.Println("encoding:") log.Println("encoding:")
encoder := tape.NewEncoder(&buffer) encoder := tape.NewEncoder(&buffer)
n, err := message.Encode(encoder) n, err := message.Encode(encoder)
@ -93,13 +93,30 @@ func testGenerateRun(test *testing.T, protocol *Protocol, title, imports, testCa
if n != len(got) { if n != len(got) {
log.Fatalf("n incorrect: %d != %d\n", n, len(got)) log.Fatalf("n incorrect: %d != %d\n", n, len(got))
} }
if ok, n := data.Check(got); !ok {
log.Fatalln("not equal at", n) var flat []byte
switch data := data.(type) {
case []byte:
flat = data
if ok, n := snake.Check(snake.L(data...), got); !ok {
log.Fatalln("not equal at", n)
}
case tu.Snake:
flat = data.Flatten()
if ok, n := data.Check(got); !ok {
log.Fatalln("not equal at", n)
}
case snake.Node:
flat = data.Flatten()
if ok, n := snake.Check(data, got); !ok {
log.Fatalln("not equal at", n)
}
default:
panic("AUSIAUGH AAAUUGUHGHGHH OUHGHGJDSGK")
} }
log.Println("decoding:") log.Println("decoding:")
destination := reflect.New(reflect.ValueOf(message).Elem().Type()).Interface().(Message) destination := reflect.New(reflect.ValueOf(message).Elem().Type()).Interface().(Message)
flat := data.Flatten()
log.Println("before: ", tu.Describe(destination)) log.Println("before: ", tu.Describe(destination))
decoder := tape.NewDecoder(bytes.NewBuffer(flat)) decoder := tape.NewDecoder(bytes.NewBuffer(flat))
n, err = destination.Decode(decoder) n, err = destination.Decode(decoder)