a0246ccee2
Includes a few minor fixes: - frontend: support setting port for WHOIS server - proxy: fix handling of very long lines - proxy: refactor IP allowlist logic, parse allow IP list at startup
85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"path"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func readDataFile(t *testing.T, filename string) string {
|
|
_, sourceName, _, _ := runtime.Caller(0)
|
|
projectRoot := path.Join(path.Dir(sourceName), "..")
|
|
dir := path.Join(projectRoot, filename)
|
|
|
|
data, err := ioutil.ReadFile(dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return string(data)
|
|
}
|
|
|
|
func TestBirdRouteToGraphvizXSS(t *testing.T) {
|
|
setting.dnsInterface = ""
|
|
|
|
// Don't change formatting of the following strings!
|
|
|
|
fakeResult := `<script>alert("evil!")</script>`
|
|
|
|
result := birdRouteToGraphviz([]string{
|
|
"alpha",
|
|
}, []string{
|
|
fakeResult,
|
|
}, fakeResult)
|
|
|
|
if strings.Contains(result, "<script>") {
|
|
t.Errorf("XSS injection succeeded: %s", result)
|
|
}
|
|
}
|
|
|
|
func TestBirdRouteToGraph(t *testing.T) {
|
|
setting.dnsInterface = ""
|
|
|
|
input := readDataFile(t, "frontend/test_data/bgpmap_case1.txt")
|
|
result := birdRouteToGraph([]string{"node"}, []string{input}, "target")
|
|
|
|
// Source node must exist
|
|
if result.GetPoint("node") == nil {
|
|
t.Error("Result doesn't contain point node")
|
|
}
|
|
// Last hop must exist
|
|
if result.GetPoint("4242423914") == nil {
|
|
t.Error("Result doesn't contain point 4242423914")
|
|
}
|
|
// Destination must exist
|
|
if result.GetPoint("target") == nil {
|
|
t.Error("Result doesn't contain point target")
|
|
}
|
|
|
|
// Verify that a few paths exist
|
|
if result.GetEdge("node", "4242423914") == nil {
|
|
t.Error("Result doesn't contain edge from node to 4242423914")
|
|
}
|
|
if result.GetEdge("node", "4242422688") == nil {
|
|
t.Error("Result doesn't contain edge from node to 4242422688")
|
|
}
|
|
if result.GetEdge("4242422688", "4242423914") == nil {
|
|
t.Error("Result doesn't contain edge from 4242422688 to 4242423914")
|
|
}
|
|
if result.GetEdge("4242423914", "target") == nil {
|
|
t.Error("Result doesn't contain edge from 4242423914 to target")
|
|
}
|
|
}
|
|
|
|
func TestBirdRouteToGraphviz(t *testing.T) {
|
|
setting.dnsInterface = ""
|
|
|
|
input := readDataFile(t, "frontend/test_data/bgpmap_case1.txt")
|
|
result := birdRouteToGraphviz([]string{"node"}, []string{input}, "target")
|
|
|
|
if !strings.Contains(result, "digraph {") {
|
|
t.Error("Response is not Graphviz data")
|
|
}
|
|
}
|