5 files changed, 143 insertions, 10 deletions

diff --git a/src/dkv/api/initialise.go b/src/dkv/api/initialise.go
index ac8825e..228a46f 100644
--- a/src/dkv/api/initialise.go
+++ b/src/dkv/api/initialise.go
@@ -18,6 +18,7 @@ package api
 
 import (
 	"errors"
+	"log"
 	"os"
 )
 
@@ -31,6 +32,7 @@ func Initialise() error {
 	if os.Getenv("DATASTORE") == "" {
 		return errors.New("DATASTORE environment variable not set.")
 	}
+
 	if os.Getenv("DATASTORE") == "consul" {
 		Datastore = &ConsulStruct{}
 	} else if os.Getenv("DATASTORE") == "cassandra" {
@@ -38,10 +40,20 @@ func Initialise() error {
 	} else {
 		return errors.New("Unrecognised Datastore. Supports only consul or cassandra")
 	}
+
+	jsonExists, err := JsonChecker(JSONPATH)
+	if jsonExists == false {
+		log.Println("[INFO] token_service_map.json not found. Creating.")
+		err = JsonCreate(JSONPATH)
+		if err != nil {
+			return err
+		}
+	}
+
 	KeyValues = &KeyValuesStruct{}
 	Directory = &DirectoryStruct{directory: ""}
 
-	err := Datastore.InitializeDatastoreClient()
+	err = Datastore.InitializeDatastoreClient()
 	if err != nil {
 		return err
 	}
diff --git a/src/dkv/api/initialise_test.go b/src/dkv/api/initialise_test.go
index 363edce..2d54bec 100644
--- a/src/dkv/api/initialise_test.go
+++ b/src/dkv/api/initialise_test.go
@@ -22,41 +22,93 @@ import (
 	"testing"
 )
 
-func TestInitialise_cassandra(t *testing.T) {
+func TestInitialise_consul(t *testing.T) {
 	oldDatastore_ip := os.Getenv("DATASTORE_IP")
 	oldDatastore_type := os.Getenv("DATASTORE")
+	oldJsonExists := JsonChecker
+	oldJsonCreate := JsonCreate
 
 	os.Setenv("DATASTORE_IP", "localhost")
-	os.Setenv("DATASTORE", "cassandra")
+	os.Setenv("DATASTORE", "consul")
 
 	defer func() {
 		os.Setenv("DATASTORE_IP", oldDatastore_ip)
 		os.Setenv("DATASTORE", oldDatastore_type)
+		JsonCreate = oldJsonCreate
+		JsonChecker = oldJsonExists
 	}()
 
+	JsonChecker = func(path string) (bool, error) {
+		return false, nil
+	}
+
+	JsonCreate = func(path string) error {
+		return nil
+	}
+
 	err := Initialise()
-	assert.Nil(t, err)
+	assert.NotNil(t, err)
 }
-func TestInitialise_consulError(t *testing.T) {
+
+func TestInitialise_cassandra(t *testing.T) {
 	oldDatastore_ip := os.Getenv("DATASTORE_IP")
 	oldDatastore_type := os.Getenv("DATASTORE")
+	oldMOUNTPATH := os.Getenv("MOUNTPATH")
+	oldJsonChecker := JsonChecker
 
 	os.Setenv("DATASTORE_IP", "localhost")
-	os.Setenv("DATASTORE", "consul")
+	os.Setenv("DATASTORE", "cassandra")
 
 	defer func() {
 		os.Setenv("DATASTORE_IP", oldDatastore_ip)
 		os.Setenv("DATASTORE", oldDatastore_type)
+		os.Setenv("MOUNTPATH", oldMOUNTPATH)
+		JsonChecker = oldJsonChecker
 	}()
 
+	JsonChecker = func(path string) (bool, error) {
+		return true, nil
+	}
+
+	err := Initialise()
+	assert.Nil(t, err)
+}
+
+func TestInitialise_datastoreUnknown(t *testing.T) {
+	datastore := os.Getenv("DATASTORE")
+	defer os.Setenv("DATASTORE", datastore)
+	os.Setenv("DATASTORE", "test")
+
 	err := Initialise()
 	assert.NotNil(t, err)
 }
 
-func TestInitialise_datastoreEmptyError(t *testing.T) {
+func TestInitialise_datastoreEmpty(t *testing.T) {
 	datastore := os.Getenv("DATASTORE")
-	os.Unsetenv("DATASTORE")
 	defer os.Setenv("DATASTORE", datastore)
+	os.Setenv("DATASTORE", "")
+
+	err := Initialise()
+	assert.NotNil(t, err)
+}
+
+func TestInitialise_noJSON(t *testing.T) {
+	oldDatastore_ip := os.Getenv("DATASTORE_IP")
+	oldDatastore_type := os.Getenv("DATASTORE")
+	oldJsonChecker := JsonChecker
+
+	os.Setenv("DATASTORE_IP", "localhost")
+	os.Setenv("DATASTORE", "consul")
+
+	defer func() {
+		os.Setenv("DATASTORE_IP", oldDatastore_ip)
+		os.Setenv("DATASTORE", oldDatastore_type)
+		JsonChecker = oldJsonChecker
+	}()
+
+	JsonChecker = func(path string) (bool, error) {
+		return false, nil
+	}
 
 	err := Initialise()
 	assert.NotNil(t, err)
diff --git a/src/dkv/api/token_service_map.json b/src/dkv/api/token_service_map.json
deleted file mode 100644
index 51d4dbb..0000000
--- a/src/dkv/api/token_service_map.json
+++ /dev/null
@@ -1 +0,0 @@
-[{"token":"default","service":"default"}]
\ No newline at end of file
diff --git a/src/dkv/api/utils.go b/src/dkv/api/utils.go
index c1094ab..383f99c 100644
--- a/src/dkv/api/utils.go
+++ b/src/dkv/api/utils.go
@@ -27,6 +27,8 @@ var (
 	IoutilRead  = ioutil.ReadFile
 	IoutilWrite = ioutil.WriteFile
 	JsonReader  = ReadJSON
+	JsonChecker = CheckJSONExists
+	JsonCreate  = CreateJSON
 )
 
 type Token_service_map struct {
@@ -34,6 +36,34 @@ type Token_service_map struct {
 	Service string `json:"service"`
 }
 
+func CheckJSONExists(path string) (bool, error) {
+	_, err := IoutilRead(path)
+
+	if err != nil {
+		return false, err
+	} else {
+		return true, nil
+	}
+}
+
+func CreateJSON(path string) error {
+	var tsm Token_service_map
+	var tsm_list []Token_service_map
+
+	tsm.Token = "default"
+	tsm.Service = "default"
+	tsm_list = append(tsm_list, tsm)
+	raw, err := json.Marshal(tsm_list)
+	if err != nil {
+		return err
+	}
+	err = IoutilWrite(path, raw, 0644)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 func ReadJSON(path string) ([]Token_service_map, error) {
 	var tsm_list []Token_service_map
 	// raw, err := ioutil.ReadFile("./token_service_map.json")
diff --git a/src/dkv/api/utils_test.go b/src/dkv/api/utils_test.go
index 0bca7c5..8b21e52 100644
--- a/src/dkv/api/utils_test.go
+++ b/src/dkv/api/utils_test.go
@@ -21,6 +21,47 @@ import (
 	"testing"
 )
 
+func TestCheckJSONExists(t *testing.T) {
+	oldIoutilRead := IoutilRead
+
+	defer func() {
+		IoutilRead = oldIoutilRead
+	}()
+
+	IoutilRead = func(path string) ([]byte, error) {
+		return []byte("test"), nil
+	}
+
+	_, err := JsonChecker("path")
+	assert.Equal(t, nil, err, "Error should be nil.")
+}
+
+func TestCreateJSON(t *testing.T) {
+	oldIoutilWrite := IoutilWrite
+
+	defer func() {
+		IoutilWrite = oldIoutilWrite
+	}()
+
+	IoutilWrite = func(val string, b []byte, f os.FileMode) error {
+		return nil
+	}
+
+	err := JsonCreate("path")
+	assert.Equal(t, nil, err, "Error should be nil.")
+}
+
+func TestCheckJSONExists_err(t *testing.T) {
+	oldIoutilRead := IoutilRead
+
+	defer func() {
+		IoutilRead = oldIoutilRead
+	}()
+
+	_, err := JsonChecker("path")
+	assert.NotNil(t, err, "Err should not be nil.")
+}
+
 func TestReadJSON(t *testing.T) {
 	oldIoutilRead := IoutilRead
 
@@ -77,7 +118,6 @@ func TestWriteJSON(t *testing.T) {
 	assert.Equal(t, nil, err, "Error should be nil.")
 
 }
-
 func TestDeleteInJSON(t *testing.T) {
 	oldReadJson := JsonReader
 	defer func() {