mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-05 16:05:36 +00:00
08acb502e5
If a test calls log.Printf, 'go test' horrifyingly rearranges the output to no longer be in chronological order, which makes debugging virtually impossible. Let's stop that from happening by making log.Printf panic if called from any module, no matter how deep, during tests. This required us to change the default error handler in at least one http.Server, as well as plumbing a bunch of logf functions around, especially in magicsock and wgengine, but also in logtail and backoff. To add insult to injury, 'go test' also rearranges the output when a parent test has multiple sub-tests (all the sub-test's t.Logf is always printed after all the parent tests t.Logf), so we need to screw around with a special Logf that can point at the "current" t (current_t.Logf) in some places. Probably our entire way of using subtests is wrong, since 'go test' would probably like to run them all in parallel if you called t.Parallel(), but it definitely can't because the're all manipulating the shared state created by the parent test. They should probably all be separate toplevel tests instead, with common setup/teardown logic. But that's a job for another time. Signed-off-by: Avery Pennarun <apenwarr@tailscale.com>
128 lines
2.4 KiB
Go
128 lines
2.4 KiB
Go
// Copyright (c) 2020 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 ipn
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"tailscale.com/tstest"
|
|
)
|
|
|
|
func testStoreSemantics(t *testing.T, store StateStore) {
|
|
t.Helper()
|
|
|
|
tests := []struct {
|
|
// if true, data is data to write. If false, data is expected
|
|
// output of read.
|
|
write bool
|
|
id StateKey
|
|
data string
|
|
// If write=false, true if we expect a not-exist error.
|
|
notExists bool
|
|
}{
|
|
{
|
|
id: "foo",
|
|
notExists: true,
|
|
},
|
|
{
|
|
write: true,
|
|
id: "foo",
|
|
data: "bar",
|
|
},
|
|
{
|
|
id: "foo",
|
|
data: "bar",
|
|
},
|
|
{
|
|
id: "baz",
|
|
notExists: true,
|
|
},
|
|
{
|
|
write: true,
|
|
id: "baz",
|
|
data: "quux",
|
|
},
|
|
{
|
|
id: "foo",
|
|
data: "bar",
|
|
},
|
|
{
|
|
id: "baz",
|
|
data: "quux",
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
if test.write {
|
|
if err := store.WriteState(test.id, []byte(test.data)); err != nil {
|
|
t.Errorf("writing %q to %q: %v", test.data, test.id, err)
|
|
}
|
|
} else {
|
|
bs, err := store.ReadState(test.id)
|
|
if err != nil {
|
|
if test.notExists && err == ErrStateNotExist {
|
|
continue
|
|
}
|
|
t.Errorf("reading %q: %v", test.id, err)
|
|
continue
|
|
}
|
|
if string(bs) != test.data {
|
|
t.Errorf("reading %q: got %q, want %q", test.id, string(bs), test.data)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMemoryStore(t *testing.T) {
|
|
tstest.PanicOnLog()
|
|
|
|
store := &MemoryStore{}
|
|
testStoreSemantics(t, store)
|
|
}
|
|
|
|
func TestFileStore(t *testing.T) {
|
|
tstest.PanicOnLog()
|
|
|
|
f, err := ioutil.TempFile("", "test_ipn_store")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
path := f.Name()
|
|
f.Close()
|
|
if err := os.Remove(path); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
store, err := NewFileStore(path)
|
|
if err != nil {
|
|
t.Fatalf("creating file store failed: %v", err)
|
|
}
|
|
|
|
testStoreSemantics(t, store)
|
|
|
|
// Build a brand new file store and check that both IDs written
|
|
// above are still there.
|
|
store, err = NewFileStore(path)
|
|
if err != nil {
|
|
t.Fatalf("creating second file store failed: %v", err)
|
|
}
|
|
|
|
expected := map[StateKey]string{
|
|
"foo": "bar",
|
|
"baz": "quux",
|
|
}
|
|
for id, want := range expected {
|
|
bs, err := store.ReadState(id)
|
|
if err != nil {
|
|
t.Errorf("reading %q (2nd store): %v", id, err)
|
|
}
|
|
if string(bs) != want {
|
|
t.Errorf("reading %q (2nd store): got %q, want %q", id, string(bs), want)
|
|
}
|
|
}
|
|
}
|