add uuid field and tests
This commit is contained in:
@ -6,6 +6,7 @@ import databases
|
||||
import pydantic
|
||||
import pytest
|
||||
import sqlalchemy
|
||||
import uuid
|
||||
|
||||
import ormar
|
||||
from ormar.exceptions import QueryDefinitionError, NoMatch
|
||||
@ -25,6 +26,16 @@ class JsonSample(ormar.Model):
|
||||
test_json: ormar.JSON(nullable=True)
|
||||
|
||||
|
||||
class UUIDSample(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "uuids"
|
||||
metadata = metadata
|
||||
database = database
|
||||
|
||||
id: ormar.UUID(primary_key=True, default=uuid.uuid4)
|
||||
test_text: ormar.Text()
|
||||
|
||||
|
||||
class User(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "users"
|
||||
@ -113,6 +124,28 @@ async def test_json_column():
|
||||
assert items[1].test_json == dict(aa=12)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_uuid_column():
|
||||
async with database:
|
||||
async with database.transaction(force_rollback=True):
|
||||
u1 = await UUIDSample.objects.create(test_text="aa")
|
||||
u2 = await UUIDSample.objects.create(test_text="bb")
|
||||
|
||||
items = await UUIDSample.objects.all()
|
||||
assert len(items) == 2
|
||||
|
||||
assert isinstance(items[0].id, uuid.UUID)
|
||||
assert isinstance(items[1].id, uuid.UUID)
|
||||
|
||||
assert items[0].id in (u1.id, u2.id)
|
||||
assert items[1].id in (u1.id, u2.id)
|
||||
|
||||
assert items[0].id != items[1].id
|
||||
|
||||
item = await UUIDSample.objects.filter(id=u1.id).get()
|
||||
assert item.id == u1.id
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_model_crud():
|
||||
async with database:
|
||||
|
||||
Reference in New Issue
Block a user