123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132 |
- package bullet
- import (
- "fmt"
- "net"
- "net/http"
- "net/http/httptest"
- "testing"
- )
- var standardError string = `{
- "error": {
- "cat": "~(=^‥^)",
- "message": "The resource could not be found.",
- "type": "invalid_request"
- }
- }`
- var standardErrorMessage string = "invalid_request: The resource could not be found."
- func fakeServer(statusCode int, data string) *httptest.Server {
- function := func(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- w.WriteHeader(statusCode)
- fmt.Fprint(w, data)
- }
- 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}
- err := b.SendNote("test", "test")
- if err != nil {
- t.Error(err)
- }
- }
- func TestSendNoteFail(t *testing.T) {
- server := fakeServer(http.StatusBadRequest, standardError)
- server.Start()
- defer server.Close()
- b := Bullet{token: "", baseURL: server.URL}
- err := b.SendNote("test", "test")
- if err == nil {
- t.Error("There should be error")
- }
- if err.Error() != standardErrorMessage {
- t.Error(err)
- }
- }
- func TestSendLink(t *testing.T) {
- server := fakeServer(http.StatusOK, "")
- server.Start()
- defer server.Close()
- b := Bullet{token: "", baseURL: server.URL}
- err := b.SendLink("test", "test", "url")
- if err != nil {
- t.Error(err)
- }
- }
- func TestSendLinkFail(t *testing.T) {
- server := fakeServer(http.StatusBadRequest, standardError)
- server.Start()
- defer server.Close()
- b := Bullet{token: "", baseURL: server.URL}
- err := b.SendLink("test", "test", "url")
- if err == nil {
- t.Error("There should be error")
- }
- if err.Error() != standardErrorMessage {
- 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)
- }
- }
- func TestSendFileFail(t *testing.T) {
- server := fakeServer(http.StatusBadRequest, standardError)
- server.Start()
- defer server.Close()
- b := Bullet{token: "", baseURL: server.URL}
- err := b.SendFile("test", "test", "./README.md")
- if err == nil {
- t.Error("There should be error")
- }
- if err.Error() != standardErrorMessage {
- t.Error(err)
- }
- }
|