106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
package drive
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"google.golang.org/api/drive/v3"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const exportFormats = `{
|
|
"application/vnd.google-apps.document": [
|
|
"application/rtf",
|
|
"application/vnd.oasis.opendocument.text",
|
|
"text/html",
|
|
"application/pdf",
|
|
"application/epub+zip",
|
|
"application/zip",
|
|
"application/vnd.openxmlformats-officedocument.wordprocessingml.document",
|
|
"text/plain"
|
|
],
|
|
"application/vnd.google-apps.spreadsheet": [
|
|
"application/x-vnd.oasis.opendocument.spreadsheet",
|
|
"text/tab-separated-values",
|
|
"application/pdf",
|
|
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
|
"text/csv",
|
|
"application/zip",
|
|
"application/vnd.oasis.opendocument.spreadsheet"
|
|
],
|
|
"application/vnd.google-apps.jam": [
|
|
"application/pdf"
|
|
],
|
|
"application/vnd.google-apps.script": [
|
|
"application/vnd.google-apps.script+json"
|
|
],
|
|
"application/vnd.google-apps.presentation": [
|
|
"application/vnd.oasis.opendocument.presentation",
|
|
"application/pdf",
|
|
"application/vnd.openxmlformats-officedocument.presentationml.presentation",
|
|
"text/plain"
|
|
],
|
|
"application/vnd.google-apps.form": [
|
|
"application/zip"
|
|
],
|
|
"application/vnd.google-apps.drawing": [
|
|
"image/svg+xml",
|
|
"image/png",
|
|
"application/pdf",
|
|
"image/jpeg"
|
|
]
|
|
}`
|
|
|
|
func TestInternalParseExtensions(t *testing.T) {
|
|
for _, test := range []struct {
|
|
in string
|
|
want []string
|
|
wantErr error
|
|
}{
|
|
{"doc", []string{"doc"}, nil},
|
|
{" docx ,XLSX, pptx,svg", []string{"docx", "xlsx", "pptx", "svg"}, nil},
|
|
{"docx,svg,Docx", []string{"docx", "svg"}, nil},
|
|
{"docx,potato,docx", []string{"docx"}, errors.New(`couldn't find mime type for extension "potato"`)},
|
|
} {
|
|
f := new(Fs)
|
|
gotErr := f.parseExtensions(test.in)
|
|
if test.wantErr == nil {
|
|
assert.NoError(t, gotErr)
|
|
} else {
|
|
assert.EqualError(t, gotErr, test.wantErr.Error())
|
|
}
|
|
assert.Equal(t, test.want, f.extensions)
|
|
}
|
|
|
|
// Test it is appending
|
|
f := new(Fs)
|
|
assert.Nil(t, f.parseExtensions("docx,svg"))
|
|
assert.Nil(t, f.parseExtensions("docx,svg,xlsx"))
|
|
assert.Equal(t, []string{"docx", "svg", "xlsx"}, f.extensions)
|
|
|
|
}
|
|
|
|
func TestInternalFindExportFormat(t *testing.T) {
|
|
item := new(drive.File)
|
|
item.MimeType = "application/vnd.google-apps.document"
|
|
for _, test := range []struct {
|
|
extensions []string
|
|
wantExtension string
|
|
wantMimeType string
|
|
}{
|
|
{[]string{}, "", ""},
|
|
{[]string{"pdf"}, "pdf", "application/pdf"},
|
|
{[]string{"pdf", "rtf", "xls"}, "pdf", "application/pdf"},
|
|
{[]string{"xls", "rtf", "pdf"}, "rtf", "application/rtf"},
|
|
{[]string{"xls", "csv", "svg"}, "", ""},
|
|
} {
|
|
f := new(Fs)
|
|
f.extensions = test.extensions
|
|
assert.NoError(t, json.Unmarshal([]byte(exportFormats), &f.exportFormats))
|
|
gotExtension, gotMimeType := f.findExportFormat("file", f.exportFormats[item.MimeType])
|
|
assert.Equal(t, test.wantExtension, gotExtension)
|
|
assert.Equal(t, test.wantMimeType, gotMimeType)
|
|
}
|
|
}
|