diff options
Diffstat (limited to 'media')
-rw-r--r-- | media/mediaType.go | 13 | ||||
-rw-r--r-- | media/mediaType_test.go | 44 |
2 files changed, 32 insertions, 25 deletions
diff --git a/media/mediaType.go b/media/mediaType.go index 07dbcc211..f6597fdb6 100644 --- a/media/mediaType.go +++ b/media/mediaType.go @@ -47,11 +47,14 @@ func (m Type) String() string { } var ( - CalendarType = Type{"text", "calendar", "ics"} - CSSType = Type{"text", "css", "css"} - HTMLType = Type{"text", "html", "html"} - JSONType = Type{"application", "json", "json"} - RSSType = Type{"application", "rss", "xml"} + CalendarType = Type{"text", "calendar", "ics"} + CSSType = Type{"text", "css", "css"} + HTMLType = Type{"text", "html", "html"} + JavascriptType = Type{"application", "javascript", "js"} + JSONType = Type{"application", "json", "json"} + RSSType = Type{"application", "rss", "xml"} + XMLType = Type{"application", "xml", "xml"} + TextType = Type{"text", "plain", "txt"} ) // TODO(bep) output mime.AddExtensionType diff --git a/media/mediaType_test.go b/media/mediaType_test.go index 40efc4f06..7636af59c 100644 --- a/media/mediaType_test.go +++ b/media/mediaType_test.go @@ -20,25 +20,29 @@ import ( ) func TestDefaultTypes(t *testing.T) { - require.Equal(t, "text", CalendarType.MainType) - require.Equal(t, "calendar", CalendarType.SubType) - require.Equal(t, "ics", CalendarType.Suffix) - - require.Equal(t, "text/calendar+ics", CalendarType.String()) - require.Equal(t, "text/calendar", CalendarType.Type()) - - require.Equal(t, "text", HTMLType.MainType) - require.Equal(t, "html", HTMLType.SubType) - require.Equal(t, "html", HTMLType.Suffix) - - require.Equal(t, "text/html", HTMLType.Type()) - require.Equal(t, "text/html+html", HTMLType.String()) - - require.Equal(t, "application", RSSType.MainType) - require.Equal(t, "rss", RSSType.SubType) - require.Equal(t, "xml", RSSType.Suffix) - - require.Equal(t, "application/rss", RSSType.Type()) - require.Equal(t, "application/rss+xml", RSSType.String()) + for _, test := range []struct { + tp Type + expectedMainType string + expectedSubType string + expectedSuffix string + expectedType string + expectedString string + }{ + {CalendarType, "text", "calendar", "ics", "text/calendar", "text/calendar+ics"}, + {CSSType, "text", "css", "css", "text/css", "text/css+css"}, + {HTMLType, "text", "html", "html", "text/html", "text/html+html"}, + {JavascriptType, "application", "javascript", "js", "application/javascript", "application/javascript+js"}, + {JSONType, "application", "json", "json", "application/json", "application/json+json"}, + {RSSType, "application", "rss", "xml", "application/rss", "application/rss+xml"}, + {TextType, "text", "plain", "txt", "text/plain", "text/plain+txt"}, + } { + require.Equal(t, test.expectedMainType, test.tp.MainType) + require.Equal(t, test.expectedSubType, test.tp.SubType) + require.Equal(t, test.expectedSuffix, test.tp.Suffix) + + require.Equal(t, test.expectedType, test.tp.Type()) + require.Equal(t, test.expectedString, test.tp.String()) + + } } |