cover all cases

joshsagredo · Jan 3, 2022 · cb6d787 · cb6d787
1 parent 1deb43b
commit cb6d787
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/internal/raw/raw_test.go b/internal/raw/raw_test.go
@@ -21,9 +21,19 @@ func TestStartFlooding(t *testing.T) {
 		srcIp, dstIp                    string
 		srcMacAddr, dstMacAddr          []byte
 	}{
-		{"500byte_syn", "syn", 500, srcPorts[rand.Intn(len(srcPorts))], 443, 100,
-			srcIps[rand.Intn(len(srcIps))], "213.238.175.187",
+		{"100byte_syn", "syn", 10, srcPorts[rand.Intn(len(srcPorts))],
+			443, 10, srcIps[rand.Intn(len(srcIps))], "213.238.175.187",
 			macAddrs[rand.Intn(len(macAddrs))], macAddrs[rand.Intn(len(macAddrs))]},
+		{
+			"100byte_ack", "ack", 10, srcPorts[rand.Intn(len(srcPorts))],
+			443, 10, srcIps[rand.Intn(len(srcIps))], "213.238.175.187",
+			macAddrs[rand.Intn(len(macAddrs))], macAddrs[rand.Intn(len(macAddrs))],
+		},
+		{
+			"100byte_synack", "synAck", 10, srcPorts[rand.Intn(len(srcPorts))],
+			443, 10, srcIps[rand.Intn(len(srcIps))], "213.238.175.187",
+			macAddrs[rand.Intn(len(macAddrs))], macAddrs[rand.Intn(len(macAddrs))],
+		},
 	}
 
 	for _, tc := range cases {