Initial server without functionality

master
Bel LaPointe 2018-10-07 23:06:07 -06:00
parent 9f057891bc
commit cfb31cf6a5
2 changed files with 140 additions and 0 deletions

83
server/server.go Normal file
View File

@ -0,0 +1,83 @@
package server
import (
"net/http"
"os"
"os/signal"
"path"
"strings"
"syscall"
)
type Server struct {
addr string
}
func New(addr string) (*Server, error) {
return &Server{
addr: addr,
}, nil
}
func (s *Server) Serve() error {
var err error
go func() {
port := s.addr
if !strings.HasPrefix(port, ":") {
port = ":" + port
}
err = http.ListenAndServe(port, s)
}()
sigc := make(chan os.Signal)
signal.Notify(sigc,
syscall.SIGHUP,
syscall.SIGINT,
syscall.SIGTERM,
syscall.SIGQUIT,
)
<-sigc
return err
}
func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
switch advance(r) {
case "api":
s.api(w, r)
default:
s.bad(w, r)
}
}
func (s *Server) bad(w http.ResponseWriter, r *http.Request) {
http.NotFound(w, r)
}
func (s *Server) api(w http.ResponseWriter, r *http.Request) {
switch advance(r) {
case "feed":
s.feed(w, r)
default:
s.bad(w, r)
}
}
func (s *Server) feed(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "GET":
case "POST":
case "PUT":
default:
s.bad(w, r)
}
}
func advance(r *http.Request) string {
p := path.Clean("/" + r.URL.Path)
i := strings.Index(p[1:], "/") + 1
if i <= 0 {
r.URL.Path = "/"
return p[1:]
}
r.URL.Path = p[i:]
return p[1:i]
}

57
server/server_test.go Normal file
View File

@ -0,0 +1,57 @@
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
}