From 23f0a9e1f5b6a99d1c685bb3939b754326cceaea Mon Sep 17 00:00:00 2001
From: "M. J. Fromberger" <fromberger@tailscale.com>
Date: Tue, 22 Oct 2024 08:40:24 -0700
Subject: [PATCH] tskagent: update tests, again

---
 tskagent_test.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tskagent_test.go b/tskagent_test.go
index b98372e..36f7ba2 100644
--- a/tskagent_test.go
+++ b/tskagent_test.go
@@ -140,6 +140,7 @@ func TestAgent(t *testing.T) {
 	})
 
 	t.Run("RemoveAll", func(t *testing.T) {
+		defer mustUpdate(t)
 		if err := ac.RemoveAll(); err != nil {
 			t.Errorf("RemoveAll: unexpected error: %v", err)
 		}
@@ -149,7 +150,6 @@ func TestAgent(t *testing.T) {
 		} else if len(lst) != 0 {
 			t.Errorf("List: got %+v, want empty", lst)
 		}
-		mustUpdate(t)
 	})
 
 	t.Run("RemoveMissing", func(t *testing.T) {
@@ -163,6 +163,7 @@ func TestAgent(t *testing.T) {
 	})
 
 	t.Run("RemovePresent", func(t *testing.T) {
+		defer mustUpdate(t)
 		if err := ac.Remove(pubKey); err != nil {
 			t.Errorf("Remove: unexpected error: %v", err)
 		}