"""Basic tests flows folder.""" import importlib from unittest.mock import MagicMock from accounting import flows class Mock_Flow(object): """Mocked flow.""" def __init__(self): """Initialize mock flow object.""" self.msg = 'i am a test workflow class' class Mock_Class(object): """Mock flow class.""" @property def Flow(self): """Property returning flow class.""" return Mock_Flow def test_get_flow_get_a_class(monkeypatch): """Test get_flow as a class.""" mock_flow = Mock_Class() monkeypatch.setattr( importlib, 'import_module', MagicMock(return_value=mock_flow)) actual = flows.get_flow('Mock_Flow') assert isinstance(actual, Mock_Flow)