def test_all():
"""Test all items are returned from MessagePack-formatted table."""
with get_fixture_uri(
'databases/objects/music/artists.mpack') as artists_uri:
artists = Table(Artist,
artists_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.msgpack
])
# Fetch artists, and validate properties
assert_that(
list(artists.all().iterator()),
all_of(
has_length(5),
has_items(
has_properties({
'id': 1,
'title': 'Gorillaz'
}), has_properties({
'id': 2,
'title': 'Daft Punk'
}), has_properties({
'id': 3,
'title': 'Friendly Fires'
}), has_properties({
'id': 4,
'title': 'Miike Snow'
}), has_properties({
'id': 5,
'title': 'LCD Soundsystem'
}))))
def test_get_relations():
"""Test relations can be resolved in json-formatted collections."""
with get_fixture_uri(
('collections/artists.json', 'collections/albums.json',
'collections/tracks.json')) as (artists_uri, albums_uri, tracks_uri):
# Artists
artists = Collection(Artist,
artists_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.json
])
# Albums
albums = Collection(Album,
albums_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.json
])
albums.connect(Album.Properties.artist, artists)
# Tracks
tracks = Collection(Track,
tracks_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.json
])
tracks.connect(Track.Properties.album, albums)
tracks.connect(Track.Properties.artist, artists)
# Fetch track, and ensure relations can be resolved
assert_that(
tracks.get(Track['id'] == 1),
has_properties({
'id':
1,
'title':
'Ascension (feat. Vince Staples)',
'artist':
has_properties({
'id': 1,
'title': 'Gorillaz'
}),
'album':
has_properties({
'id':
1,
'title':
'Humanz',
'artist':
has_properties({
'id': 1,
'title': 'Gorillaz'
})
})
}))
def test_all():
"""Test all items are returned from json-formatted collection."""
with get_fixture_uri('collections/artists.json') as artists_uri:
artists = Collection(Artist,
artists_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.json
])
# Fetch artists, and validate properties
assert_that(
list(artists.all().iterator()),
all_of(
has_length(5),
has_items(
has_properties({
'id': 1,
'title': 'Gorillaz'
}), has_properties({
'id': 2,
'title': 'Daft Punk'
}), has_properties({
'id': 3,
'title': 'Friendly Fires'
}), has_properties({
'id': 4,
'title': 'Miike Snow'
}), has_properties({
'id': 5,
'title': 'LCD Soundsystem'
}))))
def test_where():
"""Test a MessagePack-formatted table can be filtered with expressions."""
with get_fixture_uri('collections/objects/cities.mpack') as cities_uri:
cities = Table(City,
cities_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.msgpack
])
# Fetch cities, and validate properties
items = list(
cities.select().where(City['country'] == 'New Zealand').iterator())
assert_that(
items,
all_of(
has_length(35),
has_items(
has_properties({
'id': '2179537',
'name': 'Wellington',
'country': 'New Zealand',
'subcountry': 'Wellington'
}),
has_properties({
'id': '2179670',
'name': 'Wanganui',
'country': 'New Zealand',
'subcountry': 'Manawatu-Wanganui'
}),
has_properties({
'id': '2181133',
'name': 'Timaru',
'country': 'New Zealand',
'subcountry': 'Canterbury'
}),
has_properties({
'id': '2181742',
'name': 'Taupo',
'country': 'New Zealand',
'subcountry': 'Waikato'
}),
has_properties({
'id': '2184155',
'name': 'Pukekohe East',
'country': 'New Zealand',
'subcountry': 'Auckland'
}),
)))
def test_where_objects():
"""Test filtered items can be retrieved from a objects-structured table."""
with get_fixture_uri('collections/objects/cities.mpack') as artists_uri:
cities = Table(City,
artists_uri,
format_structure=MessagePackCollectionStructure.Objects,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.msgpack
])
# Fetch artists, and validate properties
assert_that(
list(cities.select().where(
City['country'] == 'New Zealand').iterator()),
all_of(
has_length(35),
has_items(
has_properties({
'id': '2179537',
'name': 'Wellington',
'country': 'New Zealand',
'subcountry': 'Wellington'
}),
has_properties({
'id': '2179670',
'name': 'Wanganui',
'country': 'New Zealand',
'subcountry': 'Manawatu-Wanganui'
}),
has_properties({
'id': '2181133',
'name': 'Timaru',
'country': 'New Zealand',
'subcountry': 'Canterbury'
}),
has_properties({
'id': '2181742',
'name': 'Taupo',
'country': 'New Zealand',
'subcountry': 'Waikato'
}),
has_properties({
'id': '2184155',
'name': 'Pukekohe East',
'country': 'New Zealand',
'subcountry': 'Auckland'
}),
)))
def test_get_basic():
"""Test items can be retrieved from json-formatted collections."""
with get_fixture_uri('collections/artists.json') as artists_uri:
artists = Collection(Artist,
artists_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.json
])
# Fetch artist, and validate properties
assert_that(artists.get(Artist['id'] == 1),
has_properties({
'id': 1,
'title': 'Gorillaz'
}))
def test_get_basic():
"""Test items can be retrieved from a MessagePack-formatted table."""
with get_fixture_uri(
'databases/objects/music/artists.mpack') as artists_uri:
artists = Table(Artist,
artists_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.msgpack
])
# Fetch artist, and validate properties
assert_that(artists.get(Artist['id'] == 1),
has_properties({
'id': 1,
'title': 'Gorillaz'
}))
def test_create():
"""Test items can be created on json-formatted collections."""
with get_fixture_uri('collections/artists.json') as artists_uri:
artists = Collection(Artist,
artists_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.json
])
# Create artist
artists.create(id=123, title='Fenech-Soler')
# Fetch artist, and validate properties
assert_that(artists.get(Artist['id'] == 123),
has_properties({
'id': 123,
'title': 'Fenech-Soler'
}))
def test_create():
"""Test items can be created on a MessagePack-formatted table."""
with get_fixture_uri(
'databases/objects/music/artists.mpack') as artists_uri:
artists = Table(Artist,
artists_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.msgpack
])
# Create artist
artists.create(id=123, title='Fenech-Soler')
# Fetch artist, and validate properties
assert_that(artists.get(Artist['id'] == 123),
has_properties({
'id': 123,
'title': 'Fenech-Soler'
}))
def test_get_relations():
"""Test relations can be resolved in a MessagePack-formatted table."""
with get_fixture_uri(
('databases/objects/music/artists.mpack',
'databases/objects/music/albums.mpack',
'databases/objects/music/tracks.mpack')) as (artists_uri, albums_uri,
tracks_uri):
# Artists
artists = Table(Artist,
artists_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.msgpack
])
# Albums
albums = Table(Album,
albums_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.msgpack
])
albums.connect(artist=artists)
# Tracks
tracks = Table(Track,
tracks_uri,
plugins=[
byte.compilers.operation, byte.executors.file,
byte.formats.msgpack
])
tracks.connect(artist=artists, album=albums)
# Fetch track, and ensure relations can be resolved
assert_that(
tracks.get(Track['id'] == 1),
has_properties({
'id':
1,
'title':
'Ascension (feat. Vince Staples)',
'artist':
has_properties({
'id': 1,
'title': 'Gorillaz'
}),
'album':
has_properties({
'id':
1,
'title':
'Humanz',
'artist':
has_properties({
'id': 1,
'title': 'Gorillaz'
})
})
}))