Skip to content

Commit 6dc4106

Browse files
committed
process_tracker_python-9 Process can have more than one source
✨ Process now can have more than one source. Can also have more than one target Processes now have the ability to be registered with more than one source and more than one target (which is an alias of source).
1 parent cda4cce commit 6dc4106

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/test_process_tracker.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
from datetime import datetime, timedelta
44
import unittest
5+
from unittest.mock import patch
56

67
import boto3
78
from sqlalchemy.orm import aliased, Session
@@ -300,7 +301,7 @@ def test_register_extracts_by_location_local(self):
300301
process_status = aliased(ExtractStatus)
301302
extract_status = aliased(ExtractStatus)
302303

303-
with unittest.mock.patch('os.listdir') as mocked_os_listdir:
304+
with patch('os.listdir') as mocked_os_listdir:
304305
mocked_os_listdir.return_value = ['test_local_dir_1.csv', 'test_local_dir_2.csv']
305306

306307
self.process_tracker.register_extracts_by_location(location_path='/test/local/dir/')

0 commit comments

Comments
 (0)