path: root/internal/timer/operations_test.go

package timer

import (
	"fmt"
	"os"
	"path/filepath"
	"strconv"
	"testing"
	"time"
)

// setup sets up a temporary state file for testing.
func setup(t *testing.T) {
	t.Helper()
	tempDir := t.TempDir()
	StateFilePathOverride = filepath.Join(tempDir, ".timr_state")
	t.Cleanup(func() {
		StateFilePathOverride = ""
	})
}

func TestStartTimer(t *testing.T) {
	setup(t)

	// Start the timer
	msg, err := StartTimer()
	if err != nil {
		t.Fatalf("StartTimer() error = %v", err)
	}
	if msg != "Timer started." {
		t.Errorf("StartTimer() msg = %v, want %v", msg, "Timer started.")
	}

	// Check the state
	state, err := LoadState()
	if err != nil {
		t.Fatalf("LoadState() error = %v", err)
	}
	if !state.Running {
		t.Error("state.Running = false, want true")
	}

	// Try to start again
	msg, err = StartTimer()
	if err != nil {
		t.Fatalf("StartTimer() error = %v", err)
	}
	if msg != "Timer is already running." {
		t.Errorf("StartTimer() msg = %v, want %v", msg, "Timer is already running.")
	}
}

func TestStopTimer(t *testing.T) {
	setup(t)

	// Stop before starting
	msg, err := StopTimer()
	if err != nil {
		t.Fatalf("StopTimer() error = %v", err)
	}
	if msg != "Timer is not running." {
		t.Errorf("StopTimer() msg = %v, want %v", msg, "Timer is not running.")
	}

	// Start and then stop the timer
	_, _ = StartTimer()
	time.Sleep(10 * time.Millisecond) // Simulate work
	msg, err = StopTimer()
	if err != nil {
		t.Fatalf("StopTimer() error = %v", err)
	}
	if msg != "Timer stopped." {
		t.Errorf("StopTimer() msg = %v, want %v", msg, "Timer stopped.")
	}

	// Check the state
	state, err := LoadState()
	if err != nil {
		t.Fatalf("LoadState() error = %v", err)
	}
	if state.Running {
		t.Error("state.Running = true, want false")
	}
	if state.ElapsedTime == 0 {
		t.Error("state.ElapsedTime = 0, want > 0")
	}
}

func TestGetStatus(t *testing.T) {
	setup(t)

	// Status when stopped
	msg, err := GetStatus()
	if err != nil {
		t.Fatalf("GetStatus() error = %v", err)
	}
	want := "Status: Stopped\nElapsed Time: 0s"
	if msg != want {
		t.Errorf("GetStatus() msg = %q, want %q", msg, want)
	}

	// Status when running
	_, _ = StartTimer()
	msg, err = GetStatus()
	if err != nil {
		t.Fatalf("GetStatus() error = %v", err)
	}
	want = "Status: Running\nElapsed Time: 0s"
	if msg != want {
		t.Errorf("GetStatus() msg = %q, want %q", msg, want)
	}
}

func TestGetRawStatus(t *testing.T) {
	setup(t)

	// Raw status when stopped
	msg, err := GetRawStatus()
	if err != nil {
		t.Fatalf("GetRawStatus() error = %v", err)
	}
	want := "0.000000"
	if msg != want {
		t.Errorf("GetRawStatus() msg = %q, want %q", msg, want)
	}

	// Raw status when running
	state, err := LoadState()
	if err != nil {
		t.Fatalf("LoadState() error = %v", err)
	}
	state.Running = true
	state.StartTime = time.Now().Add(-2 * time.Second) // Set start time 2 seconds ago
	state.ElapsedTime = 0                               // Reset elapsed time for this specific test
	if err := state.Save(); err != nil {
		t.Fatalf("Save() error = %v", err)
	}

	msg, err = GetRawStatus()
	if err != nil {
		t.Fatalf("GetRawStatus() error = %v", err)
	}
	// we can't know the exact float, so we check if it's close to 2
	val, err := strconv.ParseFloat(msg, 64)
	if err != nil {
		t.Fatalf("could not parse float: %v", err)
	}
	if val < 1.9 || val > 2.1 {
		t.Errorf("GetRawStatus() msg = %q, want around 2.0", msg)
	}
}

func TestGetRawMinutesStatus(t *testing.T) {
	setup(t)

	// Raw minutes status when stopped
	msg, err := GetRawMinutesStatus()
	if err != nil {
		t.Fatalf("GetRawMinutesStatus() error = %v", err)
	}
	want := "0"
	if msg != want {
		t.Errorf("GetRawMinutesStatus() msg = %q, want %q", msg, want)
	}

	// Raw minutes status when running (simulating 2 minutes)
	state, err := LoadState()
	if err != nil {
		t.Fatalf("LoadState() error = %v", err)
	}
	state.Running = true
	state.StartTime = time.Now().Add(-2 * time.Minute) // Set start time 2 minutes ago
	state.ElapsedTime = 0                               // Reset elapsed time for this specific test
	if err := state.Save(); err != nil {
		t.Fatalf("Save() error = %v", err)
	}

	msg, err = GetRawMinutesStatus()
	if err != nil {
		t.Fatalf("GetRawMinutesStatus() error = %v", err)
	}
	want = "2"
	if msg != want {
		t.Errorf("GetRawMinutesStatus() msg = %q, want %q", msg, want)
	}
}

func TestResetTimer(t *testing.T) {
	setup(t)

	// Start timer to create a state file
	_, _ = StartTimer()

	// Reset the timer
	msg, err := ResetTimer()
	if err != nil {
		t.Fatalf("ResetTimer() error = %v", err)
	}
	if msg != "Timer reset." {
		t.Errorf("ResetTimer() msg = %v, want %v", msg, "Timer reset.")
	}

	// Check the state
	state, err := LoadState()
	if err != nil {
		t.Fatalf("LoadState() error = %v", err)
	}
	if state.Running {
		t.Error("state.Running = true, want false")
	}
	if state.ElapsedTime != 0 {
		t.Errorf("state.ElapsedTime = %v, want 0", state.ElapsedTime)
	}
}

func TestTrackTime(t *testing.T) {
	setup(t)

	// Helper to create a mock task command
	createMockTaskCommand := func(t *testing.T, shouldSucceed bool) {
		t.Helper()
		
		// Create a mock script that simulates the task command
		mockScript := `#!/bin/sh
if [ "$1" = "add" ] && [ "$2" = "+timrtest" ]; then
	if [ "%s" = "true" ]; then
		echo "Created task 1."
		exit 0
	else
		echo "Error: Failed to add task"
		exit 1
	fi
fi
echo "Invalid command"
exit 1
`
		scriptContent := fmt.Sprintf(mockScript, shouldSucceed)
		
		// Create temp directory for our mock
		tempDir := t.TempDir()
		mockPath := filepath.Join(tempDir, "task")
		
		// Write the mock script
		if err := os.WriteFile(mockPath, []byte(scriptContent), 0755); err != nil {
			t.Fatalf("Failed to create mock script: %v", err)
		}
		
		// Update PATH to use our mock
		oldPath := os.Getenv("PATH")
		os.Setenv("PATH", tempDir+":"+oldPath)
		t.Cleanup(func() {
			os.Setenv("PATH", oldPath)
		})
	}

	t.Run("TrackWithRunningTimer", func(t *testing.T) {
		setup(t)
		createMockTaskCommand(t, true)
		
		// Start timer and let it run for a bit
		state, _ := LoadState()
		state.Running = true
		state.StartTime = time.Now().Add(-5 * time.Minute)
		state.ElapsedTime = 0
		state.Save()
		
		// We'll modify TrackTime to use +timrtest for testing
		// For now, test with the actual implementation
		// In a real scenario, we'd want to make the tag configurable
		
		// Since we can't easily test the actual command execution,
		// we'll test the error case when task command is not found
		msg, err := TrackTime("test description")
		
		// We expect an error because 'task' command likely doesn't exist
		// or our mock won't match the exact command
		if err == nil {
			// If it somehow succeeded, check the message
			if msg == "" {
				t.Error("TrackTime() returned empty message on success")
			}
		}
		
		// Verify timer was stopped
		state, _ = LoadState()
		if state.Running {
			t.Error("Timer should be stopped after track attempt")
		}
	})

	t.Run("TrackWithStoppedTimer", func(t *testing.T) {
		setup(t)
		createMockTaskCommand(t, true)
		
		// Set up a stopped timer with some elapsed time
		state, _ := LoadState()
		state.Running = false
		state.ElapsedTime = 10 * time.Minute
		state.Save()
		
		// Try to track time
		_, err := TrackTime("another test")
		
		// We expect an error because task command likely doesn't exist
		// but we can verify the state handling
		if err == nil {
			// Verify timer was reset on success
			state, _ = LoadState()
			if state.ElapsedTime != 0 {
				t.Error("Timer should be reset after successful track")
			}
		}
	})

	t.Run("TrackWithZeroTime", func(t *testing.T) {
		setup(t)
		
		// Fresh timer with no elapsed time
		state, _ := LoadState()
		state.Running = false
		state.ElapsedTime = 0
		state.Save()
		
		// Try to track with zero time
		_, err := TrackTime("zero time test")
		
		// Even with zero time, the command should be attempted
		// We just verify no panic occurs
		_ = err
	})
}