tcjlj/nezha
1
0
Fork 0
mirror of https://github.com/nezhahq/nezha.git synced 2025-01-22 12:48:14 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

♻️ set content type

Browse Source
This commit is contained in:
naiba 2021-11-05 12:16:41 +08:00
parent 6636d77cfd
commit 17a53f8829
2 changed files with 8 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
model/notification.go
View File

@ -75,14 +75,15 @@ func (n *Notification) reqBody(message string) (string, error) {
return "", errors.New("不支持的请求类型")
}
func (n *Notification) reqContentType() string {
func (n *Notification) setContentType(req *http.Request) {
if n.RequestMethod == NotificationRequestMethodGET {
return ""
return
}
if n.RequestType == NotificationRequestTypeForm {
return "application/x-www-form-urlencoded"
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
} else {
req.Header.Set("Content-Type", "application/json")
}
return "application/json"
}
func (n *Notification) setRequestHeader(req *http.Request) error {
@ -122,6 +123,8 @@ func (n *Notification) Send(message string) error {
return err
}
n.setContentType(req)
if err := n.setRequestHeader(req); err != nil {
return err
}

12
model/notification_test.go
View File

@ -7,9 +7,7 @@ import (
)
var (
msg = "msg"
reqTypeForm = "application/x-www-form-urlencoded"
reqTypeJSON = "application/json"
msg = "msg"
)
type testSt struct {
@ -19,7 +17,6 @@ type testSt struct {
reqMethod int
expectURL string
expectBody string
expectType string
}
func execCase(t *testing.T, item testSt) {
@ -33,7 +30,6 @@ func execCase(t *testing.T, item testSt) {
reqBody, err := n.reqBody(msg)
assert.Nil(t, err)
assert.Equal(t, item.expectBody, reqBody)
assert.Equal(t, item.expectType, n.reqContentType())
}
func TestNotification(t *testing.T) {
@ -44,7 +40,6 @@ func TestNotification(t *testing.T) {
reqMethod: NotificationRequestMethodGET,
expectURL: "https://example.com",
expectBody: "",
expectType: "",
},
{
url: "https://example.com/?m=#NEZHA#",
@ -52,7 +47,6 @@ func TestNotification(t *testing.T) {
reqMethod: NotificationRequestMethodGET,
expectURL: "https://example.com/?m=" + msg,
expectBody: "",
expectType: "",
},
{
url: "https://example.com/?m=#NEZHA#",
@ -61,7 +55,6 @@ func TestNotification(t *testing.T) {
reqType: NotificationRequestTypeForm,
expectURL: "https://example.com/?m=" + msg,
expectBody: "asd=" + msg,
expectType: reqTypeForm,
},
{
url: "https://example.com/?m=#NEZHA#",
@ -70,7 +63,6 @@ func TestNotification(t *testing.T) {
reqType: NotificationRequestTypeForm,
expectURL: "https://example.com/?m=" + msg,
expectBody: "%23NEZHA%23=" + msg,
expectType: reqTypeForm,
},
{
url: "https://example.com/?m=#NEZHA#",
@ -79,7 +71,6 @@ func TestNotification(t *testing.T) {
reqType: NotificationRequestTypeJSON,
expectURL: "https://example.com/?m=" + msg,
expectBody: `{"asd":"msg"}`,
expectType: reqTypeJSON,
},
{
url: "https://example.com/?m=#NEZHA#",
@ -88,7 +79,6 @@ func TestNotification(t *testing.T) {
reqType: NotificationRequestTypeJSON,
expectURL: "https://example.com/?m=" + msg,
expectBody: `{"msg":"msg"}`,
expectType: reqTypeJSON,
},
}