mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-02 22:45:37 +00:00
83efadee9f
Instead of converting our PortMap struct to a string during marshalling for use as a key, convert the whole collection of PortMaps to a list of PortMap objects, which improves the readability of the JSON config while still keeping the data structure we need in the code. Updates #13406 Signed-off-by: Tom Proctor <tomhjp@users.noreply.github.com>
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package egressservices
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
func Test_jsonUnmarshalConfig(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
bs []byte
|
|
wantsCfg Config
|
|
wantsErr bool
|
|
}{
|
|
{
|
|
name: "success",
|
|
bs: []byte(`{"ports":[{"protocol":"tcp","matchPort":4003,"targetPort":80}]}`),
|
|
wantsCfg: Config{Ports: map[PortMap]struct{}{{Protocol: "tcp", MatchPort: 4003, TargetPort: 80}: {}}},
|
|
},
|
|
{
|
|
name: "failure_invalid_format",
|
|
bs: []byte(`{"ports":{"tcp:80":{}}}`),
|
|
wantsCfg: Config{Ports: map[PortMap]struct{}{}},
|
|
wantsErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
cfg := Config{}
|
|
if gotErr := json.Unmarshal(tt.bs, &cfg); (gotErr != nil) != tt.wantsErr {
|
|
t.Errorf("json.Unmarshal returned error %v, wants error %v", gotErr, tt.wantsErr)
|
|
}
|
|
if diff := cmp.Diff(cfg, tt.wantsCfg); diff != "" {
|
|
t.Errorf("unexpected secrets (-got +want):\n%s", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_jsonMarshalConfig(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
protocol string
|
|
matchPort uint16
|
|
targetPort uint16
|
|
wantsBs []byte
|
|
}{
|
|
{
|
|
name: "success",
|
|
protocol: "tcp",
|
|
matchPort: 4003,
|
|
targetPort: 80,
|
|
wantsBs: []byte(`{"tailnetTarget":{"ip":"","fqdn":""},"ports":[{"protocol":"tcp","matchPort":4003,"targetPort":80}]}`),
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
cfg := Config{Ports: PortMaps{{
|
|
Protocol: tt.protocol,
|
|
MatchPort: tt.matchPort,
|
|
TargetPort: tt.targetPort}: {}}}
|
|
|
|
gotBs, gotErr := json.Marshal(&cfg)
|
|
if gotErr != nil {
|
|
t.Errorf("json.Marshal(%+#v) returned unexpected error %v", cfg, gotErr)
|
|
}
|
|
if diff := cmp.Diff(gotBs, tt.wantsBs); diff != "" {
|
|
t.Errorf("unexpected secrets (-got +want):\n%s", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|