Adds more tests for the registry

seantis · Feb 6, 2015 · b5c75c4 · b5c75c4
1 parent a4d7b6c
commit b5c75c4
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/libres/tests/test_registry.py b/libres/tests/test_registry.py
@@ -45,6 +45,18 @@ def test_autocreate():
     assert r.get_context('yo') is ctx
 
 
+def test_assert_existence():
+    r = Registry()
+
+    with pytest.raises(errors.UnknownContext):
+        r.assert_exists('foo')
+
+    r.register_context('foo')
+
+    with pytest.raises(errors.ContextAlreadyExists):
+        r.assert_does_not_exist('foo')
+
+
 def test_replace():
     r = Registry()