tests...

transcode-de · May 12, 2015 · 46d5bfd · 46d5bfd
1 parent d78a6d8
commit 46d5bfd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/test_element.py b/tests/test_element.py
@@ -28,9 +28,9 @@ def test_get_dynamic_url(resource):
     class DynamicURLElement(element.Element):
         id = '23'
 
-    with pytest.raises(AssertionError) as excinfo:
+    with pytest.raises(AttributeError) as excinfo:
         DynamicURLElement(resource, {}).get_url()
-    assert excinfo.value.msg == 'You must define an id_attribute attribute at MoviesResource.'
+    assert str(excinfo.value) == 'You must define an id_attribute attribute at MoviesResource.'
     resource.id_attribute = 'id'
     element_obj = DynamicURLElement(resource, {})
     assert element_obj.get_url() == '/'.join([resource.get_url(), element_obj.id])

diff --git a/tests/test_resource.py b/tests/test_resource.py
@@ -13,18 +13,18 @@
 
 
 def test_base_url_required():
-    with pytest.raises(AssertionError) as excinfo:
+    with pytest.raises(AttributeError) as excinfo:
         durga.Resource()
-    assert excinfo.value.msg == 'You must define a "base_url" attribute.'
+    assert str(excinfo.value) == 'You must define a "base_url" attribute.'
 
 
 def test_path_required():
     class TestResource(durga.Resource):
         base_url = 'https://api.example.com'
 
-    with pytest.raises(AssertionError) as excinfo:
+    with pytest.raises(AttributeError) as excinfo:
         TestResource()
-    assert excinfo.value.msg == 'You must define a "path" attribute.'
+    assert str(excinfo.value) == 'You must define a "path" attribute.'
 
 
 def test_get_url(resource):