schulze/polemos
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add tests for modify and delete

Browse Source
This commit is contained in:
Erik 2023-04-19 14:00:41 +02:00
parent 1a31a304d6
commit e3e72de1f4
1 changed files with 30 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
commands/commands_test.go
View File

@ -12,8 +12,7 @@ import (
func TestCommandCreateJsonParse(t *testing.T) { func TestCommandCreateJsonParse(t *testing.T) {
ip, _ := netip.ParseAddr("127.0.0.99") ip, _ := netip.ParseAddr("127.0.0.99")
uuid, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1") uuid, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
c := NewCommandCreate(5555, 6666, ip, state.CustomUUID(uuid)) m := NewCommandCreate(5555, 6666, ip, state.CustomUUID(uuid))
m := ProxyCommandCreate{c}
msg, err := json.Marshal(m) msg, err := json.Marshal(m)
if err != nil { if err != nil {
t.Fatalf(`%q`, err) t.Fatalf(`%q`, err)
@ -26,4 +25,33 @@ func TestCommandCreateJsonParse(t *testing.T) {
} }
} }
func TestCommandModifyJsonParse(t *testing.T) {
ip, _ := netip.ParseAddr("127.0.0.99")
uuid, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
m := NewCommandModify(8888, ip, state.CustomUUID(uuid))
msg, err := json.Marshal(m)
if err != nil {
t.Fatalf(`%q`, err)
}
expected := "{\"modify\":{\"destination_port\":8888,\"destination_ip\":\"127.0.0.99\",\"id\":\"87e79cbc-6df6-4462-8412-85d6c473e3b1\"}}"
if string(msg) != expected {
t.Fatalf(
"\nExpected:\t %q\nGot:\t\t %q\n", expected, msg)
}
}
func TestCommandDeleteJsonParse(t *testing.T) {
uuid, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
m := NewCommandDelete(state.CustomUUID(uuid))
msg, err := json.Marshal(m)
if err != nil {
t.Fatalf(`%q`, err)
}
expected := "{\"delete\":{\"id\":\"87e79cbc-6df6-4462-8412-85d6c473e3b1\"}}"
if string(msg) != expected {
t.Fatalf(
"\nExpected:\t %q\nGot:\t\t %q\n", expected, msg)
}
}