Added TestSendFile

bullet_test.go

@@ -2,6 +2,7 @@ package bullet
 
 import (
 	"fmt"
+	"net"
 	"net/http"
 	"net/http/httptest"
 	"testing"
@@ -14,11 +15,12 @@ func fakeServer(statusCode int, data string) *httptest.Server {
 		fmt.Fprint(w, data)
 	}
 
-	return httptest.NewServer(http.HandlerFunc(function))
+	return httptest.NewUnstartedServer(http.HandlerFunc(function))
 }
 
 func TestSendNote(t *testing.T) {
 	server := fakeServer(http.StatusOK, "")
+	server.Start()
 	defer server.Close()
 
 	b := Bullet{token: "", baseURL: server.URL}
@@ -31,6 +33,7 @@ func TestSendNote(t *testing.T) {
 
 func TestSendLink(t *testing.T) {
 	server := fakeServer(http.StatusOK, "")
+	server.Start()
 	defer server.Close()
 
 	b := Bullet{token: "", baseURL: server.URL}
@@ -40,3 +43,29 @@ func TestSendLink(t *testing.T) {
 		t.Error(err)
 	}
 }
+
+func TestSendFile(t *testing.T) {
+	l, err := net.Listen("tcp", "127.0.0.1:56022")
+	if err != nil {
+		t.Error(err)
+	}
+
+	response := `{
+  "file_name": "cat.jpg",
+  "file_type": "image/jpeg",
+  "file_url": "https://dl.pushbulletusercontent.com/034f197bc6c37cac3cc03542659d458b/cat.jpg",
+  "upload_url": "http://127.0.0.1:56022"}`
+
+	server := fakeServer(http.StatusOK, response)
+	server.Listener.Close()
+	server.Listener = l
+	server.Start()
+	defer server.Close()
+
+	b := Bullet{token: "", baseURL: server.URL}
+
+	err = b.SendFile("test", "test", "./README.md")
+	if err != nil {
+		t.Error(err)
+	}
+}