Add a Not comparator to compose with other comparators.

Patch by: Benoit Sigoure <benoits@google.com>

1 files changed, 32 insertions, 0 deletions

diff --git a/mox_test.py b/mox_test.py
index be35bae..5f00f0c 100755
--- a/mox_test.py
+++ b/mox_test.py
@@ -159,6 +159,38 @@ class InTest(unittest.TestCase):
     self.assert_(mox.In("test") == {"test" : "module"})
 
 
+class NotTest(unittest.TestCase):
+  """Test Not correctly identifies False predicates."""
+
+  def testItemInList(self):
+    """Should return True if the item is NOT in the list."""
+    self.assert_(mox.Not(mox.In(42)) == [1, 2, 3])
+
+  def testKeyInDict(self):
+    """Should return True if the item is NOT a key in a dict."""
+    self.assert_(mox.Not(mox.In("foo")) == {"key" : 42})
+
+  def testInvalidKeyWithNot(self):
+    """Should return False if they key is NOT in the dict."""
+    self.assert_(mox.Not(mox.ContainsKeyValue("qux", 1)) == {"key": 2})
+
+
+class NotTest(unittest.TestCase):
+  """Test Not correctly identifies False predicates."""
+
+  def testItemInList(self):
+    """Should return True if the item is NOT in the list."""
+    self.assert_(mox.Not(mox.In(42)) == [1, 2, 3])
+
+  def testKeyInDict(self):
+    """Should return True if the item is NOT a key in a dict."""
+    self.assert_(mox.Not(mox.In("foo")) == {"key" : 42})
+
+  def testInvalidKeyWithNot(self):
+    """Should return False if they key is NOT in the dict."""
+    self.assert_(mox.Not(mox.ContainsKeyValue("qux", 1)) == {"key": 2})
+
+
 class StrContainsTest(unittest.TestCase):
   """Test StrContains correctly checks for substring occurrence of a parameter.
   """