瀏覽代碼

Improve `test_post_process_urls_adds_decorator`

Use function as the return value of `get_url_decorator`. This change
needed because `get_url_decorator` method of `OscarConfigMixin` class
returns function or `None` and not string.
master
Basil Dubyk 5 年之前
父節點
當前提交
7c5f745919
共有 1 個文件被更改,包括 5 次插入2 次删除
  1. 5
    2
      tests/unit/core/test_application.py

+ 5
- 2
tests/unit/core/test_application.py 查看文件

30
         mock_permissions_required.assert_called_once_with('is_staff', login_url=None)
30
         mock_permissions_required.assert_called_once_with('is_staff', login_url=None)
31
 
31
 
32
     def test_post_process_urls_adds_decorator(self):
32
     def test_post_process_urls_adds_decorator(self):
33
+        def fake_callback():
34
+            pass
35
+
33
         fake_decorator = mock.Mock()
36
         fake_decorator = mock.Mock()
34
-        fake_decorator.return_value = 'fake_callback'
37
+        fake_decorator.return_value = fake_callback
35
 
38
 
36
         self.myapp.get_url_decorator = mock.Mock()
39
         self.myapp.get_url_decorator = mock.Mock()
37
         self.myapp.get_url_decorator.return_value = fake_decorator
40
         self.myapp.get_url_decorator.return_value = fake_decorator
40
         processed_patterns = self.myapp.post_process_urls([pattern])
43
         processed_patterns = self.myapp.post_process_urls([pattern])
41
 
44
 
42
         self.myapp.get_url_decorator.assert_called_once_with(pattern)
45
         self.myapp.get_url_decorator.assert_called_once_with(pattern)
43
-        self.assertEqual(processed_patterns[0].callback, 'fake_callback')
46
+        self.assertEqual(processed_patterns[0].callback, fake_callback)

Loading…
取消
儲存