api_test.go

package mitake

import (
	"fmt"
	"net/http"
	"reflect"
	"testing"
)

func TestClient_SendBatch(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/SmSendPost.asp", func(w http.ResponseWriter, r *http.Request) {
		testMethod(t, r, "POST")
		testINI(t, r, `[0]
dstaddr=0987654321
smbody=Test 1
[1]
dstaddr=0987654322
smbody=Test 2`)
		fmt.Fprint(w, `[0]
msgid=1010079522
statuscode=1
[1]
msgid=1010079523
statuscode=4
AccountPoint=98`)
	})

	messages := []Message{
		{
			Dstaddr: "0987654321",
			Smbody:  "Test 1",
		},
		{
			Dstaddr: "0987654322",
			Smbody:  "Test 2",
		},
	}

	resp, err := client.SendBatch(messages)
	if err != nil {
		t.Errorf("SendBatch returned unexpected error: %v", err)
	}

	want := []*MessageResult{
		{
			Msgid:      "1010079522",
			Statuscode: StatusCode("1"),
		},
		{
			Msgid:      "1010079523",
			Statuscode: StatusCode("4"),
		},
	}
	if !reflect.DeepEqual(resp.Results, want) {
		t.Errorf("SendBatch returned %+v, want %+v", resp.Results, want)
	}
}

func TestClient_QueryAccountPoint(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/SmQueryGet.asp", func(w http.ResponseWriter, r *http.Request) {
		testMethod(t, r, "GET")
		fmt.Fprint(w, `AccountPoint=100`)
	})

	ap, err := client.QueryAccountPoint()
	if err != nil {
		t.Errorf("QueryAccountPoint returned unexpected error: %v", err)
	}
	if ap != 100 {
		t.Errorf("QueryAccountPoint returned %+v, want %+v", ap, 100)
	}
}

func TestClient_QueryMessageStatus(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/SmQueryGet.asp", func(w http.ResponseWriter, r *http.Request) {
		testMethod(t, r, "GET")
		fmt.Fprint(w, `1010079522	1	20170101010010
1010079523	4	20170101010011`)
	})

	resp, err := client.QueryMessageStatus([]string{"1010079522", "1010079523"})
	if err != nil {
		t.Errorf("QueryMessageStatus returned unexpected error: %v", err)
	}

	want := []*MessageStatus{
		{
			MessageResult: MessageResult{
				Msgid:      "1010079522",
				Statuscode: StatusCode("1"),
			},
			StatusTime: "20170101010010",
		},
		{
			MessageResult: MessageResult{
				Msgid:      "1010079523",
				Statuscode: StatusCode("4"),
			},
			StatusTime: "20170101010011",
		},
	}
	if !reflect.DeepEqual(resp.Statuses, want) {
		t.Errorf("QueryMessageStatus returned %+v, want %+v", resp.Statuses, want)
	}
}