more unittest

Author: yachang

scamper/scamper_test.go

@@ -121,11 +121,59 @@ func TestTraceWritesMeta(t *testing.T) {
 	b, err := ioutil.ReadFile(tempdir + "/2019/04/01/20190401T034551Z_" + prefix.UnsafeString() + "_0000000000000001.jsonl")
 	rtx.Must(err, "Could not read file")
 
-	type metadata struct {
-		UUID                    string
-		TracerouteCallerVersion string
+	m := Metadata{}
+	lines := strings.Split(string(b), "\n")
+	if len(lines) < 2 {
+		t.Error("Not enough lines in", lines)
+	}
+	rtx.Must(json.Unmarshal([]byte(lines[0]), &m), "Could not unmarshal")
+
+	uuidChunks := strings.Split(m.UUID, "_")
+
+	if uuidChunks[len(uuidChunks)-1] != "0000000000000001" {
+		t.Error("Bad uuid:", m.UUID)
+	}
+
+	if m.TracerouteCallerVersion != "Fake Version" {
+		t.Error("Bad traceroute caller version:", m.TracerouteCallerVersion)
 	}
-	m := metadata{}
+}
+
+func TestCreateCacheTest(t *testing.T) {
+	tempdir, err := ioutil.TempDir("", "TestCachedTrace")
+	rtx.Must(err, "Could not create tempdir")
+	defer os.RemoveAll(tempdir)
+
+	// Temporarily set the hostname to a value for testing.
+	defer func(oldHn string) {
+		hostname = oldHn
+	}(hostname)
+	hostname = "testhostname"
+
+	d := Daemon{
+		AttachBinary:     "echo",
+		Warts2JSONBinary: "cat",
+		OutputPath:       tempdir,
+	}
+
+	c := connection.Connection{
+		Cookie:   "1",
+		RemoteIP: "1.2.3.4",
+	}
+
+	faketime := time.Date(2019, time.April, 1, 3, 45, 51, 0, time.UTC)
+	prometheusx.GitShortCommit = "Fake Version"
+	cachedTest := `{"UUID": "ndt-plh7v_1566050090_000000000004D64D"}
+	{"type":"cycle-start", "list_name":"/tmp/scamperctrl:51811", "id":1, "hostname":"ndt-plh7v", "start_time":1566691298}
+	{"type":"tracelb", "version":"0.1", "userid":0, "method":"icmp-echo", "src":"::ffff:180.87.97.101", "dst":"::ffff:1.47.236.62", "start":{"sec":1566691298, "usec":476221, "ftime":"2019-08-25 00:01:38"}, "probe_size":60, "firsthop":1, "attempts":3, "confidence":95, "tos":0, "gaplimit":3, "wait_timeout":5, "wait_probe":250, "probec":0, "probec_max":3000, "nodec":0, "linkc":0}
+	{"type":"cycle-stop", "list_name":"/tmp/scamperctrl:51811", "id":1, "hostname":"ndt-plh7v", "stop_time":1566691298}`
+	d.CreateCacheTest(c, faketime, cachedTest)
+
+	// Unmarshal the first line of the output file.
+	b, err := ioutil.ReadFile(tempdir + "/2019/04/01/20190401T034551Z_" + prefix.UnsafeString() + "_0000000000000001.jsonl")
+	rtx.Must(err, "Could not read file")
+
+	m := Metadata{}
 	lines := strings.Split(string(b), "\n")
 	if len(lines) < 2 {
 		t.Error("Not enough lines in", lines)
@@ -141,6 +189,14 @@ func TestTraceWritesMeta(t *testing.T) {
 	if m.TracerouteCallerVersion != "Fake Version" {
 		t.Error("Bad traceroute caller version:", m.TracerouteCallerVersion)
 	}
+
+	if m.CachedResult != true {
+		t.Error("Bad traceroute CachedResult value:", m.CachedResult)
+	}
+
+	if m.CachedUUID != "ndt-plh7v_1566050090_000000000004D64D" {
+		t.Error("Bad traceroute CachedUUID value:", m.CachedUUID)
+	}
 }
 
 func TestRecovery(t *testing.T) {