def test_get_should_instantiate_new_tag_selection_and_initiate_it(self): callable_factory = mock() args = ['selector'] tagSelectionMock = mock(tagselection._TagSelection) when(tagSelectionMock).__call__(anyx(), anyx()).thenReturn(tagSelectionMock) tagselection._TagSelection = utils.CallableMock(tagSelectionMock) tagselection.TagSelector(callable_factory).get(*args) verify(tagSelectionMock, times=1).__call__(anyx(collections.defaultdict), callable_factory) verify(tagSelectionMock, times=1).union(*args)
def setUp(self): self.TagSelector = core.tagselection.TagSelector self.tagSelectorMock = mock(core.tagselection.TagSelector) core.tagselection.TagSelector = utils.CallableMock(self.tagSelectorMock) when(self.tagSelectorMock).__call__(anyx()).thenReturn(self.tagSelectorMock) self.TagManager = core.tagselection.TagManager self.tagManagerMock = mock(core.tagselection.TagManager) core.tagselection.TagManager = utils.CallableMock(self.tagManagerMock) when(self.tagManagerMock).__call__(anyx()).thenReturn(self.tagManagerMock) when(self.tagManagerMock).infere_tags(anyx()).thenReturn(set()) self.engine = core.DatasourceEngine()
def test_get_should_only_use_builder_once_for_each_registered_reference(self): for i in range(2): self.engine.get('REGISTERED_KEY_1') for i in range(2): self.engine.get('REGISTERED_KEY_2') self.engine.get('REGISTERED_KEY_1') verify(self.builder_mock, times=2).build(anyx())
def setUp(self): self.TagSelector = core.tagselection.TagSelector self.tagSelectorMock = mock(core.tagselection.TagSelector) core.tagselection.TagSelector = utils.CallableMock( self.tagSelectorMock) when(self.tagSelectorMock).__call__(anyx()).thenReturn( self.tagSelectorMock) self.TagManager = core.tagselection.TagManager self.tagManagerMock = mock(core.tagselection.TagManager) core.tagselection.TagManager = utils.CallableMock(self.tagManagerMock) when(self.tagManagerMock).__call__(anyx()).thenReturn( self.tagManagerMock) when(self.tagManagerMock).infere_tags(anyx()).thenReturn(set()) self.engine = core.DatasourceEngine()
def setUp(self): self.TagManager = core.tagselection.TagManager self.tagManagerMock = mock(core.tagselection.TagManager) when(self.tagManagerMock).__call__(anyx(dict)).thenReturn(self.tagManagerMock) core.tagselection.TagManager = utils.CallableMock(self.tagManagerMock) self.engine = core.DatasourceEngine()
def test_get_should_only_use_builder_once_for_each_registered_reference( self): for i in range(2): self.engine.get('REGISTERED_KEY_1') for i in range(2): self.engine.get('REGISTERED_KEY_2') self.engine.get('REGISTERED_KEY_1') verify(self.builder_mock, times=2).build(anyx())
def setUp(self): self.TagManager = core.tagselection.TagManager self.tagManagerMock = mock(core.tagselection.TagManager) when(self.tagManagerMock).__call__(anyx(dict)).thenReturn( self.tagManagerMock) core.tagselection.TagManager = utils.CallableMock(self.tagManagerMock) self.engine = core.DatasourceEngine()
def test_get_without_passing_index_should_return_dict_with_each_item_result_from_callable(self): self.tagSelection._initial_selection = set('abcdefghij') self.tag_mappings.update({'2': set('defg')}) result = self.tagSelection.difference('2').get() assert isinstance(result, dict) assert len(result) is 6 verify(self.callable_factory, times=6).__call__(anyx())
def test_for_synched_should_delegate_to_tagManager_register_synched_method(self): selection = object() def function(): pass self.engine.for_synched(selection)(function) verify(self.tagManagerMock, times=1).register_synched(function, anyx())
def test_for_synched_should_delegate_to_tagManager_register_synched_method( self): selection = object() def function(): pass self.engine.for_synched(selection)(function) verify(self.tagManagerMock, times=1).register_synched(function, anyx())
def test_get_without_passing_index_should_return_dict_with_each_item_result_from_callable( self): self.tagSelection._initial_selection = set('abcdefghij') self.tag_mappings.update({'2': set('defg')}) result = self.tagSelection.difference('2').get() assert isinstance(result, dict) assert len(result) is 6 verify(self.callable_factory, times=6).__call__(anyx())
def test_register_should_call_listener_for_late_reference_when_there_is_one_selector(self): reference = 'REFERENCE' selection = mock() when(selection).__iter__().thenReturn([].__iter__()).thenReturn([reference].__iter__()) when(selection).is_elegible(reference).thenReturn(True) self.manager.register(self.listener_callable_mock, utils.IterableMock(selection)) verify(self.listener_mock, times=0).__call__(anyx()) verifyNoMoreInteractions(self.listener_mock) self.manager.evaluate_new_candidate(reference) verify(self.listener_mock, times=1).__call__(reference) verifyNoMoreInteractions(self.listener_mock)
def test_process_should_process_dependencies(self): instance = mock() registration = mock() dependencies = { 'a': mock(task.SimpleDatasourceTask), 'b': mock(task.SimpleDatasourceTask), 'c': mock(task.SimpleDatasourceTask), } task.SimpleDatasourceTask(instance, registration, dependencies).process() verify(dependencies['a'], times=1).process() verify(dependencies['b'], times=1).process() verify(dependencies['c'], times=1).process() verify(instance, times=1).evaluate(anyx(dict))
def test_register_should_call_listener_for_late_reference_when_there_is_one_selector( self): reference = 'REFERENCE' selection = mock() when(selection).__iter__().thenReturn([].__iter__()).thenReturn( [reference].__iter__()) when(selection).is_elegible(reference).thenReturn(True) self.manager.register(self.listener_callable_mock, utils.IterableMock(selection)) verify(self.listener_mock, times=0).__call__(anyx()) verifyNoMoreInteractions(self.listener_mock) self.manager.evaluate_new_candidate(reference) verify(self.listener_mock, times=1).__call__(reference) verifyNoMoreInteractions(self.listener_mock)
def test_process_should_process_dependencies(self): instance = mock() registration = mock() symbol = 'symbol' start = None end = None dependencies = { 'a': mock(task.DataFrameDatasourceTask), 'b': mock(task.DataFrameDatasourceTask), 'c': mock(task.DataFrameDatasourceTask), } task.DataFrameDatasourceTask(instance, registration, dependencies).process(symbol, start, end) verify(dependencies['a'], times=1).process(symbol, start, end) verify(dependencies['b'], times=1).process(symbol, start, end) verify(dependencies['c'], times=1).process(symbol, start, end) verify(instance, times=1).evaluate(anyx(task.DatasourceContext), symbol, start, end)
def test_get_passing_index_should_return_item_result_from_callable(self): self.tagSelection._initial_selection = set('abcdefghij') self.tag_mappings.update({'2': set('defg')}) self.tagSelection.difference('2').get(2) verify(self.callable_factory, times=1).__call__(anyx())
def test_register_datasource_should_instantiate_tag_selector_on_initialization( self): verify(self.tagSelectorMock, times=1).__call__(anyx())
def test_get_should_not_use_builder_when_reference_was_not_registered( self): self.assertRaises(KeyError, self.engine.get, 'NOT_REGISTERED_KEY_1') verify(self.builder_mock, times=0).build(anyx())
def test_get_should_use_builder_when_reference_was_registered(self): self.engine.get('REGISTERED_KEY_1') verify(self.builder_mock, times=1).build(anyx())
def test_register_datasource_should_instantiate_tag_selector_on_initialization(self): verify(self.tagSelectorMock, times=1).__call__(anyx())
def test_get_should_not_use_builder_when_reference_was_not_registered(self): self.assertRaises(KeyError, self.engine.get, 'NOT_REGISTERED_KEY_1') verify(self.builder_mock, times=0).build(anyx())