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 StaticDeleteModel(Model):
example_id = columns.Integer(partition_key=True,
primary_key=True,
default=uuid4)
example_static1 = columns.Integer(static=True)
example_static2 = columns.Integer(static=True)
example_clust = columns.Integer(primary_key=True)
class PrimaryKeysOnlyModel(Model):
__table_name__ = "primary_keys_only"
__options__ = {'compaction': {'class': 'LeveledCompactionStrategy'}}
first_key = columns.Integer(primary_key=True)
second_key = columns.Integer(primary_key=True)
class PrimaryKeysModelChanged(Model):
__table_name__ = "primary_keys_only"
__options__ = {"compaction": {"class": "LeveledCompactionStrategy"}}
new_first_key = columns.Integer(primary_key=True)
second_key = columns.Integer(primary_key=True)
class TestModel(Model):
id = columns.Integer(primary_key=True)
clustering_key = columns.Integer(
primary_key=True,
clustering_order='desc',
)
class TestClusteringComplexModel(Model):
id = columns.Integer(primary_key=True)
clustering_key = columns.Integer(
primary_key=True,
clustering_order='desc',
)
some_value = columns.Integer()
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()
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 PrimaryKeysAddedClusteringKey(Model):
__table_name__ = "primary_keys_only"
__options__ = {"compaction": {"class": "LeveledCompactionStrategy"}}
new_first_key = columns.Float(primary_key=True)
second_key = columns.Integer(primary_key=True)
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()
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 TestDefaultValueTracking(Model):
id = columns.Integer(partition_key=True)
int1 = columns.Integer(default=123)
int2 = columns.Integer(default=456)
int3 = columns.Integer(default=lambda: random.randint(0, 1000))
int4 = columns.Integer(default=lambda: random.randint(0, 1000))
int5 = columns.Integer()
int6 = columns.Integer()
class LeveledCompactionChangesDetectionTest(Model):
__options__ = {
'compaction': {
'class': (
'org.apache.cassandra.db.compaction.'
'LeveledCompactionStrategy'
),
'sstable_size_in_mb': '160',
'tombstone_threshold': '0.125',
'tombstone_compaction_interval': '3600',
},
}
pk = columns.Integer(primary_key=True)
class LeveledCompactionChangesDetectionTest(Model):
__options__ = {
"compaction": {
"class": ("org.apache.cassandra.db.compaction."
"LeveledCompactionStrategy"),
"sstable_size_in_mb":
"160",
"tombstone_threshold":
"0.125",
"tombstone_compaction_interval":
"3600",
}
}
pk = columns.Integer(primary_key=True)
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 SizeTieredCompactionChangesDetectionTest(Model):
__options__ = {
'compaction': {
'class': (
'org.apache.cassandra.db.compaction.'
'SizeTieredCompactionStrategy'
),
'bucket_high': '20',
'bucket_low': '10',
'max_threshold': '200',
'min_threshold': '100',
'min_sstable_size': '1000',
'tombstone_threshold': '0.125',
'tombstone_compaction_interval': '3600',
},
}
pk = columns.Integer(primary_key=True)
class SizeTieredCompactionChangesDetectionTest(Model):
__options__ = {
"compaction": {
"class": ("org.apache.cassandra.db.compaction."
"SizeTieredCompactionStrategy"),
"bucket_high":
"20",
"bucket_low":
"10",
"max_threshold":
"200",
"min_threshold":
"100",
"min_sstable_size":
"1000",
"tombstone_threshold":
"0.125",
"tombstone_compaction_interval":
"3600",
}
}
pk = columns.Integer(primary_key=True)
class AlreadyLoadedTest(ConcreteModelWithCol):
new_field = columns.Integer()
class DateTimeQueryTestModel(Model):
user = columns.Integer(primary_key=True)
day = columns.DateTime(primary_key=True)
data = columns.Text()
class LowercaseKeyModel(Model):
first_key = columns.Integer(primary_key=True)
second_key = columns.Integer(primary_key=True)
some_data = columns.Text()
class CapitalizedKeyModel(Model):
firstKey = columns.Integer(primary_key=True)
secondKey = columns.Integer(primary_key=True)
someData = columns.Text()
class StaticModel(Model):
id = columns.Integer(primary_key=True)
c = columns.Integer(primary_key=True)
name = columns.Text(static=True)
class ChangedInconsistentType(UserType):
__type_name__ = "type_inconsistent"
age = columns.Integer()
name = columns.Integer()
class BaseInconsistentType(UserType):
__type_name__ = "type_inconsistent"
age = columns.Integer()
name = columns.Text()
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')
class BaseInconsistent(Model):
__table_name__ = "inconsistent"
first_key = columns.UUID(primary_key=True)
second_key = columns.Integer(index=True)
third_key = columns.Integer(index=True)
class TestTTLModel(Model):
id = columns.UUID(primary_key=True, default=uuid4)
count = columns.Integer()
text = columns.Text(required=False)
class PrimaryKeysRemovedPk(Model):
__table_name__ = "primary_keys_only"
__options__ = {"compaction": {"class": "LeveledCompactionStrategy"}}
second_key = columns.Integer(primary_key=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 DummyUDT(Model):
__keyspace__ = "ks"
primary_key = columns.Integer(primary_key=True)
value = columns.UserDefinedType(Value)
