def test_handle_entities(self): """Call handle_entity on all matching Entity and VCSEntity pairs.""" vcs_entities = { 'match': VCSEntityFactory(), 'no_db_match': VCSEntityFactory() } db_entities = { 'match': EntityFactory(), 'no_vcs_match': EntityFactory() } with patch('pontoon.sync.core.get_vcs_entities', return_value=vcs_entities), \ patch('pontoon.sync.core.get_db_entities', return_value=db_entities), \ patch('pontoon.sync.core.handle_entity') as mock_handle_entity: sync_project(self.db_project) mock_handle_entity.assert_has_calls([ call(ANY, self.db_project, 'match', db_entities['match'], vcs_entities['match']), call(ANY, self.db_project, 'no_vcs_match', db_entities['no_vcs_match'], None), call(ANY, self.db_project, 'no_db_match', None, vcs_entities['no_db_match']), ], any_order=True)
def test_handle_project_handle_entities(self): """Call handle_entity on all matching Entity and VCSEntity pairs.""" vcs_entities = { 'match': VCSEntityFactory(), 'no_db_match': VCSEntityFactory() } self.command.get_vcs_entities = Mock(return_value=vcs_entities) db_entities = { 'match': EntityFactory(), 'no_vcs_match': EntityFactory() } self.command.get_db_entities = Mock(return_value=db_entities) self.command.handle_entity = Mock() self.command.handle_project(self.db_project) self.command.handle_entity.assert_has_calls([ call(ANY, self.db_project, 'match', db_entities['match'], vcs_entities['match']), call(ANY, self.db_project, 'no_vcs_match', db_entities['no_vcs_match'], None), call(ANY, self.db_project, 'no_db_match', None, vcs_entities['no_db_match']), ], any_order=True)