polemos/pcsdk/commands_test.go

package pcsdk

import (
	"encoding/json"
	"net/netip"
	"testing"
	"github.com/google/uuid"
	"github.com/thefeli73/polemos/state"
)

func TestCommandCreateJsonParse(t *testing.T) {
	ip, _ := netip.ParseAddr("127.0.0.99")
	id, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
	uuid := state.CustomUUID(id)
	m := create(5555, 6666, ip, uuid)
	t.Logf("%+v\n", m)
	msg, err := json.Marshal(m)
	if err != nil {
		t.Fatalf(`%q`, err)
	}

	expected := "{\"create\":{\"incoming_port\":5555,\"destination_port\":6666,\"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 TestCommandModifyJsonParse(t *testing.T) {
	ip, _ := netip.ParseAddr("127.0.0.99")
	id, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
	uuid := state.CustomUUID(id)
	m := modify(8888, ip, 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) {
	id, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
	uuid := state.CustomUUID(id)
	m := delete(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)
	}
}