mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-26 19:13:34 +00:00
81 lines
2.4 KiB
Go
81 lines
2.4 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/netip"
|
|
"testing"
|
|
|
|
"github.com/psviderski/uncloud/internal/machine/api/pb"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMachineMembersList_Info(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
publicKey := []byte{0x01, 0x02, 0x03, 0x04}
|
|
members := MachineMembersList{
|
|
{
|
|
Machine: &pb.MachineInfo{
|
|
Id: "abc123",
|
|
Name: "vm-1",
|
|
Hostname: "vm-1.example.com",
|
|
Arch: "amd64",
|
|
OsPrettyName: "Ubuntu 24.04.4 LTS",
|
|
KernelVersion: "6.8.0-31-generic",
|
|
DockerVersion: "27.1.1",
|
|
DaemonVersion: "0.9.0",
|
|
PublicIp: pb.NewIP(netip.MustParseAddr("203.0.113.5")),
|
|
Network: &pb.NetworkConfig{
|
|
Subnet: pb.NewIPPrefix(netip.MustParsePrefix("10.210.0.0/24")),
|
|
ManagementIp: pb.NewIP(netip.MustParseAddr("10.210.0.1")),
|
|
Endpoints: []*pb.IPPort{
|
|
pb.NewIPPort(netip.MustParseAddrPort("203.0.113.5:51820")),
|
|
},
|
|
PublicKey: publicKey,
|
|
},
|
|
},
|
|
State: pb.MachineMember_UP,
|
|
},
|
|
}
|
|
|
|
infos := members.ToNative()
|
|
require.Len(t, infos, 1)
|
|
|
|
assert.Equal(t, MachineMember{
|
|
ID: "abc123",
|
|
Name: "vm-1",
|
|
State: "Up",
|
|
Hostname: "vm-1.example.com",
|
|
Arch: "amd64",
|
|
OSPrettyName: "Ubuntu 24.04.4 LTS",
|
|
KernelVersion: "6.8.0-31-generic",
|
|
DockerVersion: "27.1.1",
|
|
DaemonVersion: "0.9.0",
|
|
PublicIP: netip.MustParseAddr("203.0.113.5"),
|
|
Network: MachineNetwork{
|
|
Subnet: netip.MustParsePrefix("10.210.0.0/24"),
|
|
ManagementIP: netip.MustParseAddr("10.210.0.1"),
|
|
Endpoints: []netip.AddrPort{netip.MustParseAddrPort("203.0.113.5:51820")},
|
|
PublicKey: publicKey,
|
|
},
|
|
}, infos[0])
|
|
|
|
// The JSON output must use PascalCase keys to stay consistent with `docker inspect` and the
|
|
// rest of Uncloud's JSON output.
|
|
data, err := json.Marshal(infos)
|
|
require.NoError(t, err)
|
|
out := string(data)
|
|
for _, key := range []string{
|
|
`"ID"`, `"Name"`, `"State"`, `"Hostname"`, `"Arch"`, `"OSPrettyName"`, `"KernelVersion"`,
|
|
`"DockerVersion"`, `"DaemonVersion"`, `"PublicIP"`, `"Network"`,
|
|
`"Subnet"`, `"ManagementIP"`, `"Endpoints"`, `"PublicKey"`,
|
|
} {
|
|
assert.Contains(t, out, key)
|
|
}
|
|
assert.Contains(t, out, `"PublicIP":"203.0.113.5"`)
|
|
assert.Contains(t, out, `"Subnet":"10.210.0.0/24"`)
|
|
assert.Contains(t, out, `"Endpoints":["203.0.113.5:51820"]`)
|
|
assert.Contains(t, out, `"PublicKey":"AQIDBA=="`)
|
|
}
|