-
Notifications
You must be signed in to change notification settings - Fork 0
/
api_test.go
69 lines (57 loc) · 1.83 KB
/
api_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package q1_test
import (
"bytes"
"encoding/json"
"net/http"
"net/http/httptest"
"net/url"
"q1"
"strings"
"testing"
"github.com/gin-gonic/gin"
)
func TestClaimV4(t *testing.T) {
expected := "0x123DFPER321XCAAZ"
gin.SetMode(gin.TestMode)
// From Weerasak Chongnguluam
formFields := url.Values{}
formFields.Add("address_id", expected)
reader := strings.NewReader(formFields.Encode())
// reader := strings.NewReader(fmt.Sprintf("address_id=%s", expected))
res := httptest.NewRecorder()
c, r := gin.CreateTestContext(res)
c.Request = httptest.NewRequest(http.MethodPost, "/claim", reader)
c.Request.Header.Set("Content-Type", "application/x-www-form-urlencoded")
r.POST("/claim", q1.ClaimRewards)
r.ServeHTTP(res, c.Request)
if status := res.Code; status != http.StatusOK {
t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusOK)
}
var response q1.Response
json.Unmarshal(res.Body.Bytes(), &response)
if response.Message != expected {
t.Errorf("Message: got %v want %v", response.Message, expected)
}
}
func TestClaimV4_with_JSON(t *testing.T) {
expected := "0x123DFPER321XCAAZ"
gin.SetMode(gin.TestMode)
request := q1.Request{
Address_id: expected,
}
jsonStr, _ := json.Marshal(request)
res := httptest.NewRecorder()
c, r := gin.CreateTestContext(res)
c.Request = httptest.NewRequest(http.MethodPost, "/claim/json", bytes.NewBuffer(jsonStr))
c.Request.Header.Set("Content-Type", "application/json")
r.POST("/claim/json", q1.ClaimRewardsJson)
r.ServeHTTP(res, c.Request)
if status := res.Code; status != http.StatusOK {
t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusOK)
}
var response q1.Response
json.Unmarshal(res.Body.Bytes(), &response)
if response.Message != expected {
t.Errorf("Message: got %v want %v", response.Message, expected)
}
}