Merge branch 'master' of github.com:natict/cqlengine into pull-171

This commit is contained in:
Blake Eggleston
2014-03-20 16:34:55 -07:00
2 changed files with 22 additions and 2 deletions

View File

@@ -537,7 +537,7 @@ class BaseContainerColumn(Column):
Returns a column definition for CQL table definition
"""
db_type = self.db_type.format(self.value_type.db_type)
return '{} {}'.format(self.db_field_name, db_type)
return '{} {}'.format(self.cql, db_type)
def get_update_statement(self, val, prev, ctx):
"""
@@ -813,7 +813,7 @@ class Map(BaseContainerColumn):
self.key_type.db_type,
self.value_type.db_type
)
return '{} {}'.format(self.db_field_name, db_type)
return '{} {}'.format(self.cql, db_type)
def validate(self, value):
val = super(Map, self).validate(value)

View File

@@ -522,3 +522,23 @@ class TestMapColumn(BaseCassEngTestCase):
# assert len([v for v in ctx.values() if [7,8,9] == v.value]) == 1
# assert len([s for s in statements if '"TEST" = "TEST" +' in s]) == 1
# assert len([s for s in statements if '+ "TEST"' in s]) == 1
class TestCamelMapModel(Model):
partition = columns.UUID(primary_key=True, default=uuid4)
camelMap = columns.Map(columns.Text, columns.Integer, required=False)
class TestCamelMapColumn(BaseCassEngTestCase):
@classmethod
def setUpClass(cls):
super(TestCamelMapColumn, cls).setUpClass()
drop_table(TestCamelMapModel)
sync_table(TestCamelMapModel)
@classmethod
def tearDownClass(cls):
super(TestCamelMapColumn, cls).tearDownClass()
drop_table(TestCamelMapModel)
def test_camelcase_column(self):
TestCamelMapModel.create(partition=None, camelMap={'blah': 1})