gocustomurls/handlers_test.go

package main

import (
	"encoding/json"
	"fmt"
	"net/http"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"golang.org/x/net/html"
)

func TestIntegrationReloadRules(t *testing.T) {
	if testing.Short() {
		t.Skip("skipping reload rules integration test")
	}

	c := &Config{}

	tmpDir := t.TempDir()

	tempLogsDir := fmt.Sprintf("%s/tmp/logs", tmpDir)
	tempConfigDir := fmt.Sprintf("%s/tmp/config", tmpDir)

	confFile := fmt.Sprintf("%s/config.json", tempConfigDir)
	rulesFile := fmt.Sprintf("%s/rules.json", tempConfigDir)
	lFp := fmt.Sprintf("%s/app.log", tempLogsDir)

	mkDirForTest(t, tempConfigDir)
	mkDirForTest(t, tempLogsDir)

	rulesJsonFp := "testData/rules.json"
	cpFileForTest(t, rulesJsonFp, rulesFile)

	p := ParsedConf{
		RulesFp:      rulesFile,
		LogFp:        lFp,
		Port:         "9050",
		Compression:  true,
		SizeToRotate: "5MB",
	}
	jsonString, _ := json.Marshal(p)
	writeForTest(t, confFile, jsonString)

	_, err := c.LoadMainConfigFile(confFile)
	assert.Equal(t, err, nil)

	l, err := newFileLogger(p.LogFp, p.SizeToRotate, p.Compression)
	assert.NotEmpty(t, l)
	assert.Equal(t, err, nil)

	isEmpty := isFileEmpty(t, lFp)
	assert.Equal(t, isEmpty, true)

	app := newTestApp(t, c, l)
	app.Routes()
	ts := newTestServer(t, app.Mux)
	defer ts.Close()

	code, _, body := ts.get(t, "/reloadRules")
	assert.Equal(t, code, http.StatusOK)
	assert.Equal(t, string(body), "ok")

	isEmpty = isFileEmpty(t, lFp)
	assert.Equal(t, isEmpty, true)

	oldRules := app.Config.MappingRules.Mappings
	assert.NotEmpty(t, oldRules)

	nRulesJsonFp := "testData/rules2.json"
	cpFileForTest(t, nRulesJsonFp, rulesFile)

	code, _, body = ts.get(t, "/reloadRules")
	assert.Equal(t, code, http.StatusOK)
	assert.Equal(t, string(body), "ok")

	isEmpty = isFileEmpty(t, lFp)
	assert.Equal(t, isEmpty, true)

	newRules := app.Config.MappingRules.Mappings
	assert.NotEmpty(t, newRules)

	assert.NotEqualValues(t, oldRules, newRules)
}

func TestIntegrationServeRules(t *testing.T) {
	if testing.Short() {
		t.Skip("skipping serve rules integration test")
	}

	t.Run("return default html if go-get is missing", func(t *testing.T) {
		c := &Config{}

		tmpDir := t.TempDir()

		tempLogsDir := fmt.Sprintf("%s/tmp/logs", tmpDir)
		tempConfigDir := fmt.Sprintf("%s/tmp/config", tmpDir)

		confFile := fmt.Sprintf("%s/config.json", tempConfigDir)
		rulesFile := fmt.Sprintf("%s/rules.json", tempConfigDir)
		lFp := fmt.Sprintf("%s/app.log", tempLogsDir)

		mkDirForTest(t, tempConfigDir)
		mkDirForTest(t, tempLogsDir)

		rulesJsonFp := "testData/rules2.json"
		cpFileForTest(t, rulesJsonFp, rulesFile)

		p := ParsedConf{
			RulesFp:      rulesFile,
			LogFp:        lFp,
			Port:         "9050",
			Compression:  true,
			SizeToRotate: "5MB",
		}
		jsonString, _ := json.Marshal(p)
		writeForTest(t, confFile, jsonString)

		_, err := c.LoadMainConfigFile(confFile)
		assert.Equal(t, err, nil)

		err = c.LoadMappingFile(p.RulesFp)
		assert.Equal(t, err, nil)

		l, err := newFileLogger(p.LogFp, p.SizeToRotate, p.Compression)
		assert.NotEmpty(t, l)
		assert.Equal(t, err, nil)

		isEmpty := isFileEmpty(t, lFp)
		assert.Equal(t, isEmpty, true)

		app := newTestApp(t, c, l)
		app.Routes()

		ts := newTestServer(t, app.Mux)
		defer ts.Close()

		code, _, body := ts.get(t, "/touche")
		assert.Equal(t, code, http.StatusOK)
		expected, _ := getDefaultHtml()
		assert.Equal(t, string(body), string(expected))

		isEmpty = isFileEmpty(t, lFp)
		assert.Equal(t, isEmpty, false)

	})

	t.Run("return normal html if go-get is included and the package exists", func(t *testing.T) {
		c := &Config{}

		tmpDir := t.TempDir()

		tempLogsDir := fmt.Sprintf("%s/tmp/logs", tmpDir)
		tempConfigDir := fmt.Sprintf("%s/tmp/config", tmpDir)

		confFile := fmt.Sprintf("%s/config.json", tempConfigDir)
		rulesFile := fmt.Sprintf("%s/rules.json", tempConfigDir)
		lFp := fmt.Sprintf("%s/app.log", tempLogsDir)

		mkDirForTest(t, tempConfigDir)
		mkDirForTest(t, tempLogsDir)

		rulesJsonFp := "testData/rules3.json"
		cpFileForTest(t, rulesJsonFp, rulesFile)

		p := ParsedConf{
			RulesFp:      rulesFile,
			LogFp:        lFp,
			Port:         "9050",
			Compression:  true,
			SizeToRotate: "5MB",
		}
		jsonString, _ := json.Marshal(p)
		writeForTest(t, confFile, jsonString)

		_, err := c.LoadMainConfigFile(confFile)
		assert.Equal(t, err, nil)

		err = c.LoadMappingFile(p.RulesFp)
		assert.Equal(t, err, nil)

		l, err := newFileLogger(p.LogFp, p.SizeToRotate, p.Compression)
		assert.NotEmpty(t, l)
		assert.Equal(t, err, nil)

		isEmpty := isFileEmpty(t, lFp)
		assert.Equal(t, isEmpty, true)

		app := newTestApp(t, c, l)
		app.Routes()
		ts := newTestServer(t, app.Mux)
		defer ts.Close()

		code, _, body := ts.get(t, "/x/touche?go-get=1")
		assert.Equal(t, code, http.StatusOK)
		assert.NotEmpty(t, string(body))

		_, err = html.Parse(strings.NewReader(string(body)))
		assert.Equal(t, err, nil)

		t.Logf("Printing returned html => %s", string(body))
		isEmpty = isFileEmpty(t, lFp)
		assert.Equal(t, isEmpty, false)
	})

	t.Run("return default html if go-get is included and the package does not exists", func(t *testing.T) {
		c := &Config{}

		tmpDir := t.TempDir()

		tempLogsDir := fmt.Sprintf("%s/tmp/logs", tmpDir)
		tempConfigDir := fmt.Sprintf("%s/tmp/config", tmpDir)

		confFile := fmt.Sprintf("%s/config.json", tempConfigDir)
		rulesFile := fmt.Sprintf("%s/rules.json", tempConfigDir)
		lFp := fmt.Sprintf("%s/app.log", tempLogsDir)

		mkDirForTest(t, tempConfigDir)
		mkDirForTest(t, tempLogsDir)

		rulesJsonFp := "testData/rules3.json"
		cpFileForTest(t, rulesJsonFp, rulesFile)

		p := ParsedConf{
			RulesFp:      rulesFile,
			LogFp:        lFp,
			Port:         "9050",
			Compression:  true,
			SizeToRotate: "5MB",
		}
		jsonString, _ := json.Marshal(p)
		writeForTest(t, confFile, jsonString)

		_, err := c.LoadMainConfigFile(confFile)
		assert.Equal(t, err, nil)

		err = c.LoadMappingFile(p.RulesFp)
		assert.Equal(t, err, nil)

		l, err := newFileLogger(p.LogFp, p.SizeToRotate, p.Compression)
		assert.NotEmpty(t, l)
		assert.Equal(t, err, nil)

		isEmpty := isFileEmpty(t, lFp)
		assert.Equal(t, isEmpty, true)

		app := newTestApp(t, c, l)
		app.Routes()
		ts := newTestServer(t, app.Mux)
		defer ts.Close()

		code, _, body := ts.get(t, "/x/fuckoff?go-get=1")
		assert.Equal(t, code, http.StatusOK)
		expected, _ := getDefaultHtml()
		assert.Equal(t, string(body), string(expected))

		isEmpty = isFileEmpty(t, lFp)
		assert.Equal(t, isEmpty, false)

		t.Logf("Printing returned html => %s", string(body))
	})
}