blob: bae9318228956a490e894ef9804e3cd705d8eee9 [file] [log] [blame]
// Copyright 2021 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package main
import (
"testing"
"github.com/google/go-cmp/cmp"
)
func TestHostTypeToVersion(t *testing.T) {
testCases := []struct {
desc string
hostType string
wantVersion *Version
}{
{
desc: "valid original darwin host type",
hostType: "host-darwin-10_11",
wantVersion: &Version{
Major: 10,
Minor: 11,
Arch: "amd64",
},
},
{
desc: "valid original darwin host type",
hostType: "host-darwin-10_12",
wantVersion: &Version{
Major: 10,
Minor: 12,
Arch: "amd64",
},
},
{
desc: "valid newer darwin host ARM64",
hostType: "host-darwin-arm64-11_0",
wantVersion: &Version{
Major: 11,
Minor: 0,
Arch: "arm64",
},
},
{
desc: "valid newer darwin host AMD64",
hostType: "host-darwin-amd64-11_1",
wantVersion: &Version{
Major: 11,
Minor: 1,
Arch: "amd64",
},
},
}
for _, tc := range testCases {
t.Run(tc.desc, func(t *testing.T) {
got, err := hostTypeToVersion(tc.hostType)
if err != nil {
t.Fatalf("hostTypeToVersion(%q) = %v, %s; want no error", tc.hostType, got, err)
}
if diff := cmp.Diff(tc.wantVersion, got); diff != "" {
t.Errorf("hostTypeToVersion(%q) = (-want +got):\n%s", tc.hostType, diff)
}
})
}
}
func TestHostTypeToVersionError(t *testing.T) {
testCases := []struct {
desc string
hostType string
wantErr error
}{
{
desc: "empty string",
hostType: "",
},
{
desc: "invalid newer darwin host ARM64",
hostType: "host-darwin-amd64-11_1-toothrot",
},
}
for _, tc := range testCases {
t.Run(tc.desc, func(t *testing.T) {
if got, gotErr := hostTypeToVersion(tc.hostType); got != nil || gotErr == nil {
t.Errorf("hostTypeToVersion(%q) = %+v, %s; want nil, error", tc.hostType, got, gotErr)
}
})
}
}
func TestHostOnMacStadium(t *testing.T) {
testCases := []struct {
desc string
hostType string
want bool
}{
{
desc: "empty string",
hostType: "",
want: false,
},
{
desc: "valid original darwin host",
hostType: "host-darwin-10_14",
want: true,
},
{
desc: "valid newer darwin host",
hostType: "host-darwin-amd64-11_0",
want: true,
},
{
desc: "invalid newer darwin host type",
hostType: "host-darwin-arm64-11_0-toothrot",
want: false,
},
}
for _, tc := range testCases {
t.Run(tc.desc, func(t *testing.T) {
if got := hostOnMacStadium(tc.hostType); got != tc.want {
t.Errorf("hostOnMacStadium(%q) = %t; want %t", tc.hostType, got, tc.want)
}
})
}
}
func TestVMNameReg(t *testing.T) {
testCases := []struct {
desc string
vmName string
want bool
}{
{
desc: "empty string",
vmName: "",
want: false,
},
{
desc: "invalid original darwin host",
vmName: "mac_10_11_host01b",
want: false,
},
{
desc: "valid newer darwin host",
vmName: "mac_11_12_amd64_host01b",
want: true,
},
{
desc: "valid newer darwin host",
vmName: "mac_11_20_amd64_host05a",
want: true,
},
{
desc: "invalid newer darwin host type",
vmName: "host-darwin-arm64-11_0-toothrot",
want: false,
},
{
desc: "invalid bastion host",
vmName: "dns_server",
want: false,
},
}
for _, tc := range testCases {
t.Run(tc.desc, func(t *testing.T) {
if got := vmNameReg.MatchString(tc.vmName); got != tc.want {
t.Errorf("vmNameReg.MatchString(%q) = %t; want %t", tc.vmName, got, tc.want)
}
})
}
}