mirror of
https://github.com/Infinidat/infi.clickhouse_orm.git
synced 2024-11-28 19:53:44 +03:00
30 lines
715 B
Python
30 lines
715 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from infi.clickhouse_orm.models import BufferModel
|
||
|
|
from infi.clickhouse_orm.engines import *
|
||
|
|
from .base_test_with_data import *
|
||
|
|
|
||
|
|
|
||
|
|
class BufferTestCase(TestCaseWithData):
|
||
|
|
|
||
|
|
def _insert_and_check_buffer(self, data, count):
|
||
|
|
self.database.insert(data)
|
||
|
|
self.assertEquals(count, self.database.count(PersonBuffer))
|
||
|
|
|
||
|
|
def _sample_buffer_data(self):
|
||
|
|
for entry in data:
|
||
|
|
yield PersonBuffer(**entry)
|
||
|
|
|
||
|
|
def test_insert_buffer(self):
|
||
|
|
self.database.create_table(PersonBuffer)
|
||
|
|
self._insert_and_check_buffer(self._sample_buffer_data(), len(data))
|
||
|
|
|
||
|
|
|
||
|
|
class PersonBuffer(BufferModel, Person):
|
||
|
|
|
||
|
|
engine = Buffer(Person)
|
||
|
|
|
||
|
|
|