WIP

peer/network_state_test.go

@@ -0,0 +1,54 @@
+package peer
+
+import (
+	"path/filepath"
+	"reflect"
+	"testing"
+
+	"vppn/m"
+)
+
+func TestNetworkState_RoundTrip(t *testing.T) {
+	path := filepath.Join(t.TempDir(), "network.json")
+
+	var state m.NetworkState
+	state.Peers[1] = &m.Peer{
+		PeerIP:     1,
+		Version:    7,
+		Name:       "hub",
+		Addr4:      []byte{10, 11, 12, 1},
+		Port:       51820,
+		Relay:      true,
+		WGPubKey:   make([]byte, 32),
+		SignPubKey: make([]byte, 32),
+	}
+	state.Peers[10] = &m.Peer{
+		PeerIP:     10,
+		Version:    3,
+		Name:       "laptop",
+		Addr4:      []byte{10, 11, 12, 10},
+		Port:       51820,
+		WGPubKey:   []byte("0123456789abcdef0123456789abcdef"),
+		SignPubKey: []byte("fedcba9876543210fedcba9876543210"),
+	}
+
+	if err := saveNetworkState(path, state); err != nil {
+		t.Fatal(err)
+	}
+
+	got, err := loadNetworkState(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if !reflect.DeepEqual(got, state) {
+		t.Errorf("round-trip mismatch:\n got: %+v\nwant: %+v", got.Peers[1], state.Peers[1])
+	}
+}
+
+func TestNetworkState_LoadMissing(t *testing.T) {
+	path := filepath.Join(t.TempDir(), "does-not-exist.json")
+	if _, err := loadNetworkState(path); err == nil {
+		t.Fatal("expected error loading missing cache, got nil")
+	}
+}