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.
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
import unittest
|
|
|
|
from datastore_orm.database import Database
|
|
from datastore_orm.models import Model
|
|
from datastore_orm.fields import *
|
|
from datastore_orm.engines import *
|
|
|
|
|
|
class FixedStringFieldsTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.database = Database('test-db', log_statements=True)
|
|
self.database.create_table(FixedStringModel)
|
|
|
|
def tearDown(self):
|
|
self.database.drop_database()
|
|
|
|
def _insert_sample_data(self):
|
|
self.database.insert([
|
|
FixedStringModel(date_field='2016-08-30', fstr_field=''),
|
|
FixedStringModel(date_field='2016-08-30'),
|
|
FixedStringModel(date_field='2016-08-31', fstr_field='foo'),
|
|
FixedStringModel(date_field='2016-08-31', fstr_field=u'לילה')
|
|
])
|
|
|
|
def _assert_sample_data(self, results):
|
|
self.assertEqual(len(results), 4)
|
|
self.assertEqual(results[0].fstr_field, '')
|
|
self.assertEqual(results[1].fstr_field, 'ABCDEFGHIJK')
|
|
self.assertEqual(results[2].fstr_field, 'foo')
|
|
self.assertEqual(results[3].fstr_field, u'לילה')
|
|
|
|
def test_insert_and_select(self):
|
|
self._insert_sample_data()
|
|
query = 'SELECT * from $table ORDER BY date_field'
|
|
results = list(self.database.select(query, FixedStringModel))
|
|
self._assert_sample_data(results)
|
|
|
|
def test_ad_hoc_model(self):
|
|
self._insert_sample_data()
|
|
query = 'SELECT * from $db.fixedstringmodel ORDER BY date_field'
|
|
results = list(self.database.select(query))
|
|
self._assert_sample_data(results)
|
|
|
|
def test_assignment_error(self):
|
|
for value in (17, 'this is too long', u'זה ארוך', None, 99.9):
|
|
with self.assertRaises(ValueError):
|
|
FixedStringModel(fstr_field=value)
|
|
|
|
|
|
class FixedStringModel(Model):
|
|
|
|
date_field = DateField()
|
|
fstr_field = FixedStringField(12, default='ABCDEFGHIJK')
|
|
|
|
engine = MergeTree('date_field', ('date_field',))
|