Merge pull request #66 from BarbarossaTM/testing/contributing_ASNs

Add more tests for contributing ASNs.

routingtable/contributing_asn_list_test.go

@@ -24,7 +24,7 @@ func TestFancy(t *testing.T) {
 		{
 			runCmd: func() { c.Add(41981) },
 			expect: func() bool { return c.IsContributingASN(41981) },
-			msg:    "AS41981 should be contributing",
+			msg:    "AS41981 should be contributing.",
 		},
 		{
 			runCmd: func() { c.Remove(41981) },
@@ -36,30 +36,49 @@ func TestFancy(t *testing.T) {
 		{
 			runCmd: func() { c.Add(41981) },
 			expect: func() bool { return c.IsContributingASN(41981) },
-			msg:    "AS41981 should be contributing",
+			msg:    "AS41981 should be contributing.",
 		},
 		{
 			runCmd: func() { c.Add(201701) },
 			expect: func() bool { return c.IsContributingASN(41981) },
-			msg:    "AS201701 should be contributing",
+			msg:    "AS201701 should be contributing.",
 		},
 
 		// Add AS41981 2nd time
 		{
 			runCmd: func() { c.Add(41981) },
 			expect: func() bool { return c.IsContributingASN(41981) },
-			msg:    "AS41981 should be still contributing",
+			msg:    "AS41981 should be still contributing.",
 		},
 		{
 			runCmd: func() {},
 			expect: func() bool { return c.contributingASNs[0].asn == 41981 },
-			msg:    "AS41981 is first ASN in list",
+			msg:    "AS41981 is first ASN in list.",
 		},
 		{
 			runCmd: func() { fmt.Printf("%+v", c.contributingASNs) },
 			expect: func() bool { return c.contributingASNs[0].count == 2 },
 			msg:    "AS41981 should be present twice.",
 		},
+
+		// Remove 2nd AS41981
+		{
+			runCmd: func() { c.Remove(41981) },
+			expect: func() bool { return c.IsContributingASN(41981) },
+			msg:    "AS41981 should still be contributing.",
+		},
+		{
+			runCmd: func() {},
+			expect: func() bool { return c.contributingASNs[0].count == 1 },
+			msg:    "S41981 should be present once.",
+		},
+
+		// Remove AS201701
+		{
+			runCmd: func() { c.Remove(201701) },
+			expect: func() bool { return !c.IsContributingASN(201701) },
+			msg:    "AS201701 shouldn't be contributing no more.",
+		},
 	}
 
 	for i, test := range tests {