package serve

import (
	"io/ioutil"
	"local/dynamodb/server/config"
	"net/http"
	"net/http/httptest"
	"os"
	"strings"
	"testing"
)

var validNS = "ns"
var validKey = "key"
var validValue = "value"

func TestRoutes(t *testing.T) {
	os.Setenv("DB", "MAP")
	config.New()
	s := New()
	if err := s.Routes(); err != nil {
		t.Fatalf("cannot routes(): %v", err)
	}
	config.Values().DB.Set(validKey, []byte("anything"))

	cases := []struct {
		path   string
		method string
		status int
	}{
		{
			path:   "/" + validKey,
			method: "GET",
			status: http.StatusOK,
		},
		{
			path:   "/" + validKey,
			method: "PUT",
			status: http.StatusOK,
		},
		{
			path:   "/" + validKey,
			method: "DELETE",
			status: http.StatusOK,
		},
		{
			path:   "/" + validKey,
			method: "POST",
			status: http.StatusNotFound,
		},
		{
			path:   "/",
			method: "GET",
			status: http.StatusNotFound,
		},
		{
			path:   "/",
			method: "PUT",
			status: http.StatusNotFound,
		},
		{
			path:   "/",
			method: "DELETE",
			status: http.StatusNotFound,
		},
		{
			path:   "/",
			method: "POST",
			status: http.StatusNotFound,
		},
		{
			path:   "",
			method: "GET",
			status: http.StatusNotFound,
		},
		{
			path:   "",
			method: "PUT",
			status: http.StatusNotFound,
		},
		{
			path:   "",
			method: "DELETE",
			status: http.StatusNotFound,
		},
		{
			path:   "",
			method: "POST",
			status: http.StatusNotFound,
		},
	}

	for _, c := range cases {
		//t.Logf("CASE %v", c)
		r, err := http.NewRequest(c.method, c.path, strings.NewReader(validValue))
		if err != nil {
			t.Fatalf("err making request: %v", err)
		}
		w := httptest.NewRecorder()
		s.ServeHTTP(w, r)
		if w.Code != c.status {
			t.Errorf("wrong status for %q on %q: %v, want %v", c.method, c.path, w.Code, c.status)
		}
	}
}

func TestPutGet(t *testing.T) {
	os.Setenv("DB", "MAP")
	config.New()
	s := New()
	if err := s.Routes(); err != nil {
		t.Errorf("cannot routes(): %v", err)
	}

	w := httptest.NewRecorder()
	r, err := http.NewRequest("PUT", "/"+validNS+"/"+validKey, strings.NewReader(validValue))
	if err != nil {
		t.Fatalf("err making put request: %v", err)
	}
	s.put(w, r)
	if w.Code != http.StatusOK {
		t.Fatalf("err status on put: %v", w.Code)
	}

	w = httptest.NewRecorder()
	r, err = http.NewRequest("GET", "/not_"+validNS+"/"+validKey, nil)
	if err != nil {
		t.Fatalf("err making get request: %v", err)
	}
	s.get(w, r)
	if w.Code != http.StatusNotFound {
		t.Fatalf("err status on bad get: %v", w.Code)
	}

	w = httptest.NewRecorder()
	r, err = http.NewRequest("GET", "/"+validNS+"/"+validKey, nil)
	if err != nil {
		t.Fatalf("err making get request: %v", err)
	}
	s.get(w, r)
	if w.Code != http.StatusOK {
		t.Fatalf("err status on get: %v", w.Code)
	} else if v, err := ioutil.ReadAll(w.Body); err != nil {
		t.Fatalf("err reading response body on get: %v", err)
	} else if string(v) != validValue {
		t.Fatalf("wrong response body on get: %v", string(v))
	}
}