fix coverage, bump version, update relaese, update docs
This commit is contained in:
@ -11,6 +11,25 @@ database = databases.Database(DATABASE_URL)
|
||||
metadata = sqlalchemy.MetaData()
|
||||
|
||||
|
||||
class Band(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "bands"
|
||||
metadata = metadata
|
||||
database = database
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=100)
|
||||
|
||||
|
||||
class ArtistsBands(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "artists_x_bands"
|
||||
metadata = metadata
|
||||
database = database
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
|
||||
|
||||
class Artist(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "artists"
|
||||
@ -19,6 +38,7 @@ class Artist(ormar.Model):
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=100)
|
||||
bands = ormar.ManyToMany(Band, through=ArtistsBands)
|
||||
|
||||
|
||||
class Album(ormar.Model):
|
||||
@ -52,8 +72,16 @@ def create_test_database():
|
||||
metadata.drop_all(engine)
|
||||
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
async def cleanup():
|
||||
yield
|
||||
async with database:
|
||||
await Band.objects.delete(each=True)
|
||||
await Artist.objects.delete(each=True)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_simple_cascade():
|
||||
async def test_simple_cascade(cleanup):
|
||||
async with database:
|
||||
artist = await Artist(name="Dr Alban").save()
|
||||
await Album(name="Jamaica", artist=artist).save()
|
||||
@ -63,3 +91,66 @@ async def test_simple_cascade():
|
||||
|
||||
albums = await Album.objects.all()
|
||||
assert len(albums) == 0
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_nested_cascade(cleanup):
|
||||
async with database:
|
||||
artist = await Artist(name="Dr Alban").save()
|
||||
album = await Album(name="Jamaica", artist=artist).save()
|
||||
await Track(title="Yuhu", album=album).save()
|
||||
|
||||
await Artist.objects.delete(id=artist.id)
|
||||
|
||||
artists = await Artist.objects.all()
|
||||
assert len(artists) == 0
|
||||
|
||||
albums = await Album.objects.all()
|
||||
assert len(albums) == 0
|
||||
|
||||
tracks = await Track.objects.all()
|
||||
assert len(tracks) == 0
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_many_to_many_cascade(cleanup):
|
||||
async with database:
|
||||
artist = await Artist(name="Dr Alban").save()
|
||||
band = await Band(name="Scorpions").save()
|
||||
await artist.bands.add(band)
|
||||
|
||||
check = await Artist.objects.select_related("bands").get()
|
||||
assert check.bands[0].name == "Scorpions"
|
||||
|
||||
await Artist.objects.delete(id=artist.id)
|
||||
|
||||
artists = await Artist.objects.all()
|
||||
assert len(artists) == 0
|
||||
|
||||
bands = await Band.objects.all()
|
||||
assert len(bands) == 1
|
||||
|
||||
connections = await ArtistsBands.objects.all()
|
||||
assert len(connections) == 0
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_reverse_many_to_many_cascade(cleanup):
|
||||
async with database:
|
||||
artist = await Artist(name="Dr Alban").save()
|
||||
band = await Band(name="Scorpions").save()
|
||||
await artist.bands.add(band)
|
||||
|
||||
check = await Artist.objects.select_related("bands").get()
|
||||
assert check.bands[0].name == "Scorpions"
|
||||
|
||||
await Band.objects.delete(id=band.id)
|
||||
|
||||
artists = await Artist.objects.all()
|
||||
assert len(artists) == 1
|
||||
|
||||
connections = await ArtistsBands.objects.all()
|
||||
assert len(connections) == 0
|
||||
|
||||
bands = await Band.objects.all()
|
||||
assert len(bands) == 0
|
||||
|
||||
@ -118,6 +118,29 @@ def test_model_attribute_json_access(example):
|
||||
assert example.test_json == dict(aa=12)
|
||||
|
||||
|
||||
def test_missing_metadata():
|
||||
with pytest.raises(ModelDefinitionError):
|
||||
|
||||
class JsonSample2(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "jsons2"
|
||||
database = database
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
test_json = ormar.JSON(nullable=True)
|
||||
|
||||
|
||||
def test_missing_database():
|
||||
with pytest.raises(ModelDefinitionError):
|
||||
|
||||
class JsonSample3(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "jsons3"
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
test_json = ormar.JSON(nullable=True)
|
||||
|
||||
|
||||
def test_non_existing_attr(example):
|
||||
with pytest.raises(ValueError):
|
||||
example.new_attr = 12
|
||||
@ -143,12 +166,13 @@ def test_sqlalchemy_table_is_created(example):
|
||||
|
||||
|
||||
@typing.no_type_check
|
||||
def test_no_pk_in_model_definition(): # type: ignore
|
||||
def test_no_pk_in_model_definition():
|
||||
with pytest.raises(ModelDefinitionError): # type: ignore
|
||||
|
||||
class ExampleModel2(Model): # type: ignore
|
||||
class Meta:
|
||||
tablename = "example2"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
test_string: str = ormar.String(max_length=250) # type: ignore
|
||||
@ -162,6 +186,7 @@ def test_two_pks_in_model_definition():
|
||||
class ExampleModel2(Model):
|
||||
class Meta:
|
||||
tablename = "example3"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
@ -175,6 +200,7 @@ def test_setting_pk_column_as_pydantic_only_in_model_definition():
|
||||
class ExampleModel2(Model):
|
||||
class Meta:
|
||||
tablename = "example4"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
test: int = ormar.Integer(primary_key=True, pydantic_only=True)
|
||||
@ -187,6 +213,7 @@ def test_decimal_error_in_model_definition():
|
||||
class ExampleModel2(Model):
|
||||
class Meta:
|
||||
tablename = "example5"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
test: decimal.Decimal = ormar.Decimal(primary_key=True)
|
||||
@ -199,6 +226,7 @@ def test_string_error_in_model_definition():
|
||||
class ExampleModel2(Model):
|
||||
class Meta:
|
||||
tablename = "example6"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
test: str = ormar.String(primary_key=True)
|
||||
|
||||
Reference in New Issue
Block a user