From: Brian Wellington <bwelling@xbill.org>
Date: Thu, 2 Apr 2020 19:54:29 +0000 (-0700)
Subject: Update Set tests.
X-Git-Tag: v2.0.0rc1~307
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=474eebc175817657db2af844db23abcf814cbe8f;p=thirdparty%2Fdnspython.git

Update Set tests.

Add tests to cover untested functionality, and update existing tests to
actually use the fact that these sets are ordered.
---

diff --git a/tests/test_set.py b/tests/test_set.py
index 08d3997f..d3fa5ea1 100644
--- a/tests/test_set.py
+++ b/tests/test_set.py
@@ -22,7 +22,7 @@ import dns.set
 # for convenience
 S = dns.set.Set
 
-class SimpleSetTestCase(unittest.TestCase):
+class SetTestCase(unittest.TestCase):
 
     def testLen1(self):
         s1 = S()
@@ -194,17 +194,33 @@ class SimpleSetTestCase(unittest.TestCase):
     def testDelitem(self):
         s1 = S([1, 2, 3])
         del s1[0]
-        i1 = s1[0]
-        i2 = s1[1]
-        self.assertTrue(i1 != i2)
-        self.assertTrue(i1 == 1 or i1 == 2 or i1 == 3)
-        self.assertTrue(i2 == 1 or i2 == 2 or i2 == 3)
+        self.assertEqual(list(s1), [2, 3])
 
     def testDelslice(self):
         s1 = S([1, 2, 3])
         del s1[0:2]
-        i1 = s1[0]
-        self.assertTrue(i1 == 1 or i1 == 2 or i1 == 3)
+        self.assertEqual(list(s1), [3])
+
+    def testRemoveNonexistent(self):
+        s1 = S([1, 2, 3])
+        s2 = S([1, 2, 3])
+        with self.assertRaises(ValueError):
+            s1.remove(4)
+        self.assertEqual(s1, s2)
+
+    def testDiscardNonexistent(self):
+        s1 = S([1, 2, 3])
+        s2 = S([1, 2, 3])
+        s1.discard(4)
+        self.assertEqual(s1, s2)
+
+    def testCopy(self):
+        s1 = S([1, 2, 3])
+        s2 = s1.copy()
+        s1.remove(1)
+        self.assertNotEqual(s1, s2)
+        s1.add(1)
+        self.assertEqual(s1, s2)
 
 if __name__ == '__main__':
     unittest.main()