nezhahq-agent/cmd/agent/main_test.go
UUBulb ea6f119e72
feat: read config from env (#104)
* feat: read config from env

* ci: skip some tests
2024-11-30 11:40:48 +08:00

55 lines
1.1 KiB
Go

package main
import (
"fmt"
"net"
"os"
"reflect"
"testing"
)
func Test(t *testing.T) {
cases := []struct {
start, size int
want []int
}{
{0, 2, []int{0, 1}},
{1, 2, []int{1, 0}},
{0, 3, []int{0, 1, 2}},
{1, 3, []int{1, 2, 0}},
{2, 3, []int{2, 0, 1}},
}
for _, c := range cases {
if !reflect.DeepEqual(c.want, generateQueue(c.start, c.size)) {
t.Errorf("generateQueue(%d, %d) == %d, want %d", c.start, c.size, generateQueue(c.start, c.size), c.want)
}
}
}
func TestLookupIP(t *testing.T) {
if ci := os.Getenv("CI"); ci != "" { // skip if test on CI
return
}
ip, err := lookupIP("www.google.com")
fmt.Printf("ip: %v, err: %v\n", ip, err)
if err != nil {
t.Errorf("lookupIP failed: %v", err)
}
_, err = net.ResolveIPAddr("ip", "www.google.com")
if err != nil {
t.Errorf("ResolveIPAddr failed: %v", err)
}
ip, err = lookupIP("ipv6.google.com")
fmt.Printf("ip: %v, err: %v\n", ip, err)
if err != nil {
t.Errorf("lookupIP failed: %v", err)
}
_, err = net.ResolveIPAddr("ip", "ipv6.google.com")
if err != nil {
t.Errorf("ResolveIPAddr failed: %v", err)
}
}