Skip to content

Commit 5eb7468

Browse files
FernandoCelmerFernandoCelmer
committed
❤️ TEST: Updated unit tests
1 parent 3c6c7f0 commit 5eb7468

File tree

3 files changed

+118
-43
lines changed

3 files changed

+118
-43
lines changed

‎tests/core/test_config.py‎

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
"""Test of config"""
2+
3+
import unittest
4+
5+
from dotflow.core.config import Config, ConfigInstance
6+
from dotflow.abc.storage import Storage
7+
from dotflow.abc.notify import Notify
8+
from dotflow.abc.logs import Logs
9+
10+
11+
class TestConfigInstance(unittest.TestCase):
12+
13+
def setUp(self):
14+
self.instance = ConfigInstance()
15+
16+
def test_instantiating_config_instance_class(self):
17+
self.assertIsNone(self.instance._storage)
18+
self.assertIsNone(self.instance._notify)
19+
self.assertIsNone(self.instance._logs)
20+
21+
22+
class TestConfig(unittest.TestCase):
23+
24+
def setUp(self):
25+
self.config = Config()
26+
self.index = 0
27+
28+
def test_instantiating_config_class(self):
29+
self.assertIsInstance(self.config.storage, Storage)
30+
self.assertIsInstance(self.config.notify, Notify)
31+
self.assertIsInstance(self.config.logs, list)
32+
self.assertIsInstance(self.config.logs[self.index], Logs)
33+
34+
def test_config_storage_object(self):
35+
self.assertIsNotNone(self.config.storage)
36+
self.assertFalse(callable(self.config.storage))
37+
38+
def test_config_notify_object(self):
39+
self.assertIsNotNone(self.config.notify)
40+
self.assertFalse(callable(self.config.notify))
41+
42+
def test_config_logs_object(self):
43+
self.assertIsNotNone(self.config.logs)
44+
self.assertFalse(callable(self.config.logs[self.index]))

‎tests/core/test_dotflow.py‎

Lines changed: 69 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -1,63 +1,103 @@
1-
"""Test context of workflow"""
1+
"""Test of dotflow"""
22

33
import unittest
44

5+
from uuid import UUID
6+
7+
from dotflow.core.config import Config
58
from dotflow.core.workflow import Manager
69
from dotflow.core.context import Context
710
from dotflow.core.task import Task, TaskBuilder
8-
from dotflow.core.dotflow import DotFlow
11+
from dotflow.core.dotflow import DotFlow, DotflowInstance
912
from dotflow.core.types.status import StatusTaskType
1013

1114
from tests.mocks import action_step
1215

1316

17+
class TestDotflowInstance(unittest.TestCase):
18+
19+
def setUp(self):
20+
self.instance = DotflowInstance()
21+
22+
def test_instantiating_dotflow_instance_class(self):
23+
self.assertIsNone(self.instance._config)
24+
self.assertIsNone(self.instance._workflow_id)
25+
self.assertIsNone(self.instance._task)
26+
self.assertIsNone(self.instance._add)
27+
self.assertIsNone(self.instance._start)
28+
29+
1430
class TestDotFlow(unittest.TestCase):
1531

1632
def setUp(self):
17-
self.workflow = DotFlow()
18-
self.workflow.add(step=action_step)
33+
self.dotflow = DotFlow()
34+
self.dotflow.add(step=action_step)
35+
self.index = 0
36+
self.size = 1
1937

2038
def test_instantiating_dotflow_class(self):
21-
self.assertIsInstance(self.workflow.task, TaskBuilder)
22-
self.assertIsInstance(self.workflow.start(), Manager)
39+
self.assertIsInstance(self.dotflow.config, Config)
40+
self.assertIsInstance(self.dotflow.workflow_id, UUID)
41+
self.assertIsInstance(self.dotflow.task, TaskBuilder)
42+
self.assertIsInstance(self.dotflow.start(), Manager)
2343

2444
def test_result_task_with_start(self):
25-
self.workflow.start()
26-
result = self.workflow.result_task()
45+
self.dotflow.start()
46+
result = self.dotflow.result_task()
2747

28-
self.assertEqual(len(result), 1)
29-
self.assertIsInstance(result[0], Task)
30-
self.assertEqual(result[0].status, StatusTaskType.COMPLETED)
48+
self.assertEqual(len(result), self.size)
49+
self.assertIsInstance(result[self.index], Task)
50+
self.assertEqual(result[self.index].status, StatusTaskType.COMPLETED)
3151

3252
def test_result_context_with_start(self):
33-
self.workflow.start()
34-
result = self.workflow.result_context()
53+
self.dotflow.start()
54+
result = self.dotflow.result_context()
3555

36-
self.assertEqual(len(result), 1)
37-
self.assertIsInstance(result[0], Context)
56+
self.assertEqual(len(result), self.size)
57+
self.assertIsInstance(result[self.index], Context)
3858

3959
def test_result_storage_with_start(self):
40-
self.workflow.start()
41-
result = self.workflow.result_storage()
60+
self.dotflow.start()
61+
result = self.dotflow.result_storage()
4262

43-
self.assertEqual(len(result), 1)
44-
self.assertEqual(result[0], {"foo": "bar"})
63+
self.assertEqual(len(result), self.size)
64+
self.assertEqual(result[self.index], {"foo": "bar"})
4565

4666
def test_result_task_without_start(self):
47-
result = self.workflow.result_task()
67+
result = self.dotflow.result_task()
4868

49-
self.assertEqual(len(result), 1)
50-
self.assertIsInstance(result[0], Task)
51-
self.assertEqual(result[0].status, StatusTaskType.NOT_STARTED)
69+
self.assertEqual(len(result), self.size)
70+
self.assertIsInstance(result[self.index], Task)
71+
self.assertEqual(result[self.index].status, StatusTaskType.NOT_STARTED)
5272

5373
def test_result_context_without_start(self):
54-
result = self.workflow.result_context()
74+
result = self.dotflow.result_context()
5575

56-
self.assertEqual(len(result), 1)
57-
self.assertIsInstance(result[0], Context)
76+
self.assertEqual(len(result), self.size)
77+
self.assertIsInstance(result[self.index], Context)
5878

5979
def test_result_storage_without_start(self):
60-
result = self.workflow.result_storage()
80+
result = self.dotflow.result_storage()
81+
82+
self.assertEqual(len(result), self.size)
83+
self.assertEqual(result[self.index], None)
84+
85+
def test_dotflow_config_object(self):
86+
self.assertIsNotNone(self.dotflow.config)
87+
self.assertFalse(callable(self.dotflow.config))
88+
89+
def test_dotflow_workflow_id_object(self):
90+
self.assertIsNotNone(self.dotflow.workflow_id)
91+
self.assertFalse(callable(self.dotflow.workflow_id))
92+
93+
def test_dotflow_task_object(self):
94+
self.assertIsNotNone(self.dotflow.task)
95+
self.assertFalse(callable(self.dotflow.task))
96+
97+
def test_dotflow_add_object(self):
98+
self.assertIsNotNone(self.dotflow.add)
99+
self.assertTrue(callable(self.dotflow.add))
61100

62-
self.assertEqual(len(result), 1)
63-
self.assertEqual(result[0], None)
101+
def test_dotflow_start_object(self):
102+
self.assertIsNotNone(self.dotflow.start)
103+
self.assertTrue(callable(self.dotflow.start))

‎tests/core/test_queue.py‎

Lines changed: 5 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -15,24 +15,22 @@ def setUp(self):
1515
step=action_step,
1616
callback=simple_callback
1717
)
18+
self.index = 0
1819

1920
def test_instantiating_queue_class(self):
2021
expected_tasks = []
2122
queue = Queue()
22-
2323
self.assertIsInstance(queue, Queue)
2424
self.assertListEqual(queue.tasks, expected_tasks)
2525

2626
def test_queue_add(self):
2727
queue = Queue()
28-
2928
queue.add(item=self.task)
3029

31-
self.assertEqual(queue.tasks[0], self.task)
30+
self.assertEqual(queue.tasks[self.index], self.task)
3231

3332
def test_queue_remove(self):
3433
queue = Queue()
35-
3634
queue.add(item=self.task)
3735
queue.remove()
3836

@@ -41,7 +39,6 @@ def test_queue_remove(self):
4139
def test_queue_size(self):
4240
expected_size = 1
4341
queue = Queue()
44-
4542
queue.add(item=self.task)
4643

4744
self.assertEqual(queue.size(), expected_size)
@@ -53,26 +50,21 @@ def test_queue_reverse(self):
5350
callback=simple_callback
5451
)
5552
queue = Queue()
56-
5753
queue.add(item=self.task)
5854
queue.add(item=expected_task)
59-
6055
queue.reverse()
6156

62-
self.assertEqual(queue.tasks[0], expected_task)
57+
self.assertEqual(queue.tasks[self.index], expected_task)
6358

6459
def test_queue_clear(self):
6560
queue = Queue()
66-
6761
queue.add(item=self.task)
68-
6962
queue.clear()
7063

7164
self.assertFalse(queue.tasks)
7265

7366
def test_queue_get(self):
7467
queue = Queue()
75-
7668
queue.add(item=self.task)
7769

7870
self.assertListEqual(queue.get(), [self.task])
@@ -86,6 +78,7 @@ def setUp(self):
8678
step=action_step,
8779
callback=simple_callback
8880
)
81+
self.index = 0
8982

9083
def test_instantiating_queue_group_class(self):
9184
expected_queue = {}
@@ -101,7 +94,7 @@ def test_queue_group_add(self):
10194

10295
self.assertIsNotNone(queue_group.queue.get(TASK_GROUP_NAME))
10396
self.assertIsInstance(queue_group.queue.get(TASK_GROUP_NAME), Queue)
104-
self.assertEqual(queue_group.queue[TASK_GROUP_NAME].tasks[0], self.task)
97+
self.assertEqual(queue_group.queue[TASK_GROUP_NAME].tasks[self.index], self.task)
10598

10699
def test_queue_group_size(self):
107100
expected_size = 1
@@ -120,15 +113,13 @@ def test_queue_group_count(self):
120113
group_name="new_group"
121114
)
122115
queue = QueueGroup()
123-
124116
queue.add(item=self.task)
125117
queue.add(item=task)
126118

127119
self.assertEqual(queue.count(), expected_count)
128120

129121
def test_queue_group_tasks(self):
130122
queue = QueueGroup()
131-
132123
queue.add(item=self.task)
133124

134125
self.assertIsNotNone(queue.tasks())

0 commit comments

Comments
 (0)