handler_test: make test more reliable

Linux gives "text/x-sh", and not "application/x-sh".
tsuru · Sep 9, 2014 · 3ac04f8 · 3ac04f8
1 parent 5b5b689
commit 3ac04f8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/api/handler_test.go b/api/handler_test.go
@@ -10,6 +10,7 @@ import (
 	"fmt"
 	"io"
 	"io/ioutil"
+	"mime"
 	"net/http"
 	"net/http/httptest"
 	"os"
@@ -1621,7 +1622,8 @@ func (s *S) TestGetMimeTypeFromExtension(c *gocheck.C) {
 	path = "my-text-file.sh"
 	content = new(bytes.Buffer)
 	content.WriteString("")
-	c.Assert(getMimeType(path, content.Bytes()), gocheck.Equals, "application/x-sh")
+	expected := mime.TypeByExtension(".sh")
+	c.Assert(getMimeType(path, content.Bytes()), gocheck.Equals, expected)
 }
 
 func (s *S) TestGetMimeTypeFromContent(c *gocheck.C) {