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.
92 lines
3.5 KiB
Python
92 lines
3.5 KiB
Python
import unittest
|
|
|
|
from datastore_orm.database import Database
|
|
from datastore_orm.models import Model
|
|
from datastore_orm.fields import *
|
|
from datastore_orm.engines import *
|
|
|
|
from enum import Enum
|
|
|
|
|
|
class EnumFieldsTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.database = Database('test-db', log_statements=True)
|
|
self.database.create_table(ModelWithEnum)
|
|
self.database.create_table(ModelWithEnumArray)
|
|
|
|
def tearDown(self):
|
|
self.database.drop_database()
|
|
|
|
def test_insert_and_select(self):
|
|
self.database.insert([
|
|
ModelWithEnum(date_field='2016-08-30', enum_field=Fruit.apple),
|
|
ModelWithEnum(date_field='2016-08-31', enum_field=Fruit.orange),
|
|
ModelWithEnum(date_field='2016-08-31', enum_field=Fruit.cherry)
|
|
])
|
|
query = 'SELECT * from $table ORDER BY date_field'
|
|
results = list(self.database.select(query, ModelWithEnum))
|
|
self.assertEqual(len(results), 3)
|
|
self.assertEqual(results[0].enum_field, Fruit.apple)
|
|
self.assertEqual(results[1].enum_field, Fruit.orange)
|
|
self.assertEqual(results[2].enum_field, Fruit.cherry)
|
|
|
|
def test_ad_hoc_model(self):
|
|
self.database.insert([
|
|
ModelWithEnum(date_field='2016-08-30', enum_field=Fruit.apple),
|
|
ModelWithEnum(date_field='2016-08-31', enum_field=Fruit.orange),
|
|
ModelWithEnum(date_field='2016-08-31', enum_field=Fruit.cherry)
|
|
])
|
|
query = 'SELECT * from $db.modelwithenum ORDER BY date_field'
|
|
results = list(self.database.select(query))
|
|
self.assertEqual(len(results), 3)
|
|
self.assertEqual(results[0].enum_field.name, Fruit.apple.name)
|
|
self.assertEqual(results[0].enum_field.value, Fruit.apple.value)
|
|
self.assertEqual(results[1].enum_field.name, Fruit.orange.name)
|
|
self.assertEqual(results[1].enum_field.value, Fruit.orange.value)
|
|
self.assertEqual(results[2].enum_field.name, Fruit.cherry.name)
|
|
self.assertEqual(results[2].enum_field.value, Fruit.cherry.value)
|
|
|
|
def test_conversion(self):
|
|
self.assertEqual(ModelWithEnum(enum_field=3).enum_field, Fruit.orange)
|
|
self.assertEqual(ModelWithEnum(enum_field=-7).enum_field, Fruit.cherry)
|
|
self.assertEqual(ModelWithEnum(enum_field='apple').enum_field, Fruit.apple)
|
|
self.assertEqual(ModelWithEnum(enum_field=Fruit.banana).enum_field, Fruit.banana)
|
|
|
|
def test_assignment_error(self):
|
|
for value in (0, 17, 'pear', '', None, 99.9):
|
|
with self.assertRaises(ValueError):
|
|
ModelWithEnum(enum_field=value)
|
|
|
|
def test_default_value(self):
|
|
instance = ModelWithEnum()
|
|
self.assertEqual(instance.enum_field, Fruit.apple)
|
|
|
|
def test_enum_array(self):
|
|
instance = ModelWithEnumArray(date_field='2016-08-30', enum_array=[Fruit.apple, Fruit.apple, Fruit.orange])
|
|
self.database.insert([instance])
|
|
query = 'SELECT * from $table ORDER BY date_field'
|
|
results = list(self.database.select(query, ModelWithEnumArray))
|
|
self.assertEqual(len(results), 1)
|
|
self.assertEqual(results[0].enum_array, instance.enum_array)
|
|
|
|
|
|
Fruit = Enum('Fruit', [('apple', 1), ('banana', 2), ('orange', 3), ('cherry', -7)])
|
|
|
|
|
|
class ModelWithEnum(Model):
|
|
|
|
date_field = DateField()
|
|
enum_field = Enum8Field(Fruit)
|
|
|
|
engine = MergeTree('date_field', ('date_field',))
|
|
|
|
|
|
class ModelWithEnumArray(Model):
|
|
|
|
date_field = DateField()
|
|
enum_array = ArrayField(Enum16Field(Fruit))
|
|
|
|
engine = MergeTree('date_field', ('date_field',))
|
|
|