tskagent: update tests, again

tailscale · Oct 22, 2024 · 23f0a9e · 23f0a9e
1 parent 6f23dfc
commit 23f0a9e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)
 		}