Add unit test for TestFilterSingleIPv6

parser/filter_test.go

@@ -208,7 +208,21 @@ func TestFilterSingleIP(t *testing.T) {
 	}
 }
 
-// In alwaysIncluded, just leave it as an empty slice, replace
-// util.ParseSubnets with []*net.IPNet
-// In neverIncluded, include the IPv6 address without CIDR.
-// Test case will be modeled after the "never" test case
+func TestFilterSingleIPv6(t *testing.T) {
+	fsTest := filter{
+		// purposely omitting internal subnet definition
+		alwaysIncluded: []*net.IPNet{},
+		neverIncluded: util.ParseSubnets([]string{"::1"}),
+	}
+
+	never:= "::1"
+
+	testCases := []testCaseSingleIP{
+		{never, true, "NeverInclude IP should be filtered"},
+	}
+
+	for _, test := range testCases {
+		output := fsTest.filterSingleIP(net.ParseIP(test.ip))
+		assert.Equal(t, test.out, output, test.msg)
+	}
+}

util/ip.go

@@ -34,7 +34,7 @@ func ParseSubnets(subnets []string) []*net.IPNet {
 		// Try to parse out CIDR range
 		_, block, err := net.ParseCIDR(entry)
 
-		// If there was an error, check if entry was an IP, not a range
+		// If there was an error, check if entry was an IP
 		if err != nil {
 			ipAddr := net.ParseIP(entry)
 			if ipAddr == nil {
@@ -45,9 +45,9 @@ func ParseSubnets(subnets []string) []*net.IPNet {
 			// Check if it's an IPv4 or IPv6 address and append the appropriate subnet mask
 			var subnetMask string
 			if ipAddr.To4() != nil {
-				subnetMask = "/32" // IPv4
+				subnetMask = "/32"
 			} else {
-				subnetMask = "/128" // IPv6
+				subnetMask = "/128"
 			}
 
 			// Append the subnet mask and parse as a CIDR range