mirror of
https://github.com/hanzoai/datastore-orm.git
synced 2026-09-30 21:40:30 +00:00
zeekay
dda3979851
This is our fork, so it should read like it. `import datastore_orm` instead of `from infi.clickhouse_orm import ...`: `infi` was the original vendor's org namespace, and carrying someone else's org name through every import in the consuming codebase is not a fork, it is a vendored copy that never got renamed. The `infi` namespace package is gone entirely (src/infi/__init__.py with it), the distribution is `datastore-orm`, and the metadata points at Hanzo rather than at the upstream repo and author. NOT renamed, deliberately: the migration history tables are still `infi_clickhouse_orm_migrations`. That name is persisted state — 448 rows in production key off it — so changing it is a data migration, not a rename. Doing it carelessly reproduces the exact failure this fork just hit, where a package rename orphaned the history and silently blocked every migration.
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
import unittest
|
|
from datastore_orm.database import Database
|
|
from datastore_orm.fields import Field, Int16Field
|
|
from datastore_orm.models import Model
|
|
from datastore_orm.engines import Memory
|
|
|
|
|
|
class CustomFieldsTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.database = Database('test-db', log_statements=True)
|
|
|
|
def tearDown(self):
|
|
self.database.drop_database()
|
|
|
|
def test_boolean_field(self):
|
|
# Create a model
|
|
class TestModel(Model):
|
|
i = Int16Field()
|
|
f = BooleanField()
|
|
engine = Memory()
|
|
self.database.create_table(TestModel)
|
|
# Check valid values
|
|
for index, value in enumerate([1, '1', True, 0, '0', False]):
|
|
rec = TestModel(i=index, f=value)
|
|
self.database.insert([rec])
|
|
self.assertEqual([rec.f for rec in TestModel.objects_in(self.database).order_by('i')],
|
|
[True, True, True, False, False, False])
|
|
# Check invalid values
|
|
for value in [None, 'zzz', -5, 7]:
|
|
with self.assertRaises(ValueError):
|
|
TestModel(i=1, f=value)
|
|
|
|
|
|
class BooleanField(Field):
|
|
|
|
# The ClickHouse column type to use
|
|
db_type = 'UInt8'
|
|
|
|
# The default value if empty
|
|
class_default = False
|
|
|
|
def to_python(self, value, timezone_in_use):
|
|
# Convert valid values to bool
|
|
if value in (1, '1', True):
|
|
return True
|
|
elif value in (0, '0', False):
|
|
return False
|
|
else:
|
|
raise ValueError('Invalid value for BooleanField: %r' % value)
|
|
|
|
def to_db_string(self, value, quote=True):
|
|
# The value was already converted by to_python, so it's a bool
|
|
return '1' if value else '0'
|
|
|