Move Linux client & common packages into a public repo.

wgengine/watchdog_test.go

@@ -0,0 +1,71 @@
+// 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 wgengine
+
+import (
+	"bytes"
+	"fmt"
+	"strings"
+	"testing"
+	"time"
+)
+
+func TestWatchdog(t *testing.T) {
+	t.Parallel()
+
+	t.Run("default watchdog does not fire", func(t *testing.T) {
+		t.Parallel()
+		tun := NewFakeTun()
+		e, err := NewUserspaceEngineAdvanced(t.Logf, tun, NewFakeRouter, 0, false)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		e = NewWatchdog(e)
+		e.(*watchdogEngine).maxWait = 150 * time.Millisecond
+
+		e.RequestStatus()
+		e.RequestStatus()
+		e.RequestStatus()
+		e.Close()
+	})
+
+	t.Run("watchdog fires on blocked getStatus", func(t *testing.T) {
+		t.Parallel()
+		tun := NewFakeTun()
+		e, err := NewUserspaceEngineAdvanced(t.Logf, tun, NewFakeRouter, 0, false)
+		if err != nil {
+			t.Fatal(err)
+		}
+		usEngine := e.(*userspaceEngine)
+		e = NewWatchdog(e)
+		wdEngine := e.(*watchdogEngine)
+		wdEngine.maxWait = 100 * time.Millisecond
+
+		logBuf := new(bytes.Buffer)
+		fatalCalled := make(chan struct{})
+		wdEngine.logf = func(format string, args ...interface{}) {
+			fmt.Fprintf(logBuf, format+"\n", args...)
+		}
+		wdEngine.fatalf = func(format string, args ...interface{}) {
+			t.Logf("FATAL: %s", fmt.Sprintf(format, args...))
+			fatalCalled <- struct{}{}
+		}
+
+		usEngine.wgLock.Lock() // blocks getStatus so the watchdog will fire
+
+		go e.RequestStatus()
+
+		select {
+		case <-fatalCalled:
+			if !strings.Contains(logBuf.String(), "goroutine profile: total ") {
+				t.Errorf("fatal called without watchdog stacks, got: %s", logBuf.String())
+			}
+			// expected
+		case <-time.After(3 * time.Second):
+			t.Fatalf("watchdog failed to fire")
+		}
+	})
+}