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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/*
Copyright (c) 2019 Ben Morrison (gbmor)
This file is part of Getwtxt.
Getwtxt is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Getwtxt is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with Getwtxt. If not, see <https://www.gnu.org/licenses/>.
*/
package svc // import "git.sr.ht/~gbmor/getwtxt/svc"
import (
"fmt"
"net/http"
"net/http/httptest"
"net/url"
"strings"
"testing"
"git.sr.ht/~gbmor/getwtxt/registry"
)
var apiPostUserCases = []struct {
name string
nick string
uri string
wantErr bool
}{
{
name: "Known Good User",
nick: "getwtxttest",
uri: testTwtxtURL,
wantErr: false,
},
{
name: "Missing URI",
nick: "missinguri",
uri: "",
wantErr: true,
},
{
name: "Missing Nickname",
nick: "",
uri: "https://example.com/twtxt.txt",
wantErr: true,
},
{
name: "Missing URI and Nickname",
nick: "",
uri: "",
wantErr: true,
},
}
func Test_apiPostUser(t *testing.T) {
initTestConf()
portnum := fmt.Sprintf(":%v", confObj.Port)
twtxtCache = registry.New(nil)
for _, tt := range apiPostUserCases {
t.Run(tt.name, func(t *testing.T) {
params := url.Values{}
params.Set("url", tt.uri)
params.Set("nickname", tt.nick)
req, err := http.NewRequest("POST", "https://localhost"+portnum+"/api/plain/users", strings.NewReader(params.Encode()))
if err != nil {
t.Errorf("%v\n", err)
}
req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
rr := httptest.NewRecorder()
apiEndpointPOSTHandler(rr, req)
if !tt.wantErr {
if rr.Code != http.StatusOK {
t.Errorf("Received unexpected non-200 response: %v\n", rr.Code)
}
} else {
if rr.Code != http.StatusBadRequest {
t.Errorf("Expected 400 Bad Request, but received: %v\n", rr.Code)
}
}
})
}
}
func Benchmark_apiPostUser(b *testing.B) {
initTestConf()
portnum := fmt.Sprintf(":%v", confObj.Port)
twtxtCache = registry.New(nil)
params := url.Values{}
params.Set("url", testTwtxtURL)
params.Set("nickname", "gbmor")
req, _ := http.NewRequest("POST", "https://localhost"+portnum+"/api/plain/users", strings.NewReader(params.Encode()))
req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
rr := httptest.NewRecorder()
for i := 0; i < b.N; i++ {
apiEndpointPOSTHandler(rr, req)
b.StopTimer()
twtxtCache = registry.New(nil)
b.StartTimer()
}
}
|