53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
|
# -*- coding: utf-8; -*-
|
||
|
|
|
||
|
|
import uuid as _uuid
|
||
|
|
|
||
|
|
from wuttjamaican.testing import DataTestCase
|
||
|
|
|
||
|
|
try:
|
||
|
|
import sqlalchemy as sa
|
||
|
|
from wuttjamaican.db import model
|
||
|
|
from wuttjamaican.db.model import batch as mod
|
||
|
|
except ImportError:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
|
||
|
|
class TestBatchMixin(DataTestCase):
|
||
|
|
|
||
|
|
def test_basic(self):
|
||
|
|
|
||
|
|
class MyBatch(mod.BatchMixin, model.Base):
|
||
|
|
__tablename__ = 'testing_mybatch'
|
||
|
|
|
||
|
|
model.Base.metadata.create_all(bind=self.session.bind)
|
||
|
|
metadata = sa.MetaData()
|
||
|
|
metadata.reflect(self.session.bind)
|
||
|
|
self.assertIn('testing_mybatch', metadata.tables)
|
||
|
|
|
||
|
|
batch = MyBatch(id=42, uuid=_uuid.UUID('0675cdac-ffc9-7690-8000-6023de1c8cfd'))
|
||
|
|
self.assertEqual(repr(batch), "MyBatch(uuid=UUID('0675cdac-ffc9-7690-8000-6023de1c8cfd'))")
|
||
|
|
self.assertEqual(str(batch), "00000042")
|
||
|
|
|
||
|
|
|
||
|
|
class TestBatchRowMixin(DataTestCase):
|
||
|
|
|
||
|
|
def test_basic(self):
|
||
|
|
|
||
|
|
class MyBatch2(mod.BatchMixin, model.Base):
|
||
|
|
__tablename__ = 'testing_mybatch2'
|
||
|
|
|
||
|
|
class MyBatchRow2(mod.BatchRowMixin, model.Base):
|
||
|
|
__tablename__ = 'testing_mybatch_row2'
|
||
|
|
__batch_class__ = MyBatch2
|
||
|
|
|
||
|
|
model.Base.metadata.create_all(bind=self.session.bind)
|
||
|
|
metadata = sa.MetaData()
|
||
|
|
metadata.reflect(self.session.bind)
|
||
|
|
self.assertIn('testing_mybatch2', metadata.tables)
|
||
|
|
self.assertIn('testing_mybatch_row2', metadata.tables)
|
||
|
|
|
||
|
|
# nb. this gives coverage but doesn't really test much
|
||
|
|
batch = MyBatch2(id=42, uuid=_uuid.UUID('0675cdac-ffc9-7690-8000-6023de1c8cfd'))
|
||
|
|
row = MyBatchRow2()
|
||
|
|
batch.rows.append(row)
|