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
