Fix session test

safing · Dec 22, 2023 · 540fe8e · 540fe8e
1 parent d7ab4d6
commit 540fe8e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/terminal/session_test.go b/terminal/session_test.go
@@ -47,10 +47,12 @@ func TestSuspicionLimit(t *testing.T) {
 
 	// Somewhere here we should rate limiting.
 	for i := 0; i < rateLimitMaxSuspicionPerSecond; i++ {
-		tErr = s.RateLimit()
 		s.ReportSuspiciousActivity(SusFactorCommon)
+		tErr = s.RateLimit()
+	}
+	if tErr == nil {
+		t.Error("should rate limit")
 	}
-	assert.ErrorIs(t, tErr, ErrRateLimited, "should rate limit")
 }
 
 func TestConcurrencyLimit(t *testing.T) {