1
|
|
|
"""Test kytos.core.helpers module.""" |
2
|
|
|
from unittest import TestCase |
3
|
|
|
from unittest.mock import patch |
4
|
|
|
|
5
|
|
|
from kytos.core.helpers import get_time, run_on_thread |
6
|
|
|
|
7
|
|
|
|
8
|
|
|
class TestHelpers(TestCase): |
9
|
|
|
"""Test the helpers methods.""" |
10
|
|
|
|
11
|
|
|
@staticmethod |
12
|
|
|
@patch('kytos.core.helpers.Thread') |
13
|
|
|
def test_run_on_thread(mock_thread): |
14
|
|
|
"""Test run_on_thread decorator.""" |
15
|
|
|
|
16
|
|
|
@run_on_thread |
17
|
|
|
def test(): |
18
|
|
|
pass |
19
|
|
|
|
20
|
|
|
test() |
21
|
|
|
|
22
|
|
|
mock_thread.return_value.start.assert_called() |
23
|
|
|
|
24
|
|
|
def test_get_time__str(self): |
25
|
|
|
"""Test get_time method passing a string as parameter.""" |
26
|
|
|
date = get_time("2000-01-01T00:30:00") |
27
|
|
|
|
28
|
|
|
self.assertEqual(date.year, 2000) |
29
|
|
|
self.assertEqual(date.month, 1) |
30
|
|
|
self.assertEqual(date.day, 1) |
31
|
|
|
self.assertEqual(date.hour, 0) |
32
|
|
|
self.assertEqual(date.minute, 30) |
33
|
|
|
self.assertEqual(date.second, 0) |
34
|
|
|
|
35
|
|
|
def test_get_time__dict(self): |
36
|
|
|
"""Test get_time method passing a dict as parameter.""" |
37
|
|
|
date = get_time({"year": 2000, |
38
|
|
|
"month": 1, |
39
|
|
|
"day": 1, |
40
|
|
|
"hour": 00, |
41
|
|
|
"minute": 30, |
42
|
|
|
"second": 00}) |
43
|
|
|
|
44
|
|
|
self.assertEqual(date.year, 2000) |
45
|
|
|
self.assertEqual(date.month, 1) |
46
|
|
|
self.assertEqual(date.day, 1) |
47
|
|
|
self.assertEqual(date.hour, 0) |
48
|
|
|
self.assertEqual(date.minute, 30) |
49
|
|
|
self.assertEqual(date.second, 0) |
50
|
|
|
|
51
|
|
|
def test_get_time__none(self): |
52
|
|
|
"""Test get_time method by not passing a parameter.""" |
53
|
|
|
date = get_time() |
54
|
|
|
|
55
|
|
|
self.assertIsNone(date) |
56
|
|
|
|