82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
package cgi_test
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"tildegit.org/tjp/gus"
|
|
"tildegit.org/tjp/gus/contrib/cgi"
|
|
"tildegit.org/tjp/gus/gemini"
|
|
)
|
|
|
|
func TestCGIDirectory(t *testing.T) {
|
|
tlsconf, err := gemini.FileTLS("testdata/server.crt", "testdata/server.key")
|
|
require.Nil(t, err)
|
|
|
|
handler := cgi.GeminiCGIDirectory("/cgi-bin", "./testdata")
|
|
server, err := gemini.NewServer(context.Background(), "localhost", "tcp", "127.0.0.1:0", handler, nil, tlsconf)
|
|
require.Nil(t, err)
|
|
|
|
go func() { assert.Nil(t, server.Serve()) }()
|
|
defer server.Close()
|
|
|
|
tests := []struct {
|
|
requestPath string
|
|
responseCode gus.Status
|
|
responseBody string
|
|
}{
|
|
{
|
|
requestPath: "/cgi-bin/hello_world",
|
|
responseCode: gemini.StatusSuccess,
|
|
responseBody: "hello, world!\n",
|
|
},
|
|
{
|
|
requestPath: "/cgi-bin/server.key",
|
|
responseCode: gemini.StatusNotFound,
|
|
},
|
|
{
|
|
requestPath: "/cgi-bin/non-existent",
|
|
responseCode: gemini.StatusNotFound,
|
|
},
|
|
{
|
|
requestPath: "/cgi-bin/fails",
|
|
responseCode: gemini.StatusCGIError,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.requestPath, func(t *testing.T) {
|
|
conn, err := tls.Dial(
|
|
server.Network(),
|
|
server.Address(),
|
|
&tls.Config{InsecureSkipVerify: true},
|
|
)
|
|
require.Nil(t, err)
|
|
|
|
_, err = fmt.Fprintf(conn, "gemini://%s%s\r\n", server.Address(), test.requestPath)
|
|
require.Nil(t, err)
|
|
|
|
response, err := io.ReadAll(conn)
|
|
require.Nil(t, err)
|
|
|
|
code, err := strconv.Atoi(string(response[:2]))
|
|
if assert.Nil(t, err) {
|
|
assert.Equal(t, test.responseCode, gus.Status(code))
|
|
}
|
|
|
|
_, body, found := strings.Cut(string(response), "\r\n")
|
|
if assert.True(t, found) && test.responseBody != "" {
|
|
assert.Equal(t, test.responseBody, body)
|
|
}
|
|
})
|
|
}
|
|
}
|