dnscontrol/js/js_test.go

package js

import (
	"encoding/json"
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"
	"unicode"

	"github.com/StackExchange/dnscontrol/models"
)

const (
	testDir  = "js/parse_tests"
	errorDir = "js/error_tests"
)

func init() {
	os.Chdir("..") // go up a directory so we helpers.js is in a consistent place.
}

func TestParsedFiles(t *testing.T) {
	files, err := ioutil.ReadDir(testDir)
	if err != nil {
		t.Fatal(err)
	}
	for _, f := range files {
		//run all js files that start with a number. Skip others.
		if filepath.Ext(f.Name()) != ".js" || !unicode.IsNumber(rune(f.Name()[0])) {
			continue
		}
		t.Log(f.Name(), "------")
		content, err := ioutil.ReadFile(filepath.Join(testDir, f.Name()))
		if err != nil {
			t.Fatal(err)
		}
		conf, err := ExecuteJavascript(string(content), true)
		if err != nil {
			t.Fatal(err)
		}
		actualJson, err := json.MarshalIndent(conf, "", "  ")
		if err != nil {
			t.Fatal(err)
		}
		expectedFile := filepath.Join(testDir, f.Name()[:len(f.Name())-3]+".json")
		expectedData, err := ioutil.ReadFile(expectedFile)
		if err != nil {
			t.Fatal(err)
		}
		conf = &models.DNSConfig{}
		err = json.Unmarshal(expectedData, conf)
		if err != nil {
			t.Fatal(err)
		}
		expectedJson, err := json.MarshalIndent(conf, "", "  ")
		if err != nil {
			t.Fatal(err)
		}
		if string(expectedJson) != string(actualJson) {
			t.Error("Expected and actual json don't match")
			t.Log("Expected:", string(expectedJson))
			t.Log("Actual:", string(actualJson))
			t.FailNow()
		}
	}
}

func TestErrors(t *testing.T) {
	tests := []struct{ desc, text string }{
		{"old dsp style", `D("foo.com","reg","dsp")`},
		{"MX no priority", `D("foo.com","reg",MX("@","test."))`},
		{"MX reversed", `D("foo.com","reg",MX("@","test.", 5))`},
		{"Bad cidr", `D(reverse("foo.com"), "reg")`},
	}
	for _, tst := range tests {
		t.Run(tst.desc, func(t *testing.T) {
			if _, err := ExecuteJavascript(tst.text, true); err == nil {
				t.Fatal("Expected error but found none")
			}
		})

	}
}