def test_index_by_fk_1(self): columns = {'id': '1', 'area': '2', 'type': '3'} parsed_message = Message(1, 'area_alias', columns, 'delete') handler.SCHEMA = SCHEMA self.handler = handler.Handler(SCHEMA.keys()) for entity_type, entity in SCHEMA.items(): self.handler.cores[entity_type] = mock.Mock() self.handler._index_by_fk(parsed_message) calls = self.handler.db_session().execute.call_args_list self.assertEqual(len(calls), 6) actual_queries = [str(call[0][0]) for call in calls] expected_queries = [ 'SELECT place_1.id AS place_1_id \n' 'FROM musicbrainz.place AS place_1 JOIN musicbrainz.area ON musicbrainz.area.id = place_1.area \n' 'WHERE musicbrainz.area.id = :id_1', 'SELECT label_1.id AS label_1_id \n' 'FROM musicbrainz.label AS label_1 JOIN musicbrainz.area ON musicbrainz.area.id = label_1.area \n' 'WHERE musicbrainz.area.id = :id_1', 'SELECT artist_1.id AS artist_1_id \n' 'FROM musicbrainz.artist AS artist_1 JOIN musicbrainz.area ON musicbrainz.area.id = artist_1.end_area \n' 'WHERE musicbrainz.area.id = :id_1', 'SELECT artist_1.id AS artist_1_id \n' 'FROM musicbrainz.artist AS artist_1 JOIN musicbrainz.area ON musicbrainz.area.id = artist_1.area \n' 'WHERE musicbrainz.area.id = :id_1', 'SELECT artist_1.id AS artist_1_id \n' 'FROM musicbrainz.artist AS artist_1 JOIN musicbrainz.area ON musicbrainz.area.id = artist_1.begin_area \n' 'WHERE musicbrainz.area.id = :id_1', 'SELECT musicbrainz.area.id AS musicbrainz_area_id \n' 'FROM musicbrainz.area \n' 'WHERE musicbrainz.area.id = :id_1' ] self.assertEqual(expected_queries, actual_queries)
def setUp(self): super(HandlerTest, self).setUp() handler.SCHEMA = {self.entity_type: None} solr_version_check_patcher = mock.patch("sir.amqp.handler.solr_version_check") self.addCleanup(solr_version_check_patcher.stop) solr_version_check_patcher.start() self.handler = handler.Handler() self.handler.cores[self.entity_type] = mock.Mock()
def test_ack(self): def wrapped_f(*args, **kwargs): pass f = handler.callback_wrapper(wrapped_f) _handler = handler.Handler() f(_handler, self.message, "search.index") # Process messages to ack _handler.process_messages() self.message.channel.basic_ack.assert_called_once_with(self.delivery_tag)
def test_index_by_fk_3(self): columns = {'release_group': 1} parsed_message = Message(1, 'release', columns, 'delete') handler.SCHEMA = SCHEMA self.handler = handler.Handler() for entity_type, entity in SCHEMA.items(): self.handler.cores[entity_type] = mock.Mock() entity.build_entity_query = mock.MagicMock() self.handler._index_by_fk(parsed_message) calls = self.handler.db_session().execute.call_args_list self.assertEqual(len(calls), 1) actual_queries = [str(call[0][0]) for call in calls] expected_queries = [ 'SELECT musicbrainz.release_group.id AS musicbrainz_release_group_id \n' 'FROM musicbrainz.release_group \n' 'WHERE musicbrainz.release_group.id = :id_1'] self.assertEqual(expected_queries, actual_queries)
def setUp(self): super(HandlerTest, self).setUp() handler.SCHEMA = {self.entity_type: None} solr_version_check_patcher = mock.patch( "sir.amqp.handler.solr_version_check") self.addCleanup(solr_version_check_patcher.stop) solr_version_check_patcher.start() self.handler = handler.Handler(handler.SCHEMA.keys()) self.channel = self.handler.channel = mock.MagicMock() self.handler.connection = mock.MagicMock() self.handler.cores[self.entity_type] = mock.Mock() for entity_type, entity in SCHEMA.items(): patcher = mock.patch.object(entity, 'build_entity_query') patcher.start() self.addCleanup(patcher.stop)
def setUp(self): super(CallbackWrapperTest, self).setUp() self.handler = handler.Handler(SCHEMA.keys()) self.channel = self.handler.channel = mock.MagicMock() self.handler.connection = mock.MagicMock()