Fix tests.

socialwifi · May 9, 2017 · a757313 · a757313
1 parent 4e698fd
commit a757313
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_application.py b/tests/test_application.py
@@ -31,7 +31,7 @@
 
 @mock.patch('dila.data.add_resource')
 def test_add_resource(add_resource):
-    dila.application.resources.add_resource('Next')
+    dila.application.add_resource('Next')
     assert add_resource.mock_calls == [
         mock.call(
             'Next',
@@ -53,6 +53,14 @@ def test_get_resources_evaluates_iterator(get_resources):
     assert result == []
 
 
+@mock.patch('dila.data.get_resource')
+def test_get_resource(get_resource):
+    get_resource.return_value = 'data_result'
+    result = application.get_resource('pk')
+    get_resource.assert_called_with('pk')
+    assert result == 'data_result'
+
+
 @mock.patch('dila.data.get_resource')
 def test_get_translated_string(get_resource):
     get_resource.return_value = 'data_result'