💾 Archived View for source.community › ckaznocha › gemini › blob › main › response_test.go captured on 2024-05-10 at 11:03:31. Gemini links have been rewritten to link to archived content
⬅️ Previous capture (2024-02-05)
-=-=-=-=-=-=-
.
,-. ,-. . . ,-. ,-. ,-. ,-. ,-. ,-,-. ,-,-. . . ,-. . |- . .
`-. | | | | | | |-' | | | | | | | | | | | | | | | | |
`-' `-' `-^ ' `-' `-' :: `-' `-' ' ' ' ' ' ' `-^ ' ' ' `' `-|
/|
`-'
git clone https://source.community/ckaznocha/gemini.git
View raw contents of /response_test.go (main)
────╮
1│ package gemini_test
2│
3│ import (
4│ "context"
5│ "crypto/tls"
6│ "fmt"
7│ "io"
8│ "mime"
9│ "net"
10│ "strings"
11│ "testing"
12│ "time"
13│
14│ "source.community/ckaznocha/gemini"
15│ "source.community/ckaznocha/gemini/geminitest"
16│ )
17│
18│ func Test_response_Input(t *testing.T) {
19│ t.Parallel()
20│
21│ type args struct {
22│ prompt string
23│ isSensitive bool
24│ }
25│
26│ tests := []struct {
27│ name string
28│ want string
29│ args args
30│ }{
31│ {
32│ name: "",
33│ args: args{
34│ prompt: "Hello",
35│ },
36│ want: "10 Hello\r\n",
37│ },
38│ {
39│ name: "",
40│ args: args{
41│ prompt: "Hello",
42│ isSensitive: true,
43│ },
44│ want: "11 Hello\r\n",
45│ },
46│ }
47│
48│ for _, tt := range tests {
49│ tt := tt
50│ t.Run(tt.name, func(t *testing.T) {
51│ t.Parallel()
52│
53│ s := geminitest.NewServer(t, gemini.HandlerFunc(func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
54│ w.Input(ctx, tt.args.prompt, tt.args.isSensitive)
55│ }))
56│
57│ defer s.Close()
58│
59│ conn, err := tls.DialWithDialer(
60│ &net.Dialer{Timeout: 1 * time.Second},
61│ "tcp",
62│ s.Addr,
63│ &tls.Config{
64│ InsecureSkipVerify: true,
65│ },
66│ )
67│ if err != nil {
68│ t.Fatalf("Unable to dial: error = %v", err)
69│ }
70│
71│ if _, err = fmt.Fprint(conn, "gemini://localhost.com/\r\n"); err != nil {
72│ t.Fatalf("Unable to write request: error = %v", err)
73│ }
74│
75│ got, err := io.ReadAll(conn)
76│ if err != nil {
77│ t.Fatalf("Unable to read response: error = %v", err)
78│ }
79│
80│ if string(got) != tt.want {
81│ t.Fatalf("*response.Input(%s, %t) = %s, want %s", tt.args.prompt, tt.args.isSensitive, string(got), tt.want)
82│ }
83│ })
84│ }
85│ }
86│
87│ func Test_response_Redirect(t *testing.T) {
88│ t.Parallel()
89│
90│ type args struct {
91│ redirectURL string
92│ isPermanant bool
93│ }
94│
95│ tests := []struct {
96│ name string
97│ want string
98│ args args
99│ }{
100│ {
101│ name: "",
102│ args: args{
103│ redirectURL: "gemini://foo.bar",
104│ isPermanant: false,
105│ },
106│ want: "30 gemini://foo.bar\r\n",
107│ },
108│ {
109│ name: "",
110│ args: args{
111│ redirectURL: "gemini://foo.bar",
112│ isPermanant: true,
113│ },
114│ want: "31 gemini://foo.bar\r\n",
115│ },
116│ }
117│
118│ for _, tt := range tests {
119│ tt := tt
120│ t.Run(tt.name, func(t *testing.T) {
121│ t.Parallel()
122│
123│ s := geminitest.NewServer(t, gemini.HandlerFunc(func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
124│ w.Redirect(ctx, tt.args.redirectURL, tt.args.isPermanant)
125│ }))
126│
127│ defer s.Close()
128│
129│ conn, err := tls.DialWithDialer(
130│ &net.Dialer{Timeout: 1 * time.Second},
131│ "tcp",
132│ s.Addr,
133│ &tls.Config{
134│ InsecureSkipVerify: true,
135│ },
136│ )
137│ if err != nil {
138│ t.Fatalf("Unable to dial: error = %v", err)
139│ }
140│
141│ if _, err = fmt.Fprint(conn, "gemini://localhost.com/\r\n"); err != nil {
142│ t.Fatalf("Unable to write request: error = %v", err)
143│ }
144│
145│ got, err := io.ReadAll(conn)
146│ if err != nil {
147│ t.Fatalf("Unable to read response: error = %v", err)
148│ }
149│
150│ if string(got) != tt.want {
151│ t.Fatalf("*response.Redirect(%s, %t) = %s, want %s", tt.args.redirectURL, tt.args.isPermanant, string(got), tt.want)
152│ }
153│ })
154│ }
155│ }
156│
157│ func Test_response_Success(t *testing.T) {
158│ t.Parallel()
159│
160│ type args struct {
161│ body io.Reader
162│ mimeType string
163│ }
164│
165│ tests := []struct {
166│ name string
167│ args args
168│ want string
169│ }{
170│ {
171│ name: "With mime type and body",
172│ args: args{
173│ mimeType: mime.FormatMediaType("text/gemini", map[string]string{"charset": "utf-8", "lang": "en"}),
174│ body: strings.NewReader("Hello, Gemini!"),
175│ },
176│ want: "20 text/gemini; charset=utf-8; lang=en\r\nHello, Gemini!",
177│ },
178│ {
179│ name: "With body only",
180│ args: args{
181│ mimeType: "",
182│ body: strings.NewReader("Hello, Gemini!"),
183│ },
184│ want: "20 \r\nHello, Gemini!",
185│ },
186│ {
187│ name: "With mime type only",
188│ args: args{
189│ mimeType: mime.FormatMediaType("text/gemini", map[string]string{"charset": "utf-8", "lang": "en"}),
190│ body: nil,
191│ },
192│ want: "20 text/gemini; charset=utf-8; lang=en\r\n",
193│ },
194│ {
195│ name: "With no mime type or body",
196│ args: args{
197│ mimeType: "",
198│ body: nil,
199│ },
200│ want: "20 \r\n",
201│ },
202│ }
203│
204│ for _, tt := range tests {
205│ tt := tt
206│ t.Run(tt.name, func(t *testing.T) {
207│ t.Parallel()
208│
209│ s := geminitest.NewServer(t, gemini.HandlerFunc(func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
210│ gotWriter := w.Success(ctx, tt.args.mimeType)
211│
212│ if tt.args.body != nil {
213│ _, err := io.Copy(gotWriter, tt.args.body)
214│ if err != nil {
215│ t.Fatal(err)
216│ }
217│ }
218│ }))
219│
220│ defer s.Close()
221│
222│ conn, err := tls.DialWithDialer(
223│ &net.Dialer{Timeout: 1 * time.Second},
224│ "tcp",
225│ s.Addr,
226│ &tls.Config{
227│ InsecureSkipVerify: true,
228│ },
229│ )
230│ if err != nil {
231│ t.Fatalf("Unable to dial: error = %v", err)
232│ }
233│
234│ if _, err = fmt.Fprint(conn, "gemini://localhost.com/\r\n"); err != nil {
235│ t.Fatalf("Unable to write request: error = %v", err)
236│ }
237│
238│ got, err := io.ReadAll(conn)
239│ if err != nil {
240│ t.Fatalf("Unable to read response: error = %v", err)
241│ }
242│
243│ if string(got) != tt.want {
244│ t.Fatalf("*response.Success(%s) = %s, want %s", tt.args.mimeType, string(got), tt.want)
245│ }
246│ })
247│ }
248│ }
249│
250│ func Test_response_Failure(t *testing.T) {
251│ t.Parallel()
252│
253│ type args struct {
254│ msg string
255│ code gemini.StatusCode
256│ }
257│
258│ tests := []struct {
259│ name string
260│ args args
261│ want string
262│ }{
263│ {
264│ name: "",
265│ args: args{
266│ code: gemini.StatusTemporaryFailure,
267│ msg: "oops",
268│ },
269│ want: "40 oops\r\n",
270│ },
271│ {
272│ name: "",
273│ args: args{
274│ code: gemini.StatusPermanentFailure,
275│ msg: "oops",
276│ },
277│ want: "50 oops\r\n",
278│ },
279│ {
280│ name: "",
281│ args: args{
282│ code: gemini.StatusTemporaryRedirect,
283│ msg: "oops",
284│ },
285│ want: "50 oops\r\n",
286│ },
287│ }
288│
289│ for _, tt := range tests {
290│ tt := tt
291│ t.Run(tt.name, func(t *testing.T) {
292│ t.Parallel()
293│
294│ s := geminitest.NewServer(t, gemini.HandlerFunc(func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
295│ w.Failure(ctx, tt.args.code, tt.args.msg)
296│ }))
297│
298│ defer s.Close()
299│
300│ conn, err := tls.DialWithDialer(
301│ &net.Dialer{Timeout: 1 * time.Second},
302│ "tcp",
303│ s.Addr,
304│ &tls.Config{
305│ InsecureSkipVerify: true,
306│ },
307│ )
308│ if err != nil {
309│ t.Fatalf("Unable to dial: error = %v", err)
310│ }
311│
312│ if _, err = fmt.Fprint(conn, "gemini://localhost.com/\r\n"); err != nil {
313│ t.Fatalf("Unable to write request: error = %v", err)
314│ }
315│
316│ got, err := io.ReadAll(conn)
317│ if err != nil {
318│ t.Fatalf("Unable to read response: error = %v", err)
319│ }
320│
321│ if string(got) != tt.want {
322│ t.Fatalf("*response.Failure(%d, %s) = %s, want %s", tt.args.code, tt.args.msg, string(got), tt.want)
323│ }
324│ })
325│ }
326│ }
────╯
· · ·
© 2024 source.community