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)