76 lines
1.5 KiB
Go
76 lines
1.5 KiB
Go
package notify
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/appleboy/gorush/config"
|
|
"github.com/appleboy/gorush/logx"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestEmptyFeedbackURL(t *testing.T) {
|
|
cfg, _ := config.LoadConf()
|
|
logEntry := logx.LogPushEntry{
|
|
ID: "",
|
|
Type: "",
|
|
Platform: "",
|
|
Token: "",
|
|
Message: "",
|
|
Error: "",
|
|
}
|
|
|
|
err := DispatchFeedback(logEntry, cfg.Core.FeedbackURL, cfg.Core.FeedbackTimeout)
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestHTTPErrorInFeedbackCall(t *testing.T) {
|
|
cfg, _ := config.LoadConf()
|
|
cfg.Core.FeedbackURL = "http://test.example.com/api/"
|
|
logEntry := logx.LogPushEntry{
|
|
ID: "",
|
|
Type: "",
|
|
Platform: "",
|
|
Token: "",
|
|
Message: "",
|
|
Error: "",
|
|
}
|
|
|
|
err := DispatchFeedback(logEntry, cfg.Core.FeedbackURL, cfg.Core.FeedbackTimeout)
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestSuccessfulFeedbackCall(t *testing.T) {
|
|
// Mock http server
|
|
httpMock := httptest.NewServer(
|
|
http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path == "/dispatch" {
|
|
w.Header().Add("Content-Type", "application/json")
|
|
_, err := w.Write([]byte(`{}`))
|
|
if err != nil {
|
|
log.Println(err)
|
|
panic(err)
|
|
}
|
|
}
|
|
}),
|
|
)
|
|
defer httpMock.Close()
|
|
|
|
cfg, _ := config.LoadConf()
|
|
cfg.Core.FeedbackURL = httpMock.URL
|
|
logEntry := logx.LogPushEntry{
|
|
ID: "",
|
|
Type: "",
|
|
Platform: "",
|
|
Token: "",
|
|
Message: "",
|
|
Error: "",
|
|
}
|
|
|
|
err := DispatchFeedback(logEntry, cfg.Core.FeedbackURL, cfg.Core.FeedbackTimeout)
|
|
assert.Nil(t, err)
|
|
}
|