tailscale/net/dns/manager_linux_test.go

85 lines
1.9 KiB
Go
Raw Normal View History

// Copyright (c) 2021 Tailscale Inc & 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 dns
import (
"bytes"
"fmt"
"io/fs"
"os"
"testing"
)
func TestLinuxNewOSConfigurator(t *testing.T) {
tests := []struct {
name string
env newOSConfigEnv
wantLog string
want string // reflect type string
}{
{
name: "no_obvious_resolv.conf_owner",
env: newOSConfigEnv{
fs: memFS{
"/etc/resolv.conf": "nameserver 10.0.0.1\n",
},
resolvOwner: resolvOwner,
},
wantLog: "dns: [rc=unknown ret=dns.directManager]\n",
want: "dns.directManager",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
var logBuf bytes.Buffer
logf := func(format string, a ...interface{}) {
fmt.Fprintf(&logBuf, format, a...)
logBuf.WriteByte('\n')
}
osc, err := newOSConfigurator(logf, "unused_if_name0", tt.env)
if err != nil {
t.Fatal(err)
}
if got := fmt.Sprintf("%T", osc); got != tt.want {
t.Errorf("got %s; want %s", got, tt.want)
}
if tt.wantLog != string(logBuf.Bytes()) {
t.Errorf("log output mismatch:\n got: %q\nwant: %q\n", logBuf.Bytes(), tt.wantLog)
}
})
}
}
type memFS map[string]interface{} // full path => string for regular files
func (m memFS) Stat(name string) (isRegular bool, err error) {
v, ok := m[name]
if !ok {
return false, fs.ErrNotExist
}
if _, ok := v.(string); ok {
return true, nil
}
return false, nil
}
func (m memFS) Rename(oldName, newName string) error { panic("TODO") }
func (m memFS) Remove(name string) error { panic("TODO") }
func (m memFS) ReadFile(name string) ([]byte, error) {
v, ok := m[name]
if !ok {
return nil, fs.ErrNotExist
}
if s, ok := v.(string); ok {
return []byte(s), nil
}
panic("TODO")
}
func (fs memFS) WriteFile(name string, contents []byte, perm os.FileMode) error {
fs[name] = string(contents)
return nil
}