opencode/yao
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update file tests to handle content type variations

Browse source 
- Modified assertions in file upload, retrieval, and content tests to account for content type variations that may include charset.
- Enhanced test robustness by checking that the content type starts with the expected value, improving validation accuracy.
This commit is contained in:
Max 2025-11-06 11:57:38 +08:00
parent 2f6cf1a08c
commit 59c32bd1f6
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
openapi/tests/file/file_test.go
View file

@ -164,7 +164,9 @@ func TestFileUpload(t *testing.T) {
assert.Contains(t, response, "path") assert.Contains(t, response, "path")
assert.Contains(t, response, "user_path") assert.Contains(t, response, "user_path")
assert.Equal(t, testFileName, response["filename"]) assert.Equal(t, testFileName, response["filename"])
assert.Equal(t, testContentType, response["content_type"]) // Content type may include charset
contentType, _ := response["content_type"].(string)
assert.True(t, strings.HasPrefix(contentType, testContentType), "Content-Type should start with %s, got %s", testContentType, contentType)
assert.Equal(t, "uploaded", response["status"]) assert.Equal(t, "uploaded", response["status"])
// The file_id should be URL-safe (no slashes) and be an MD5 hash (32 chars) // The file_id should be URL-safe (no slashes) and be an MD5 hash (32 chars)
@ -519,7 +521,9 @@ func TestFileRetrieve(t *testing.T) {
assert.Contains(t, response, "content_type") assert.Contains(t, response, "content_type")
assert.Equal(t, testFileID, response["file_id"]) assert.Equal(t, testFileID, response["file_id"])
assert.Equal(t, testFileName, response["filename"]) assert.Equal(t, testFileName, response["filename"])
assert.Equal(t, testContentType, response["content_type"]) // Content type may include charset
contentType, _ := response["content_type"].(string)
assert.True(t, strings.HasPrefix(contentType, testContentType), "Content-Type should start with %s, got %s", testContentType, contentType)
t.Logf("Successfully retrieved file metadata: %s", testFileID) t.Logf("Successfully retrieved file metadata: %s", testFileID)
}) })
@ -627,7 +631,9 @@ func TestFileContent(t *testing.T) {
defer resp.Body.Close() defer resp.Body.Close()
assert.Equal(t, http.StatusOK, resp.StatusCode) assert.Equal(t, http.StatusOK, resp.StatusCode)
assert.Equal(t, testContentType, resp.Header.Get("Content-Type")) // Content type may include charset
assert.True(t, strings.HasPrefix(resp.Header.Get("Content-Type"), testContentType),
"Content-Type should start with %s, got %s", testContentType, resp.Header.Get("Content-Type"))
// Read and verify content // Read and verify content
content, err := io.ReadAll(resp.Body) content, err := io.ReadAll(resp.Body)