test_entities.py

class EntityTestCase(TestCase):

    def setUp(self):

        mb_release_group.mb_session = MagicMock()
        mb_recording.mb_session = MagicMock()
        mb_artist.mb_session = MagicMock()
        mb_label.mb_session = MagicMock()
        mb_place.mb_session = MagicMock()
        mb_event.mb_session = MagicMock()
        mb_work.mb_session = MagicMock()

        self.release_group_query = mb_release_group.mb_session.return_value.\
            options.return_value.options.return_value
        self.recording_query = mb_recording.mb_session.return_value.\
            options.return_value.options.return_value.options.return_value.filter.return_value.all
        self.artist_query = mb_artist.mb_session.return_value.\
            options.return_value.filter.return_value.all
        self.label_query = mb_label.mb_session.return_value.\
            options.return_value.options.return_value.filter.return_value.all
        self.place_query = mb_place.mb_session.return_value.\
            options.return_value.options.return_value.filter.return_value.all    
        self.event_query = mb_event.mb_session.return_value.\
            __enter__.return_value.query.return_value.filter.return_value.all
        self.work_query = mb_work.mb_session.return_value.\
            options.return_value.filter.return_value.all

        setup_cache()

        self.entities = {
            'ebe6ce0f-22c0-4fe7-bfd4-7a0397c9fe94': ('event', {
                'id': 'ebe6ce0f-22c0-4fe7-bfd4-7a0397c9fe94',
                'name': 'Taubertal-Festival 2004, Day 1',
            })
        }

    def test_get_entities(self):
        self.event_query.return_value = [taubertal_festival_2004]
        entities = mb_entities.get_multiple_entities(
            [(mbid, self.entities[mbid][0]) for mbid in self.entities.keys()]
        )
        for mbid in self.entities:
            entity = mb_entities.get_entity_by_id(mbid, type=self.entities[mbid][0])
            self.assertDictEqual(entity, self.entities[mbid][1])
            self.assertDictEqual(entities[mbid], self.entities[mbid][1])