WIP super dirty - change to descriptors and different tries

tests/test_fastapi/test_binary_fields.py

@@ -0,0 +1,88 @@
+import base64
+import json
+import os
+import uuid
+from typing import List
+
+import databases
+import pydantic
+import pytest
+import sqlalchemy
+from fastapi import FastAPI
+from starlette.testclient import TestClient
+
+import ormar
+from tests.settings import DATABASE_URL
+
+app = FastAPI()
+
+database = databases.Database(DATABASE_URL, force_rollback=True)
+metadata = sqlalchemy.MetaData()
+app.state.database = database
+
+
+@app.on_event("startup")
+async def startup() -> None:
+    database_ = app.state.database
+    if not database_.is_connected:
+        await database_.connect()
+
+
+@app.on_event("shutdown")
+async def shutdown() -> None:
+    database_ = app.state.database
+    if database_.is_connected:
+        await database_.disconnect()
+
+
+blob3 = b"\xc3\x28"
+blob4 = b"\xf0\x28\x8c\x28"
+blob5 = b"\xee"
+blob6 = b"\xff"
+
+
+class BaseMeta(ormar.ModelMeta):
+    metadata = metadata
+    database = database
+
+
+class BinaryThing(ormar.Model):
+    class Meta(BaseMeta):
+        tablename = "things"
+
+    id: uuid.UUID = ormar.UUID(primary_key=True, default=uuid.uuid4)
+    name: str = ormar.Text(default="")
+    bt: bytes = ormar.LargeBinary(
+        max_length=1000,
+        choices=[blob3, blob4, blob5, blob6],
+        represent_as_base64_str=True
+    )
+
+
+@app.get("/things", response_model=List[BinaryThing])
+async def read_things():
+    return await BinaryThing.objects.order_by("name").all()
+
+
+@app.post("/things", response_model=BinaryThing)
+async def create_things(thing: BinaryThing):
+    thing = await thing.save()
+    return thing
+
+
+@pytest.fixture(autouse=True, scope="module")
+def create_test_database():
+    engine = sqlalchemy.create_engine(DATABASE_URL)
+    metadata.create_all(engine)
+    yield
+    metadata.drop_all(engine)
+
+
+def test_read_main():
+    client = TestClient(app)
+    with client as client:
+        response = client.post(
+            "/things", data=json.dumps({"bt": base64.b64encode(blob3).decode()})
+        )
+        print(response.content)
+        assert response.status_code == 200

tests/test_model_definition/test_models.py

@@ -1,5 +1,6 @@
 import asyncio
 import datetime
+import os
 import uuid
 from typing import List
 
@@ -37,7 +38,23 @@ class LargeBinarySample(ormar.Model):
         database = database
 
     id: int = ormar.Integer(primary_key=True)
-    test_binary = ormar.LargeBinary(max_length=100000, choices=[blob, blob2])
+    test_binary: bytes = ormar.LargeBinary(max_length=100000, choices=[blob, blob2])
+
+
+blob3 = os.urandom(64)
+blob4 = os.urandom(100)
+
+
+class LargeBinaryStr(ormar.Model):
+    class Meta:
+        tablename = "my_str_blobs"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    test_binary: str = ormar.LargeBinary(
+        max_length=100000, choices=[blob3, blob4], represent_as_base64=True
+    )
 
 
 class UUIDSample(ormar.Model):
@@ -171,6 +188,19 @@ async def test_binary_column():
             assert items[1].test_binary == blob2
 
 
+@pytest.mark.asyncio
+async def test_binary_str_column():
+    async with database:
+        async with database.transaction(force_rollback=True):
+            await LargeBinaryStr.objects.create(test_binary=blob3)
+            await LargeBinaryStr.objects.create(test_binary=blob4)
+
+            items = await LargeBinaryStr.objects.all()
+            assert len(items) == 2
+            assert items[0].test_binary == blob3
+            assert items[1].test_binary == blob4
+
+
 @pytest.mark.asyncio
 async def test_uuid_column():
     async with database: