1 files changed, 276 insertions, 0 deletions
diff --git a/src/dkv/api/registrationHandlers_test.go b/src/dkv/api/registrationHandlers_test.go
new file mode 100644
index 0000000..3482e8d
--- /dev/null
+++ b/src/dkv/api/registrationHandlers_test.go
@@ -0,0 +1,276 @@
+/*
+ * Copyright 2018 Intel Corporation, Inc
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package api
+
+import (
+	"bytes"
+	"encoding/json"
+	"github.com/gorilla/mux"
+	"github.com/stretchr/testify/assert"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func RouterRegister() *mux.Router {
+	router := mux.NewRouter()
+	router.HandleFunc("/v1/register", HandleServiceCreate).Methods("POST")
+	router.HandleFunc("/v1/register/{token}", HandleServiceGet).Methods("GET")
+	router.HandleFunc("/v1/register/{token}", HandleServiceDelete).Methods("DELETE")
+	return router
+}
+
+func RouterRegisterSubdomain() *mux.Router {
+	router := mux.NewRouter()
+	router.HandleFunc("/v1/register/{token}/subdomain", HandleServiceSubdomainCreate).Methods("POST")
+	router.HandleFunc("/v1/register/{token}/subdomain/{subdomain}", HandleServiceSubdomainDelete).Methods("DELETE")
+	return router
+}
+
+func TestHandleServiceCreate(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	body := &CreateRegisterServiceBody{
+		Domain: "test",
+	}
+
+	b, _ := json.Marshal(body)
+
+	request, _ := http.NewRequest("POST", "/v1/register", bytes.NewBuffer(b))
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 200, response.Code, "200 response is expected")
+}
+
+func TestHandleServiceCreate_default(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	body := &CreateRegisterServiceBody{
+		Domain: "default",
+	}
+
+	b, _ := json.Marshal(body)
+
+	request, _ := http.NewRequest("POST", "/v1/register", bytes.NewBuffer(b))
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 400, response.Code, "400 response is expected")
+}
+func TestHandleServiceCreate_no_body(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	body := &CreateRegisterServiceBody{}
+
+	b, _ := json.Marshal(body)
+
+	request, _ := http.NewRequest("POST", "/v1/register", bytes.NewBuffer(b))
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 400, response.Code, "400 response is expected")
+}
+
+func TestHandleServiceCreate_no_domain(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	body := &CreateRegisterServiceBody{
+		Domain: "",
+	}
+
+	b, _ := json.Marshal(body)
+
+	request, _ := http.NewRequest("POST", "/v1/register", bytes.NewBuffer(b))
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 400, response.Code, "400 response is expected")
+}
+
+func TestHandleServiceGet(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+	defer func() { Directory = oldDirectory }()
+
+	request, _ := http.NewRequest("GET", "/v1/register/token1", nil)
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 200, response.Code, "200 response is expected")
+}
+
+func TestHandleServiceGet_err(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectoryErr{}
+	defer func() { Directory = oldDirectory }()
+
+	request, _ := http.NewRequest("GET", "/v1/register/token1", nil)
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 500, response.Code, "500 response is expected")
+}
+func TestHandleServiceDelete(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	request, _ := http.NewRequest("DELETE", "/v1/register/token1", nil)
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 200, response.Code, "200 response is expected")
+}
+
+func TestHandleServiceDelete_default(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	request, _ := http.NewRequest("DELETE", "/v1/register/default", nil)
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 406, response.Code, "406 response is expected")
+}
+
+func TestHandleServiceDelete_err(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectoryErr{}
+
+	defer func() { Directory = oldDirectory }()
+
+	request, _ := http.NewRequest("DELETE", "/v1/register/token1", nil)
+	response := httptest.NewRecorder()
+	RouterRegister().ServeHTTP(response, request)
+
+	assert.Equal(t, 500, response.Code, "500 response is expected")
+}
+
+func TestHandleServiceSuddomainCreate(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	body := &CreateServiceSubdomainBody{
+		Subdomain: "test",
+	}
+
+	b, _ := json.Marshal(body)
+
+	request, _ := http.NewRequest("POST", "/v1/register/token1/subdomain", bytes.NewBuffer(b))
+	response := httptest.NewRecorder()
+	RouterRegisterSubdomain().ServeHTTP(response, request)
+
+	assert.Equal(t, 200, response.Code, "200 response is expected")
+}
+
+func TestHandleServiceSuddomainCreate_no_body(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	body := &CreateServiceSubdomainBody{}
+
+	b, _ := json.Marshal(body)
+
+	request, _ := http.NewRequest("POST", "/v1/register/token1/subdomain", bytes.NewBuffer(b))
+	response := httptest.NewRecorder()
+	RouterRegisterSubdomain().ServeHTTP(response, request)
+
+	assert.Equal(t, 400, response.Code, "400 response is expected")
+}
+
+func TestHandleServiceSuddomainCreate_no_subdomain(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	body := &CreateServiceSubdomainBody{
+		Subdomain: "",
+	}
+
+	b, _ := json.Marshal(body)
+
+	request, _ := http.NewRequest("POST", "/v1/register/token1/subdomain", bytes.NewBuffer(b))
+	response := httptest.NewRecorder()
+	RouterRegisterSubdomain().ServeHTTP(response, request)
+
+	assert.Equal(t, 400, response.Code, "400 response is expected")
+}
+
+func TestHandleServiceSuddomainCreate_err(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectoryErr{}
+
+	defer func() { Directory = oldDirectory }()
+
+	body := &CreateServiceSubdomainBody{
+		Subdomain: "test",
+	}
+
+	b, _ := json.Marshal(body)
+
+	request, _ := http.NewRequest("POST", "/v1/register/token1/subdomain", bytes.NewBuffer(b))
+	response := httptest.NewRecorder()
+	RouterRegisterSubdomain().ServeHTTP(response, request)
+
+	assert.Equal(t, 500, response.Code, "500 response is expected")
+}
+
+func TestHandleServiceSuddomainDelete(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectory{}
+
+	defer func() { Directory = oldDirectory }()
+
+	request, _ := http.NewRequest("DELETE", "/v1/register/token1/subdomain/subdomain1", nil)
+	response := httptest.NewRecorder()
+	RouterRegisterSubdomain().ServeHTTP(response, request)
+
+	assert.Equal(t, 200, response.Code, "200 response is expected")
+}
+
+func TestHandleServiceSuddomainDelete_err(t *testing.T) {
+	oldDirectory := Directory
+	Directory = &FakeDirectoryErr{}
+
+	defer func() { Directory = oldDirectory }()
+
+	request, _ := http.NewRequest("DELETE", "/v1/register/token1/subdomain/subdomain1", nil)
+	response := httptest.NewRecorder()
+	RouterRegisterSubdomain().ServeHTTP(response, request)
+
+	assert.Equal(t, 500, response.Code, "500 response is expected")
+}