64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from unittest import skip
|
|
from cassandraengine.tests.base import BaseCassEngTestCase
|
|
|
|
from cassandraengine.models import Model
|
|
from cassandraengine import columns
|
|
|
|
class TestModel(Model):
|
|
count = columns.Integer()
|
|
text = columns.Text()
|
|
|
|
#class TestModel2(Model):
|
|
|
|
class TestModelIO(BaseCassEngTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestModelIO, self).setUp()
|
|
TestModel.objects._create_column_family()
|
|
|
|
def test_model_save_and_load(self):
|
|
"""
|
|
Tests that models can be saved and retrieved
|
|
"""
|
|
tm = TestModel.objects.create(count=8, text='123456789')
|
|
tm2 = TestModel.objects.find(tm.pk)
|
|
|
|
for cname in tm._columns.keys():
|
|
self.assertEquals(getattr(tm, cname), getattr(tm2, cname))
|
|
|
|
def test_model_updating_works_properly(self):
|
|
"""
|
|
Tests that subsequent saves after initial model creation work
|
|
"""
|
|
tm = TestModel.objects.create(count=8, text='123456789')
|
|
|
|
tm.count = 100
|
|
tm.save()
|
|
|
|
tm2 = TestModel.objects.find(tm.pk)
|
|
self.assertEquals(tm.count, tm2.count)
|
|
|
|
def test_nullable_columns_are_saved_properly(self):
|
|
"""
|
|
Tests that nullable columns save without any trouble
|
|
"""
|
|
|
|
@skip
|
|
def test_dynamic_columns(self):
|
|
"""
|
|
Tests that items put into dynamic columns are saved and retrieved properly
|
|
|
|
Note: seems I've misunderstood how arbitrary column names work in Cassandra
|
|
skipping for now
|
|
"""
|
|
#TODO:Fix this
|
|
tm = TestModel(count=8, text='123456789')
|
|
tm['other'] = 'something'
|
|
tm['number'] = 5
|
|
tm.save()
|
|
|
|
tm2 = TestModel.objects.find(tm.pk)
|
|
self.assertEquals(tm['other'], tm2['other'])
|
|
self.assertEquals(tm['number'], tm2['number'])
|
|
|