fix tests

wakatime · Jul 19, 2024 · ca990df · ca990df
1 parent df215b4
commit ca990df
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/main_test.go b/main_test.go
@@ -327,7 +327,7 @@ func TestSendHeartbeats_ExtraHeartbeats(t *testing.T) {
 	offlineCount, err := offline.CountHeartbeats(offlineQueueFile.Name())
 	require.NoError(t, err)
 
-	assert.Zero(t, offlineCount)
+	assert.Equal(t, 1, offlineCount)
 
 	assert.Eventually(t, func() bool { return numCalls == 2 }, time.Second, 50*time.Millisecond)
 }
@@ -357,6 +357,8 @@ func TestSendHeartbeats_ExtraHeartbeats_SyncLegacyOfflineActivity(t *testing.T)
 			filename = "testdata/api_heartbeats_response_extra_heartbeats_legacy_offline.json"
 		case 3:
 			filename = "testdata/api_heartbeats_response_extra_heartbeats_extra.json"
+		case 4:
+			filename = "testdata/api_heartbeats_response_extra_heartbeats_extra.json"
 		}
 
 		// write response
@@ -455,7 +457,7 @@ func TestSendHeartbeats_ExtraHeartbeats_SyncLegacyOfflineActivity(t *testing.T)
 
 	assert.Zero(t, offlineCount)
 
-	assert.Eventually(t, func() bool { return numCalls == 3 }, time.Second, 50*time.Millisecond)
+	assert.Eventually(t, func() bool { return numCalls == 4 }, time.Second, 50*time.Millisecond)
 }
 
 func TestSendHeartbeats_OfflineOnly(t *testing.T) {
@@ -497,7 +499,7 @@ func TestSendHeartbeats_OfflineOnly(t *testing.T) {
 		"--internal-config", tmpInternalConfigFile.Name(),
 		"--entity", "testdata/main.go",
 		"--extra-heartbeats", "true",
-		"--cursorpos", "12",
+		"--cursorpos", "100",
 		"--offline-queue-file", offlineQueueFile.Name(),
 		"--offline-only",
 		"--lineno", "42",