From f5e0407e9714d09e1f23d4916cec6f200563d566 Mon Sep 17 00:00:00 2001
From: David Crawshaw <crawshaw@tailscale.com>
Date: Sun, 8 Mar 2020 08:42:49 -0400
Subject: [PATCH] ipn: exercise logout in e2e test

Signed-off-by: David Crawshaw <crawshaw@tailscale.com>
---
 ipn/e2e_test.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/ipn/e2e_test.go b/ipn/e2e_test.go
index 89d2e9fa1..987e84c36 100644
--- a/ipn/e2e_test.go
+++ b/ipn/e2e_test.go
@@ -136,6 +136,32 @@ func TestIPN(t *testing.T) {
 			t.Error("no ping seen")
 		}
 	})
+
+drain:
+	for {
+		select {
+		case <-n1.NotifyCh:
+		case <-n2.NotifyCh:
+		default:
+			break drain
+		}
+	}
+
+	n1.Backend.Logout()
+
+	t.Run("logout", func(t *testing.T) {
+		select {
+		case n := <-n1.NotifyCh:
+			if n.State != nil {
+				if *n.State != NeedsLogin {
+					t.Errorf("n.State=%v, want %v", n.State, NeedsLogin)
+					return
+				}
+			}
+		case <-time.After(3 * time.Second):
+			t.Fatalf("timeout waiting for logout notification")
+		}
+	})
 }
 
 type testNode struct {