-
Notifications
You must be signed in to change notification settings - Fork 61
/
test_utils.py
59 lines (45 loc) · 2.14 KB
/
test_utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
test_django-watchman
------------
Tests for `django-watchman` decorators module.
"""
from __future__ import unicode_literals
import unittest
from watchman.utils import get_checks
class TestWatchman(unittest.TestCase):
def assertListsEqual(self, l1, l2):
try:
# Python 3.4
self.assertCountEqual(l1, l2)
except AttributeError:
# Python 2.7
self.assertItemsEqual(l1, l2)
def setUp(self):
pass
def test_get_checks_returns_all_available_checks_by_default(self):
checks = [check.__name__ for check in get_checks()]
expected_checks = ['caches_status', 'email_status', 'databases_status']
self.assertListsEqual(checks, expected_checks)
def test_get_checks_with_check_list_returns_union(self):
check_list = ['watchman.checks.caches_status']
checks = [check.__name__ for check in get_checks(check_list=check_list)]
expected_checks = ['caches_status']
self.assertListsEqual(checks, expected_checks)
def test_get_checks_with_skip_list_returns_difference(self):
skip_list = ['watchman.checks.caches_status']
checks = [check.__name__ for check in get_checks(skip_list=skip_list)]
expected_checks = ['databases_status', 'email_status']
self.assertListsEqual(checks, expected_checks)
def test_get_checks_with_matching_check_and_skip_list_returns_empty_list(self):
check_list, skip_list = ['watchman.checks.caches_status'], ['watchman.checks.caches_status']
checks = [check.__name__ for check in get_checks(check_list=check_list, skip_list=skip_list)]
expected_checks = []
self.assertListsEqual(checks, expected_checks)
def test_get_checks_with_check_and_skip_list(self):
check_list = ['watchman.checks.caches_status', 'watchman.checks.databases_status']
skip_list = ['watchman.checks.caches_status']
checks = [check.__name__ for check in get_checks(check_list=check_list, skip_list=skip_list)]
expected_checks = ['databases_status']
self.assertListsEqual(checks, expected_checks)