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()
