58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"syscall"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
const testPort = "39231"
|
|
|
|
func Test_Server(t *testing.T) {
|
|
cases := []struct {
|
|
}{
|
|
{},
|
|
}
|
|
|
|
for _, _ = range cases {
|
|
var err error
|
|
s, err := New(testPort)
|
|
if err != nil {
|
|
t.Errorf("failed to create server: %v", err)
|
|
}
|
|
go s.Serve()
|
|
time.Sleep(time.Second * 1)
|
|
if err := checkStatus("GET", "", http.StatusNotFound); err != nil {
|
|
t.Errorf(err.Error())
|
|
}
|
|
if err := checkStatus("GET", "api", http.StatusNotFound); err != nil {
|
|
t.Errorf(err.Error())
|
|
}
|
|
if err := checkStatus("GET", "api/feed", http.StatusOK); err != nil {
|
|
t.Errorf(err.Error())
|
|
}
|
|
if err := checkStatus("POST", "api/feed", http.StatusOK); err != nil {
|
|
t.Errorf(err.Error())
|
|
}
|
|
if err := checkStatus("PUT", "api/feed", http.StatusOK); err != nil {
|
|
t.Errorf(err.Error())
|
|
}
|
|
syscall.Kill(syscall.Getpid(), syscall.SIGINT)
|
|
}
|
|
}
|
|
|
|
func checkStatus(method, path string, code int) error {
|
|
client := &http.Client{}
|
|
r, _ := http.NewRequest(method, "http://localhost:"+testPort+"/"+path, nil)
|
|
resp, err := client.Do(r)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to %v server: %v", method, err)
|
|
}
|
|
if resp.StatusCode != code {
|
|
return fmt.Errorf("%s %q did not return %v: %v", method, path, code, resp.StatusCode)
|
|
}
|
|
return nil
|
|
}
|