tools_test.go

package toolkit

import (
	"io"
	"net/http"
	"net/http/httptest"
	"os"
	"testing"
)

func TestTools_RandomString(t *testing.T) {
	var testTools Tools

	tests := []struct {
		name   string
		length int
	}{
		{"Ten", 10},
		{"Twenty", 20},
		{"One", 1},
	}

	for _, entry := range tests {
		t.Run(entry.name, func(t *testing.T) {
			s := testTools.RandomString(entry.length)

			if len(s) != entry.length {
				t.Errorf("expected length %d, received %d", entry.length, len(s))
			}
		})
	}
}

func TestTools_CreateNewDirectory(t *testing.T) {
	tests := []struct {
		name    string
		dirPath string
		err     error
	}{
		{"Non-existing path", "./testFolder", nil},
		{"Non-existing path", "./testUploads", nil},
	}

	for _, entry := range tests {
		t.Run(entry.name, func(t *testing.T) {
			var tools Tools
			// Call the CreateNewDirectory function
			err := tools.CreateNewDirectory(entry.dirPath)
			if err != nil && entry.err == nil {
				t.Errorf("expected no error, but received %+v", err)
			}

			// Remove the directory if it was created
			if _, err = os.Stat(entry.dirPath); err == nil {
				err = os.RemoveAll(entry.dirPath)
				if err != nil {
					t.Errorf("failed to remove path %s. %+v", entry.dirPath, err)
				}
			}
		})
	}
}

func TestTools_Slugify(t *testing.T) {
	tests := []struct {
		name     string
		expected string
		err      bool
	}{
		{"Hello, World!", "hello-world", false},
		{"    !Hello && World!", "hello-world", false},
		// Cyrillic characters
		{"Привет, мир!Hello, World!", "hello-world", false},
		{"Привет, мир!", "", true},
		{"88GoLang!PyThon===Java?     TYPESCRIPT@   ", "88golang-python-java-typescript", false},
		{"        ", "", true},
		{"!!!!", "", true},
	}
	var tools Tools
	for _, entry := range tests {
		t.Run(entry.name, func(t *testing.T) {
			result, err := tools.Slugify(entry.name)

			if result != entry.expected {
				t.Errorf("expected %s, but received %s", entry.expected, result)
			}

			if err != nil && !entry.err {
				t.Errorf("expected no error, but received %+v", err)
			}
		})
	}

}

func TestTools_DownloadStaticFile(t *testing.T) {
	// Define and initialize response recorder and request
	resp := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodPost, "/download", nil)
	// Initalize tools
	tools := &Tools{}

	// Call TestTools_DownloadStaticFile
	tools.DownloadStaticFile(resp, req, "./testdata", "img.png", "hello-world.png")

	// Get result of the response
	result := resp.Result()
	// Close the body to ensure no resource leak
	defer result.Body.Close()

	// Check if the file downloaded entirely by checking content length
	if result.Header["Content-Length"][0] != "5003" {
		t.Error("wrong content-length of", result.Header["Content-Length"][0])
	}

	// Check headers
	if result.Header["Content-Disposition"][0] != "attachment; filename=\"hello-world.png\"" {
		t.Error("wrong content-length of", result.Header["Content-Length"][0])
	}

	// Check for an error
	_, err := io.ReadAll(result.Body)
	if err != nil {
		t.Error(err)
	}

}