feat(stats): add cost breakdown, forecast, and cache stats

Author: thomas-vilte

internal/ai/gemini/pull_requests_summarizer_service_test.go

@@ -173,7 +173,7 @@ func TestGeneratePRSummary_HappyPath(t *testing.T) {
 		summary, err := summarizer.GeneratePRSummary(ctx, "content with empty title")
 
 		assert.Error(t, err)
-		assert.Contains(t, err.Error(), "no PR title")
+		assert.Contains(t, err.Error(), "invalid AI output format")
 		assert.Empty(t, summary.Title)
 	})
 }

internal/ai/gemini/release_generator_test.go

@@ -324,6 +324,6 @@ func TestGenerateNotes(t *testing.T) {
 
 		// Assert
 		assert.Error(t, err)
-		assert.Contains(t, err.Error(), "empty response from AI")
+		assert.Contains(t, err.Error(), "invalid AI output format")
 	})
 }

internal/commands/stats/stats_test.go

@@ -189,7 +189,7 @@ func TestShowMonthlyStats_NoActivity(t *testing.T) {
 	os.Stdout = w
 
 	// Act
-	err := cmd.showMonthlyStats(manager, trans)
+	err := cmd.showMonthlyStats(manager, trans, false)
 
 	_ = w.Close()
 	os.Stdout = oldStdout
@@ -243,7 +243,7 @@ func TestShowMonthlyStats_WithActivity(t *testing.T) {
 	os.Stdout = w
 
 	// Act
-	err := cmd.showMonthlyStats(manager, trans)
+	err := cmd.showMonthlyStats(manager, trans, false)
 
 	_ = w.Close()
 	os.Stdout = oldStdout
@@ -290,7 +290,7 @@ func TestShowMonthlyStats_GroupsByDay(t *testing.T) {
 	os.Stdout = w
 
 	// Act
-	err := cmd.showMonthlyStats(manager, trans)
+	err := cmd.showMonthlyStats(manager, trans, false)
 
 	_ = w.Close()
 	os.Stdout = oldStdout
@@ -436,7 +436,7 @@ func TestStatsCommand_Integration(t *testing.T) {
 	r2, w2, _ := os.Pipe()
 	os.Stdout = w2
 
-	errMonthly := cmd.showMonthlyStats(manager, trans)
+	errMonthly := cmd.showMonthlyStats(manager, trans, false)
 
 	_ = w2.Close()
 	os.Stdout = oldStdout
@@ -479,7 +479,7 @@ func TestShowMonthlyStats_HandlesManagerErrors(t *testing.T) {
 	require.NoError(t, err)
 
 	// Act
-	err = cmd.showMonthlyStats(manager, trans)
+	err = cmd.showMonthlyStats(manager, trans, false)
 
 	// Assert
 	assert.Error(t, err, "should return an error when the history is corrupt")

internal/commands/suggests_commits/suggests_commits_test.go

@@ -43,6 +43,19 @@ func (m *MockGitService) ValidateGitConfig(ctx context.Context) error {
 	return args.Error(0)
 }
 
+func (m *MockGitService) GetChangedFiles(ctx context.Context) ([]string, error) {
+	args := m.Called(ctx)
+	if args.Get(0) == nil {
+		return nil, args.Error(1)
+	}
+	return args.Get(0).([]string), args.Error(1)
+}
+
+func (m *MockGitService) GetDiff(ctx context.Context) (string, error) {
+	args := m.Called(ctx)
+	return args.String(0), args.Error(1)
+}
+
 func setupTestEnv(t *testing.T) (*config.Config, *i18n.Translations, func()) {
 	tmpDir, err := os.MkdirTemp("", "matecommit-test-*")
 	if err != nil {

internal/errors/errors_test.go

@@ -41,15 +41,15 @@ func TestAppError_Error_Format(t *testing.T) {
 			err:  ErrNoChanges,
 			contains: []string{
 				"GIT",
-				"no staged changes detected",
+				"No staged changes detected",
 			},
 		},
 		{
 			name: "Error with underlying error",
 			err:  ErrGetBranch.WithError(errors.New("exit status 1")),
 			contains: []string{
 				"GIT",
-				"failed to get current branch",
+				"Failed to get current branch",
 				"exit status 1",
 			},
 		},
@@ -60,7 +60,7 @@ func TestAppError_Error_Format(t *testing.T) {
 				WithContext("stderr", "did not match any files"),
 			contains: []string{
 				"GIT",
-				"failed to add file to staging",
+				"Failed to add file to staging",
 				"exit status 128",
 				"did not match any files",
 			},
@@ -72,7 +72,7 @@ func TestAppError_Error_Format(t *testing.T) {
 				WithContext("stderr", "repository not found"),
 			contains: []string{
 				"GIT",
-				"failed to get diff",
+				"Failed to get diff",
 				"command failed",
 				"repository not found",
 			},

internal/git/git_service_test.go

@@ -741,7 +741,7 @@ func TestGitService_ErrorHandling(t *testing.T) {
 		var appErr *domainErrors.AppError
 		assert.True(t, errors.As(err, &appErr))
 		assert.Equal(t, domainErrors.TypeGit, appErr.Type)
-		assert.Contains(t, err.Error(), "failed to get commit count")
+		assert.Contains(t, err.Error(), "Failed to get commit count")
 	})
 }
 
@@ -970,7 +970,7 @@ func TestGitService_Push(t *testing.T) {
 		var appErr *domainErrors.AppError
 		assert.True(t, errors.As(err, &appErr))
 		assert.Equal(t, domainErrors.TypeGit, appErr.Type)
-		assert.Contains(t, err.Error(), "failed to push to remote")
+		assert.Contains(t, err.Error(), "Failed to push to remote")
 	})
 }
 

internal/github/github_service_test.go

@@ -275,7 +275,6 @@ func TestGitHubClient_UpdatePR_ErrorCases(t *testing.T) {
 
 		assert.Error(t, err)
 		assert.Contains(t, err.Error(), "insufficient permissions")
-		assert.Contains(t, err.Error(), "required scopes")
 		mockPR.AssertExpectations(t)
 	})
 }
@@ -342,9 +341,10 @@ func TestGitHubClient_GetPR_ErrorCases(t *testing.T) {
 		client := newTestClient(mockPR, mockIssues, mockRelease, mockUserService)
 
 		mockPR.On("Get", mock.Anything, "test-owner", "test-repo", 123).
-			Return(&github.PullRequest{}, &github.Response{}, assert.AnError)
+			Return(&github.PullRequest{}, &github.Response{Response: &http.Response{StatusCode: http.StatusInternalServerError}}, assert.AnError)
 
 		_, err := client.GetPR(context.Background(), 123)
+		assert.Error(t, err)
 		assert.Contains(t, err.Error(), "failed to get PR #123")
 	})
 
@@ -454,7 +454,7 @@ func TestGitHubClient_CreateRelease(t *testing.T) {
 
 		err := client.CreateRelease(context.Background(), release, notes, false, true)
 		assert.Error(t, err)
-		assert.Contains(t, err.Error(), "release already exists for version v1.0.0")
+		assert.Contains(t, err.Error(), "failed to create release")
 	})
 
 	t.Run("should return error if repo or tag not found", func(t *testing.T) {
@@ -473,7 +473,7 @@ func TestGitHubClient_CreateRelease(t *testing.T) {
 
 		err := client.CreateRelease(context.Background(), release, notes, false, true)
 		assert.Error(t, err)
-		assert.Contains(t, err.Error(), "repository or tag not found for version v1.0.0")
+		assert.Contains(t, err.Error(), "repository not found")
 	})
 
 	t.Run("should return generic error for other failures", func(t *testing.T) {
@@ -541,7 +541,7 @@ func TestGitHubClient_GetRelease(t *testing.T) {
 
 		assert.Error(t, err)
 		assert.Nil(t, release)
-		assert.Contains(t, err.Error(), "repository or tag not found for version v1.0.0")
+		assert.Contains(t, err.Error(), "repository not found")
 		mockRelease.AssertExpectations(t)
 	})
 
@@ -603,7 +603,7 @@ func TestGitHubClient_UpdateRelease(t *testing.T) {
 		err := client.UpdateRelease(context.Background(), "v1.0.0", "Updated body")
 
 		assert.Error(t, err)
-		assert.Contains(t, err.Error(), "repository or tag not found for version v1.0.0")
+		assert.Contains(t, err.Error(), "repository not found")
 		mockRelease.AssertExpectations(t)
 	})
 
@@ -1400,7 +1400,7 @@ func TestGitHubClient_CreateIssue(t *testing.T) {
 		client := newTestClient(mockPR, mockIssues, mockRelease, mockUserService)
 
 		mockIssues.On("Create", mock.Anything, "test-owner", "test-repo", mock.Anything).
-			Return((*github.Issue)(nil), &github.Response{}, assert.AnError)
+			Return((*github.Issue)(nil), &github.Response{Response: &http.Response{StatusCode: http.StatusInternalServerError}}, assert.AnError)
 
 		_, err := client.CreateIssue(context.Background(), "Title", "Body", nil, nil)