|
@@ -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)
|
|
|
+ }
|
|
|
+}
|