Simplify unit-test

time.go

@@ -44,6 +44,7 @@ func StartTimeStampUpdater() {
 }
 
 // StopTimeStampUpdater stops the timestamp updater
+// WARNING: Make sure to call this function before the program exits, otherwise it will leak goroutines
 func StopTimeStampUpdater() {
 	if stopChan != nil {
 		close(stopChan)

time_test.go

@@ -32,21 +32,14 @@ func Test_StopTimeStampUpdater(t *testing.T) {
 	// Start the timestamp updater
 	StartTimeStampUpdater()
 
-	now := uint32(time.Now().Unix())
-	checkTimeStamp(t, now, Timestamp())
-
-	// Wait for an increment
-	time.Sleep(1 * time.Second)
-	checkTimeStamp(t, now+1, Timestamp())
-
 	// Stop the updater
 	StopTimeStampUpdater()
 
 	// Capture the timestamp after stopping
 	stoppedTime := Timestamp()
 
-	// Wait again to see if it updates
-	time.Sleep(3 * time.Second)
+	// Wait TO see if it updates
+	time.Sleep(5 * time.Second)
 
 	// It should not have changed since we've stopped the updater
 	require.Equal(t, stoppedTime, Timestamp(), "timestamp should not change after stopping updater")