Function TestCalculateTimeRemaining

pkg/cli/commands_test.go:911–969

TestCalculateTimeRemaining tests the calculateTimeRemaining function

(t *testing.T)

// TestCalculateTimeRemaining tests the calculateTimeRemaining function
func TestCalculateTimeRemaining(t *testing.T) {
	tests := []struct {
		name        string
		stopTimeStr string
		expected    string
	}{
		{
			name:        "empty stop time",
			stopTimeStr: "",
			expected:    "N/A",
		},
		{
			name:        "invalid format",
			stopTimeStr: "invalid-date-format",
			expected:    "Invalid",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := calculateTimeRemaining(tt.stopTimeStr)
			if result != tt.expected {
				t.Errorf("calculateTimeRemaining(%q) = %q, want %q", tt.stopTimeStr, result, tt.expected)
			}
		})
	}

	// Test with future time - this will test the logic but the exact result depends on current time
	t.Run("future time formatting", func(t *testing.T) {
		// Create a time 2 hours and 30 minutes in the future
		// Add a small buffer to account for execution time
		futureTime := time.Now().Add(2*time.Hour + 30*time.Minute + 1*time.Second)
		stopTimeStr := futureTime.Format("2006-01-02 15:04:05")

		result := calculateTimeRemaining(stopTimeStr)

		// Should contain "h" and "m" for hours and minutes
		if !strings.Contains(result, "h") || !strings.Contains(result, "m") {
			t.Errorf("calculateTimeRemaining() for future time should contain hours and minutes, got: %q", result)
		}

		// Should not be "Expired", "Invalid", or "N/A"
		if result == "Expired" || result == "Invalid" || result == "N/A" {
			t.Errorf("calculateTimeRemaining() for future time should not be %q", result)
		}
	})

	// Test with past time
	t.Run("past time - expired", func(t *testing.T) {
		// Create a time 1 hour in the past
		pastTime := time.Now().Add(-1 * time.Hour)
		stopTimeStr := pastTime.Format("2006-01-02 15:04:05")

		result := calculateTimeRemaining(stopTimeStr)
		if result != "Expired" {
			t.Errorf("calculateTimeRemaining() for past time = %q, want %q", result, "Expired")
		}
	})