fix test

wakatime · Jul 23, 2024 · af87dc7 · af87dc7
1 parent abba4cf
commit af87dc7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/cmd/heartbeat/heartbeat.go b/cmd/heartbeat/heartbeat.go
@@ -197,7 +197,7 @@ func RateLimited(offline paramscmd.Offline) bool {
 		return false
 	}
 
-	return time.Since(offline.LastSentAt) > offline.RateLimit
+	return time.Since(offline.LastSentAt) < offline.RateLimit
 }
 
 // ResetRateLimit updates the internal.heartbeats_last_sent_at timestamp

diff --git a/cmd/heartbeat/heartbeat_test.go b/cmd/heartbeat/heartbeat_test.go
@@ -1056,8 +1056,14 @@ func TestSendHeartbeats_ObfuscateProjectNotBranch(t *testing.T) {
 func TestRateLimited(t *testing.T) {
 	assert.False(t, cmdheartbeat.RateLimited(params.Offline{Disabled: true, RateLimit: 1, LastSentAt: time.Now()}))
 	assert.False(t, cmdheartbeat.RateLimited(params.Offline{RateLimit: 0, LastSentAt: time.Now()}))
-	assert.False(t, cmdheartbeat.RateLimited(params.Offline{RateLimit: offline.RateLimitDefaultSeconds, LastSentAt: time.Now().Add(time.Duration(-offline.RateLimitDefaultSeconds) * time.Second)}))
-	assert.True(t, cmdheartbeat.RateLimited(params.Offline{RateLimit: offline.RateLimitDefaultSeconds, LastSentAt: time.Now()}))
+
+	limit := time.Duration(offline.RateLimitDefaultSeconds) * time.Second
+
+	assert.False(t, cmdheartbeat.RateLimited(params.Offline{
+		RateLimit:  limit,
+		LastSentAt: time.Now().Add(time.Duration(-offline.RateLimitDefaultSeconds*2) * time.Second),
+	}))
+	assert.True(t, cmdheartbeat.RateLimited(params.Offline{RateLimit: limit, LastSentAt: time.Now()}))
 }
 
 func setupTestServer() (string, *http.ServeMux, func()) {