39
39
from cassandra .util import uuid_from_time
40
40
from cassandra .cqlengine .connection import get_session
41
41
from tests .integration import PROTOCOL_VERSION , CASSANDRA_VERSION , greaterthancass20 , greaterthancass21 , \
42
- greaterthanorequalcass30 , TestCluster , requirescollectionindexes
42
+ greaterthanorequalcass30 , TestCluster , requires_collection_indexes
43
43
from tests .integration .cqlengine import execute_count , DEFAULT_KEYSPACE
44
44
45
45
@@ -384,7 +384,7 @@ def tearDownClass(cls):
384
384
drop_table (CustomIndexedTestModel )
385
385
drop_table (TestMultiClusteringModel )
386
386
387
- @requirescollectionindexes
387
+ @requires_collection_indexes
388
388
class TestQuerySetCountSelectionAndIteration (BaseQuerySetUsage ):
389
389
390
390
@execute_count (2 )
@@ -558,7 +558,7 @@ class NonEqualityFilteringModel(Model):
558
558
num = qa .count ()
559
559
assert num == 1 , num
560
560
561
- @requirescollectionindexes
561
+ @requires_collection_indexes
562
562
class TestQuerySetDistinct (BaseQuerySetUsage ):
563
563
564
564
@execute_count (1 )
@@ -597,7 +597,7 @@ def test_distinct_with_explicit_count(self):
597
597
self .assertEqual (q .count (), 2 )
598
598
599
599
600
- @requirescollectionindexes
600
+ @requires_collection_indexes
601
601
class TestQuerySetOrdering (BaseQuerySetUsage ):
602
602
@execute_count (2 )
603
603
def test_order_by_success_case (self ):
@@ -646,7 +646,7 @@ def test_ordering_on_multiple_clustering_columns(self):
646
646
assert [r .three for r in results ] == [1 , 2 , 3 , 4 , 5 ]
647
647
648
648
649
- @requirescollectionindexes
649
+ @requires_collection_indexes
650
650
class TestQuerySetSlicing (BaseQuerySetUsage ):
651
651
652
652
@execute_count (1 )
@@ -701,7 +701,7 @@ def test_negative_slicing(self):
701
701
self .assertEqual (model .attempt_id , expect )
702
702
703
703
704
- @requirescollectionindexes
704
+ @requires_collection_indexes
705
705
class TestQuerySetValidation (BaseQuerySetUsage ):
706
706
707
707
def test_primary_key_or_index_must_be_specified (self ):
@@ -783,7 +783,7 @@ def test_custom_indexed_field_can_be_queried(self):
783
783
list (CustomIndexedTestModel .objects .filter (test_id = 1 , description = 'test' ))
784
784
785
785
786
- @requirescollectionindexes
786
+ @requires_collection_indexes
787
787
class TestQuerySetDelete (BaseQuerySetUsage ):
788
788
789
789
@execute_count (9 )
@@ -942,7 +942,7 @@ def test_success_case(self):
942
942
assert '4' in datas
943
943
944
944
945
- @requirescollectionindexes
945
+ @requires_collection_indexes
946
946
class TestInOperator (BaseQuerySetUsage ):
947
947
@execute_count (1 )
948
948
def test_kwarg_success_case (self ):
@@ -1003,7 +1003,7 @@ class bool_model2(Model):
1003
1003
1004
1004
1005
1005
@greaterthancass20
1006
- @requirescollectionindexes
1006
+ @requires_collection_indexes
1007
1007
class TestContainsOperator (BaseQuerySetUsage ):
1008
1008
1009
1009
@execute_count (6 )
@@ -1069,7 +1069,7 @@ def test_query_expression_success_case(self):
1069
1069
self .assertEqual (q .count (), 0 )
1070
1070
1071
1071
1072
- @requirescollectionindexes
1072
+ @requires_collection_indexes
1073
1073
class TestValuesList (BaseQuerySetUsage ):
1074
1074
1075
1075
@execute_count (2 )
@@ -1082,7 +1082,7 @@ def test_values_list(self):
1082
1082
assert item == 10
1083
1083
1084
1084
1085
- @requirescollectionindexes
1085
+ @requires_collection_indexes
1086
1086
class TestObjectsProperty (BaseQuerySetUsage ):
1087
1087
@execute_count (1 )
1088
1088
def test_objects_property_returns_fresh_queryset (self ):
@@ -1113,7 +1113,7 @@ class PagingTest(Model):
1113
1113
assert len (results ) == 2
1114
1114
1115
1115
1116
- @requirescollectionindexes
1116
+ @requires_collection_indexes
1117
1117
class ModelQuerySetTimeoutTestCase (BaseQuerySetUsage ):
1118
1118
def test_default_timeout (self ):
1119
1119
with mock .patch .object (Session , 'execute' ) as mock_execute :
@@ -1131,7 +1131,7 @@ def test_none_timeout(self):
1131
1131
self .assertEqual (mock_execute .call_args [- 1 ]['timeout' ], None )
1132
1132
1133
1133
1134
- @requirescollectionindexes
1134
+ @requires_collection_indexes
1135
1135
class DMLQueryTimeoutTestCase (BaseQuerySetUsage ):
1136
1136
def setUp (self ):
1137
1137
self .model = TestModel (test_id = 1 , attempt_id = 1 , description = 'timeout test' )
0 commit comments