Add tests for AddByte and TestByte

Author: korthaj

filter_test.go

@@ -58,6 +58,60 @@ func TestFilter(t *testing.T) {
 	}
 }
 
+func TestFilterByte(t *testing.T) {
+	s1 := []byte("asöldkgjaösldkgaösldkasldgjkaösldkgjöasgkdjg")
+	s2 := []byte("elasödlnkgaölsdkfgaölsdkjfaölsdkgaölskgnaösl")
+	s3 := []byte("aölsdgkaösldkgaösldkgjaölsdkjgaölsdkgjaösldk")
+	for n := 0; n < 100; n++ {
+		for p := 1; p <= 128; p *= 2 {
+			filter := New(n, p)
+			member := filter.TestByte(s1)
+			if member {
+				t.Errorf("TestByte(s1) = %v; want false\n", member)
+			}
+			count := filter.Count()
+			if count != 0 {
+				t.Errorf("Count() = %d; want 0\n", count)
+			}
+
+			member = filter.AddByte(s1)
+			if member {
+				t.Errorf("AddByte(s1) = %v; want false\n", member)
+			}
+			count = filter.Count()
+			if count != 1 {
+				t.Errorf("Count() = %d; want 1\n", count)
+			}
+			member = filter.TestByte(s1)
+			if !member {
+				t.Errorf("TestByte(s1) = %v; want true\n", member)
+			}
+			member = filter.TestByte(s2)
+			if member {
+				t.Errorf("TestByte(s2) = %v; want false\n", member)
+			}
+
+			member = filter.AddByte(s1)
+			if !member {
+				t.Errorf("AddByte(s1) = %v; want true\n", member)
+			}
+			count = filter.Count()
+			if count != 1 {
+				t.Errorf("Count() = %d; want 1\n", count)
+			}
+
+			member = filter.AddByte(s3)
+			if member {
+				t.Errorf("AddByte(s3) = %v; want false\n", member)
+			}
+			count = filter.Count()
+			if count != 2 {
+				t.Errorf("Count() = %d; want 2\n", count)
+			}
+		}
+	}
+}
+
 func TestUnion(t *testing.T) {
 	s1 := "asöldkgjaösldkgaösldkasldgjkaösldkgjöasgkdjg"
 	s2 := "elasödlnkgaölsdkfgaölsdkjfaölsdkgaölskgnaösl"