mirror of
https://github.com/rishikanthc/Scriberr.git
synced 2026-06-28 14:55:46 +00:00
7fc7619ee6
fix: new tests for chat and user management flows fix: resolve lint errors fix: configured lefthook to check entire project
72 lines
2.2 KiB
Go
72 lines
2.2 KiB
Go
package tests
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"scriberr/internal/api"
|
|
"scriberr/internal/models"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func (suite *APIHandlerTestSuite) TestSummarize() {
|
|
// Setup
|
|
job := suite.helper.CreateTestTranscriptionJob(suite.T(), "Summary Test Transcription")
|
|
job.Status = models.StatusCompleted
|
|
transcript := "This is a transcript about lots of things."
|
|
job.Transcript = &transcript
|
|
suite.helper.DB.Save(job)
|
|
|
|
req := api.SummarizeRequest{
|
|
Model: "gpt-3.5-turbo",
|
|
Content: "This is the content to summarize",
|
|
TranscriptionID: job.ID,
|
|
}
|
|
|
|
resp := suite.makeAuthenticatedRequest("POST", "/api/v1/summarize/", req, true)
|
|
assert.Equal(suite.T(), http.StatusOK, resp.Code)
|
|
|
|
// Check headers for streaming
|
|
assert.Equal(suite.T(), "text/plain; charset=utf-8", resp.Header().Get("Content-Type"))
|
|
assert.Equal(suite.T(), "chunked", resp.Header().Get("Transfer-Encoding"))
|
|
|
|
// Check response body
|
|
body := resp.Body.String()
|
|
// Mock server returns "This is a test response..." in chunks
|
|
assert.Contains(suite.T(), body, "This")
|
|
|
|
// Verify summary saved
|
|
var summary models.Summary
|
|
err := suite.helper.DB.Where("transcription_id = ?", job.ID).First(&summary).Error
|
|
assert.NoError(suite.T(), err)
|
|
assert.NotEmpty(suite.T(), summary.Content)
|
|
}
|
|
|
|
func (suite *APIHandlerTestSuite) TestGetSummaryForTranscription() {
|
|
// Setup
|
|
job := suite.helper.CreateTestTranscriptionJob(suite.T(), "Summary Retrieval Test")
|
|
|
|
// Case 1: No summary
|
|
resp := suite.makeAuthenticatedRequest("GET", "/api/v1/transcription/"+job.ID+"/summary", nil, true)
|
|
assert.Equal(suite.T(), http.StatusOK, resp.Code)
|
|
var summaryResp models.Summary
|
|
json.Unmarshal(resp.Body.Bytes(), &summaryResp)
|
|
assert.Empty(suite.T(), summaryResp.Content)
|
|
|
|
// Case 2: Saved summary
|
|
summary := models.Summary{
|
|
TranscriptionID: job.ID,
|
|
Model: "gpt-4",
|
|
Content: "Stored summary content",
|
|
}
|
|
suite.helper.DB.Create(&summary)
|
|
|
|
resp = suite.makeAuthenticatedRequest("GET", "/api/v1/transcription/"+job.ID+"/summary", nil, true)
|
|
assert.Equal(suite.T(), http.StatusOK, resp.Code)
|
|
|
|
json.Unmarshal(resp.Body.Bytes(), &summaryResp)
|
|
assert.Equal(suite.T(), "Stored summary content", summaryResp.Content)
|
|
assert.Equal(suite.T(), "gpt-4", summaryResp.Model)
|
|
}
|