databunker/src/utils_test.go

package main

import (
	"fmt"
	"io"
	"log"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	uuid "github.com/hashicorp/go-uuid"
	"github.com/securitybunker/databunker/src/utils"
)

func TestUtilUUID(t *testing.T) {
	for id := 1; id < 11; id++ {
		recordUUID, err := uuid.GenerateUUID()
		t.Logf("Checking[%d]: %s\n", id, recordUUID)
		if err != nil {
			t.Fatalf("Failed to generate UUID %s: %s ", recordUUID, err)
		} else if utils.CheckValidUUID(recordUUID) == false {
			t.Fatalf("Failed to validate UUID: %s ", recordUUID)
		}
	}
}

func TestUtilAppNames(t *testing.T) {
	goodApps := []string{"penn", "teller", "a123", "good_app"}
	for _, value := range goodApps {
		if utils.CheckValidApp(value) == false {
			t.Fatalf("Failed to validate good app name: %s ", value)
		}
	}
	badApps := []string{"P1", "4as", "_a", "a.a", "a a", "a!b"}
	for _, value := range badApps {
		if utils.CheckValidApp(value) == true {
			t.Fatalf("Failed to validate bad app name: %s ", value)
		}
	}
}

func TestUtilStringPatternMatch(t *testing.T) {
	goodJsons := []map[string]interface{}{
		{"pattern": "*", "name": "tom", "result": true},
		{"pattern": "aa", "name": "tom", "result": false},
		{"pattern": "", "name": "aa", "result": false},
		{"pattern": "test*", "name": "123testabc", "result": false},
		{"pattern": "test*", "name": "testabc", "result": true},
		{"pattern": "*test*", "name": "test1", "result": true},
		{"pattern": "*test", "name": "123testabc", "result": false},
		{"pattern": "*test", "name": "123test", "result": true},
	}
	for _, value := range goodJsons {
		if utils.StringPatternMatch(value["pattern"].(string), value["name"].(string)) != value["result"].(bool) {
			t.Fatalf("Failed in %s match %s\n", value["pattern"].(string), value["name"].(string))
		}
	}
}

func TestUtilGetJSONPost(t *testing.T) {
	goodJsons := []string{
		`{"login":"abc","name": "tom", "pass": "mylittlepony", "admin": true}`,
		`{"login":1,"name": "tom", "pass": "mylittlepony", "admin": true}`,
		`{"login":123,"name": "tom", "pass": "mylittlepony", "admin": true}`,
		`{"login":"1234","name": "tom", "pass": "mylittlepony", "admin": true}`,
	}
	for _, value := range goodJsons {
		request := httptest.NewRequest("POST", "/user", strings.NewReader(value))
		request.Header.Set("Content-Type", "application/json")
		result, err := utils.GetUserJSONStruct(request, "IL")
		if err != nil {
			t.Fatalf("Failed to parse json: %s, err: %s\n", value, err)
		}
		if len(result.LoginIdx) == 0 {
			t.Fatalf("Failed to parse login index from json: %s ", value)
		}
	}

	badJsons := []string{
		`{"login":true,"name": "tom", "pass": "mylittlepony", "admin": true}`,
		`{"login":null,"name": "tom", "pass": "mylittlepony", "admin": true}`,
	}
	for _, value := range badJsons {
		request := httptest.NewRequest("POST", "/user", strings.NewReader(value))
		request.Header.Set("Content-Type", "application/json")
		result, err := utils.GetUserJSONStruct(request, "IL")
		if err != nil {
			t.Fatalf("Failed to parse json: %s, err: %s\n", value, err)
		}
		if len(result.LoginIdx) != 0 {
			t.Fatalf("Failed to parse login index from json: %s ", value)
		}
	}
}

func TestUtilSMS(t *testing.T) {
	server := httptest.NewServer(e.reqMiddleware(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
		rw.Header().Set("Content-Type", "application/json")
		rw.WriteHeader(200)
		defer req.Body.Close()
		bodyBytes, _ := io.ReadAll(req.Body)
		log.Printf("body: %s\n", string(bodyBytes))
		if string(bodyBytes) != "Body=Data+Bunker+code+1234&From=from1234&To=4444" {
			t.Fatalf("bad request: %s", string(bodyBytes))
		}
	})))
	// Close the server when test finishes
	defer server.Close()
	client := server.Client()
	domain := server.URL
	var cfg Config
	sendCodeByPhoneDo(domain, client, 1234, "4444", &cfg)
}

func TestUtilNotifyConsentChange(t *testing.T) {
	q := make(chan string)
	server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
		rw.Header().Set("Content-Type", "application/json")
		rw.WriteHeader(200)
		defer req.Body.Close()
		bodyBytes, _ := io.ReadAll(req.Body)
		log.Printf("body: %s\n", string(bodyBytes))
		if string(bodyBytes) != `{"action":"consentchange","brief":"brief","identity":"user3@user3.com","mode":"email","status":"no"}` {
			q <- fmt.Sprintf("bad request in notifyConsentChange: %s", string(bodyBytes))
		} else {
			q <- "ok"
		}
	}))
	// Close the server when test finishes
	defer server.Close()
	notifyConsentChange(server.URL, "brief", "no", "email", "user3@user3.com")
	response := <-q
	if response != "ok" {
		t.Fatal(response)
	}
}

func TestUtilNotifyProfileNew(t *testing.T) {
	q := make(chan string)
	server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
		rw.Header().Set("Content-Type", "application/json")
		rw.WriteHeader(200)
		defer req.Body.Close()
		bodyBytes, _ := io.ReadAll(req.Body)
		log.Printf("body: %s\n", string(bodyBytes))
		if string(bodyBytes) != `{"action":"profilenew","identity":"user3@user3.com","mode":"email","profile":{"name":"alex"}}` {
			q <- fmt.Sprintf("bad request in notifyConsentChange: %s", string(bodyBytes))
		} else {
			q <- "ok"
		}
	}))
	// Close the server when test finishes
	defer server.Close()
	profile := []byte(`{"name":"alex"}`)
	notifyProfileNew(server.URL, profile, "email", "user3@user3.com")
	response := <-q
	if response != "ok" {
		t.Fatal(response)
	}
}

func TestUtilNotifyForgetMe(t *testing.T) {
	q := make(chan string)
	server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
		rw.Header().Set("Content-Type", "application/json")
		rw.WriteHeader(200)
		defer req.Body.Close()
		bodyBytes, _ := io.ReadAll(req.Body)
		log.Printf("body: %s\n", string(bodyBytes))
		if string(bodyBytes) != `{"action":"forgetme","identity":"user3@user3.com","mode":"email","profile":{"name":"alex"}}` {
			q <- fmt.Sprintf("bad request in notifyConsentChange: %s", string(bodyBytes))
		} else {
			q <- "ok"
		}
	}))
	// Close the server when test finishes
	defer server.Close()
	profile := []byte(`{"name":"alex"}`)
	notifyForgetMe(server.URL, profile, "email", "user3@user3.com")
	response := <-q
	if response != "ok" {
		t.Fatal(response)
	}
}

func TestUtilNotifyProfileChange(t *testing.T) {
	q := make(chan string)
	server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
		rw.Header().Set("Content-Type", "application/json")
		rw.WriteHeader(200)
		defer req.Body.Close()
		bodyBytes, _ := io.ReadAll(req.Body)
		log.Printf("body: %s\n", string(bodyBytes))
		if string(bodyBytes) != `{"action":"profilechange","identity":"user3@user3.com","mode":"email","old":{"name":"alex2"},"profile":{"name":"alex3"}}` {
			q <- fmt.Sprintf("bad request in notifyConsentChange: %s", string(bodyBytes))
		} else {
			q <- "ok"
		}
	}))
	// Close the server when test finishes
	defer server.Close()
	profile := []byte(`{"name":"alex2"}`)
	profile2 := []byte(`{"name":"alex3"}`)
	notifyProfileChange(server.URL, profile, profile2, "email", "user3@user3.com")
	response := <-q
	if response != "ok" {
		t.Fatal(response)
	}
}