85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
|
package sms
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"net/http"
|
|||
|
|
"net/http/httptest"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ---------------- 模拟服务 ----------------
|
|||
|
|
|
|||
|
|
// mockServer 返回一个 httptest.Server，用于模拟API响应
|
|||
|
|
func mockServer(t *testing.T, path string, response interface{}) *httptest.Server {
|
|||
|
|
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|||
|
|
if r.URL.Path != path {
|
|||
|
|
t.Errorf("unexpected path: got %s, want %s", r.URL.Path, path)
|
|||
|
|
}
|
|||
|
|
_ = json.NewEncoder(w).Encode(response)
|
|||
|
|
}))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------------- 测试发送短信 ----------------
|
|||
|
|
|
|||
|
|
func TestSendSMS(t *testing.T) {
|
|||
|
|
client := NewClient()
|
|||
|
|
|
|||
|
|
content := "【明慧科技】提醒：您的go2ns租卡会员时长仅剩余一个月，现在续费最高立减200元！赶快进入小程序领取优惠吧！"
|
|||
|
|
returnUrl := "https://telecom.2016js.com/api/v1/sohan/notice"
|
|||
|
|
body := BusinessBody{
|
|||
|
|
Content: &content,
|
|||
|
|
SendList: []SendList{
|
|||
|
|
{
|
|||
|
|
OutOrderID: "ORDER777",
|
|||
|
|
PhoneNumber: "15019230751"},
|
|||
|
|
},
|
|||
|
|
ReturnURL: &returnUrl,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resp, err := client.SendSMS(body)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("SendSMS error: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Printf("response: %+v", resp)
|
|||
|
|
|
|||
|
|
if resp.StatusCode != 1 {
|
|||
|
|
t.Errorf("unexpected response: %+v", resp)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------------- 测试查询余额 ----------------
|
|||
|
|
|
|||
|
|
func TestGetBalance(t *testing.T) {
|
|||
|
|
client := NewClient()
|
|||
|
|
|
|||
|
|
resp, err := client.GetBalance()
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("GetBalance error: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Printf("response: %+v", resp)
|
|||
|
|
|
|||
|
|
if resp.StatusCode != 1 {
|
|||
|
|
t.Errorf("unexpected response: %+v", resp)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------------- 测试查询状态 ----------------
|
|||
|
|
|
|||
|
|
func TestQueryState(t *testing.T) {
|
|||
|
|
client := NewClient()
|
|||
|
|
|
|||
|
|
resp, err := client.QueryState("ORDER123", "15019230751")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("QueryState error: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Printf("response: %+v", resp)
|
|||
|
|
|
|||
|
|
if resp.StatusCode != 1 {
|
|||
|
|
t.Errorf("unexpected sendStatus: %s", resp.Data.SendStatus)
|
|||
|
|
}
|
|||
|
|
}
|