def test_get_by_id_missing(self): dao = mock.Mock(spec=avalon.models.ReadOnlyDao) dao.get_all_tracks.return_value = [self.song] cache = avalon.cache.TrackStore(dao).reload() songs = cache.get_by_id(uuid.UUID('72e2e340-fabc-4712-aa26-8a8f122999e8')) assert 0 == len(songs)
def test_get_by_id(self): dao = mock.Mock(spec=avalon.models.ReadOnlyDao) dao.get_all_tracks.return_value = [self.song] cache = avalon.cache.TrackStore(dao).reload() songs = cache.get_by_id(uuid.UUID("ca2e8303-69d7-53ec-907e-2f111103ba29")) for song in songs: assert uuid.UUID("ca2e8303-69d7-53ec-907e-2f111103ba29") == song.id
def test_get_by_id(self): model1 = avalon.models.Album() model1.id = uuid.UUID("2d24515c-a459-552a-b022-e85d1621425a") model1.name = 'Dookie' model2 = avalon.models.Album() model2.id = uuid.UUID("b3c204e4-445d-5812-9366-28de6770c4e1") model2.name = 'Insomniac' dao = mock.Mock(spec=avalon.models.ReadOnlyDao) dao.get_all_albums.return_value = [model1, model2] cache = avalon.cache.AlbumStore(dao).reload() res = cache.get_by_id(uuid.UUID("2d24515c-a459-552a-b022-e85d1621425a")) assert 1 == len(res) for dookie in res: assert 'Dookie' == dookie.name