class SecondModel(Model):
__table_name__ = "first_model"
first_key = columns.UUID(primary_key=True)
second_key = columns.UUID()
third_key = columns.Text()
fourth_key = columns.Text()
class ComplexModelRouting(Model):
__table_name__ = "complex_model_routing"
partition = columns.UUID(partition_key=True, default=uuid4)
cluster = columns.Integer(partition_key=True)
count = columns.Integer()
text = columns.Text(partition_key=True)
float = columns.Float(partition_key=True)
text_2 = columns.Text()
def test_instantiation_with_column_instance(self):
"""
Tests that columns instantiated with a column instance work properly
"""
column = columns.Map(columns.Text(min_length=100), columns.Integer())
self.assertIsInstance(column.key_col, columns.Text)
self.assertIsInstance(column.value_col, columns.Integer)
class TestModel(Model):
test_id = columns.Integer(primary_key=True)
attempt_id = columns.Integer(primary_key=True)
description = columns.Text()
expected_result = columns.Integer()
test_result = columns.Integer()
class FourthModel(Model):
__table_name__ = "first_model"
first_key = columns.UUID(primary_key=True)
second_key = columns.UUID()
third_key = columns.Text()
# renamed model field, but map to existing column
renamed = columns.Map(columns.Text, columns.Text, db_field="blah")
class ThirdModel(Model):
__table_name__ = "first_model"
first_key = columns.UUID(primary_key=True)
second_key = columns.UUID()
third_key = columns.Text()
# removed fourth key, but it should stay in the DB
blah = columns.Map(columns.Text, columns.Text)
class TestModelSave(Model):
partition = columns.UUID(primary_key=True, default=uuid4)
cluster = columns.Integer(primary_key=True)
count = columns.Integer(required=False)
text = columns.Text(required=False, index=True)
text_set = columns.Set(columns.Text, required=False)
text_list = columns.List(columns.Text, required=False)
text_map = columns.Map(columns.Text, columns.Text, required=False)
class AllLeveledOptionsModel(Model):
__options__ = {
"compaction": {
"class": ("org.apache.cassandra.db.compaction."
"LeveledCompactionStrategy"),
"sstable_size_in_mb":
"64",
}
}
cid = columns.UUID(primary_key=True)
name = columns.Text()
class AllLeveledOptionsModel(Model):
__options__ = {
'compaction': {
'class': ('org.apache.cassandra.db.compaction.'
'LeveledCompactionStrategy'),
'sstable_size_in_mb':
'64'
},
}
cid = columns.UUID(primary_key=True)
name = columns.Text()
class AllDatatypes(UserType):
a = columns.Ascii()
b = columns.BigInt()
c = columns.Blob()
d = columns.Boolean()
e = columns.DateTime()
f = columns.Decimal()
g = columns.Double()
h = columns.Float()
i = columns.Inet()
j = columns.Integer()
k = columns.Text()
l = columns.TimeUUID()
m = columns.UUID()
n = columns.VarInt()
class AllDatatypesModel(Model):
id = columns.Integer(primary_key=True)
a = columns.Ascii()
b = columns.BigInt()
c = columns.Blob()
d = columns.Boolean()
e = columns.DateTime()
f = columns.Decimal()
g = columns.Double()
h = columns.Float()
i = columns.Inet()
j = columns.Integer()
k = columns.Text()
l = columns.TimeUUID()
m = columns.UUID()
n = columns.VarInt()
class AllSizeTieredOptionsModel(Model):
__options__ = {
'compaction': {
'class': (
'org.apache.cassandra.db.compaction.'
'SizeTieredCompactionStrategy'
),
'bucket_low': '.3',
'bucket_high': '2',
'min_threshold': '2',
'max_threshold': '64',
'tombstone_compaction_interval': '86400',
},
}
cid = columns.UUID(primary_key=True)
name = columns.Text()
class AllSizeTieredOptionsModel(Model):
__options__ = {
"compaction": {
"class": ("org.apache.cassandra.db.compaction."
"SizeTieredCompactionStrategy"),
"bucket_low":
".3",
"bucket_high":
"2",
"min_threshold":
"2",
"max_threshold":
"64",
"tombstone_compaction_interval":
"86400",
}
}
cid = columns.UUID(primary_key=True)
name = columns.Text()
class CapitalizedKeyModel(Model):
firstKey = columns.Integer(primary_key=True)
secondKey = columns.Integer(primary_key=True)
someData = columns.Text()
class LowercaseKeyModel(Model):
first_key = columns.Integer(primary_key=True)
second_key = columns.Integer(primary_key=True)
some_data = columns.Text()
class StaticModel(Model):
id = columns.Integer(primary_key=True)
c = columns.Integer(primary_key=True)
name = columns.Text(static=True)
class BaseInconsistentType(UserType):
__type_name__ = "type_inconsistent"
age = columns.Integer()
name = columns.Text()
class RenamedTest(Model):
__table_name__ = 'manual_name'
id = columns.UUID(primary_key=True)
data = columns.Text()
class BadNames(Model):
words = columns.Text(primary_key=True)
content = columns.Text(db_field='words')
class TestTTLModel(Model):
id = columns.UUID(primary_key=True, default=uuid4)
count = columns.Integer()
text = columns.Text(required=False)
class IndexModel(Model):
__table_name__ = "index_model"
first_key = columns.UUID(primary_key=True)
second_key = columns.Text(index=True)
class TestModel(Model):
id = columns.UUID(primary_key=True, default=uuid4)
text = columns.Text()
class TestPerson(Model):
first_name = columns.Text(primary_key=True, default='kevin')
last_name = columns.Text(default='deldycke')
class Value(UserType):
t = columns.Text()
class DateTimeQueryTestModel(Model):
user = columns.Integer(primary_key=True)
day = columns.DateTime(primary_key=True)
data = columns.Text()
class IndexCaseSensitiveModel(Model):
__table_name__ = "IndexModel"
__table_name_case_sensitive__ = True
first_key = columns.UUID(primary_key=True)
second_key = columns.Text(index=True)
class ModelWithPartitionKeys(Model):
id = columns.UUID(primary_key=True, default=uuid4)
c1 = columns.Text(primary_key=True)
p1 = columns.Text(partition_key=True)
p2 = columns.Text(partition_key=True)
class ChangedInconsistent(Model):
__table_name__ = "inconsistent"
__table_name_case_sensitive__ = True
first_key = columns.UUID(primary_key=True)
second_key = columns.Text(index=True)
class TestDefaultTTLModel(Model):
__options__ = {"default_time_to_live": 20}
id = columns.UUID(primary_key=True, default=uuid4)
count = columns.Integer()
text = columns.Text(required=False)
class WildDBNames(Model):
id = columns.UUID(primary_key=True, default=uuid4)
content = columns.Text(db_field='words_and_whatnot')
numbers = columns.Integer(db_field='integers_etc')
