Merge pull request #8 from alphagov/add-csv-preview-test-case

Add test case for csv preview url
alphagov · Oct 19, 2023 · 2a504ac · 2a504ac
2 parents c4fb794 + 2835dc3
commit 2a504ac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/file/file_test.go b/internal/file/file_test.go
@@ -60,6 +60,8 @@ func TestGenerateFilePathTableDriven(t *testing.T) {
 		{"https://example.com/foo/", "text/html", "example.com/foo/index.html", nil},
 		{"https://example.com/foo/bar", "text/html", "example.com/foo/bar.html", nil},
 		{"https://example.com/foo", "text/csv", "example.com/foo.csv", nil},
+		{"https://example.com/foo.csv", "text/csv", "example.com/foo.csv", nil},
+		{"https://example.com/foo.csv/preview", "text/html", "example.com/foo.csv/preview.html", nil},
 		{"https://example.com/style.css", "text/css", "example.com/style.css", nil},
 		{"https://example.com/foo", "text/blah", "", errors.New("Error determining content type")},
 		{"https://example.com/foo", "", "", errors.New("Error determining content type")},