def test_count(self):
ss = SelectStatement('table', count=True, limit=10, order_by='d')
ss.add_where(Column(db_field='a'), EqualsOperator(), 'b')
self.assertEqual(
six.text_type(ss),
'SELECT COUNT(*) FROM table WHERE "a" = %(0)s LIMIT 10',
six.text_type(ss))
self.assertIn('LIMIT', six.text_type(ss))
self.assertNotIn('ORDER', six.text_type(ss))
def test_context_id_update(self):
""" tests that the right things happen the the context id """
ss = SelectStatement('table')
ss.add_where(Column(db_field='a'), EqualsOperator(), 'b')
self.assertEqual(ss.get_context(), {'0': 'b'})
self.assertEqual(str(ss), 'SELECT * FROM table WHERE "a" = %(0)s')
ss.update_context_id(5)
self.assertEqual(ss.get_context(), {'5': 'b'})
self.assertEqual(str(ss), 'SELECT * FROM table WHERE "a" = %(5)s')
def test_distinct(self):
ss = SelectStatement('table', distinct_fields=['field2'])
ss.add_where(Column(db_field='field1'), EqualsOperator(), 'b')
self.assertEqual(six.text_type(ss), 'SELECT DISTINCT "field2" FROM table WHERE "field1" = %(0)s', six.text_type(ss))
ss = SelectStatement('table', distinct_fields=['field1', 'field2'])
self.assertEqual(six.text_type(ss), 'SELECT DISTINCT "field1", "field2" FROM table')
ss = SelectStatement('table', distinct_fields=['field1'], count=True)
self.assertEqual(six.text_type(ss), 'SELECT DISTINCT COUNT("field1") FROM table')
def test_context_id_update(self):
""" tests that the right things happen the the context id """
ss = SelectStatement('table')
ss.add_where(Column(db_field='a'), EqualsOperator(), 'b')
self.assertEqual(ss.get_context(), {'0': 'b'})
self.assertEqual(str(ss), 'SELECT * FROM table WHERE "a" = %(0)s')
ss.update_context_id(5)
self.assertEqual(ss.get_context(), {'5': 'b'})
self.assertEqual(str(ss), 'SELECT * FROM table WHERE "a" = %(5)s')
def test_distinct(self):
ss = SelectStatement('table', distinct_fields=['field2'])
ss.add_where(Column(db_field='field1'), EqualsOperator(), 'b')
self.assertEqual(six.text_type(ss), 'SELECT DISTINCT "field2" FROM table WHERE "field1" = %(0)s', six.text_type(ss))
ss = SelectStatement('table', distinct_fields=['field1', 'field2'])
self.assertEqual(six.text_type(ss), 'SELECT DISTINCT "field1", "field2" FROM table')
ss = SelectStatement('table', distinct_fields=['field1'], count=True)
self.assertEqual(six.text_type(ss), 'SELECT DISTINCT COUNT("field1") FROM table')
def test_like_operator(self):
"""
Test to verify the like operator works appropriately
@since 3.13
@jira_ticket PYTHON-512
@expected_result the expected row is read using LIKE
@test_category data_types:object_mapper
"""
cluster = TestCluster()
session = cluster.connect()
self.addCleanup(cluster.shutdown)
session.execute("""CREATE CUSTOM INDEX text_index ON {} (text)
USING 'org.apache.cassandra.index.sasi.SASIIndex';"""
.format(self.table_name))
self.addCleanup(session.execute,
"DROP INDEX {}.text_index".format(DEFAULT_KEYSPACE))
partition = uuid4()
cluster = 1
self._insert_statement(partition, cluster)
ss = SelectStatement(self.table_name)
like_clause = "text_for_%"
ss.add_where(Column(db_field='text'), LikeOperator(), like_clause)
self.assertEqual(
six.text_type(ss),
'SELECT * FROM {} WHERE "text" LIKE %(0)s'.format(self.table_name))
result = execute(ss)
self.assertEqual(result[0]["text"], self.text)
q = TestQueryUpdateModel.objects.filter(
text__like=like_clause).allow_filtering()
self.assertEqual(q[0].text, self.text)
q = TestQueryUpdateModel.objects.filter(text__like=like_clause)
self.assertEqual(q[0].text, self.text)
def test_like_operator(self):
"""
Test to verify the like operator works appropriately
@since 3.13
@jira_ticket PYTHON-512
@expected_result the expected row is read using LIKE
@test_category data_types:object_mapper
"""
cluster = Cluster()
session = cluster.connect()
self.addCleanup(cluster.shutdown)
session.execute("""CREATE CUSTOM INDEX text_index ON {} (text)
USING 'org.apache.cassandra.index.sasi.SASIIndex';""".format(self.table_name))
self.addCleanup(session.execute, "DROP INDEX {}.text_index".format(DEFAULT_KEYSPACE))
partition = uuid4()
cluster = 1
self._insert_statement(partition, cluster)
ss = SelectStatement(self.table_name)
like_clause = "text_for_%"
ss.add_where(Column(db_field='text'), LikeOperator(), like_clause)
self.assertEqual(six.text_type(ss),
'SELECT * FROM {} WHERE "text" LIKE %(0)s'.format(self.table_name))
result = execute(ss)
self.assertEqual(result[0]["text"], self.text)
q = TestQueryUpdateModel.objects.filter(text__like=like_clause).allow_filtering()
self.assertEqual(q[0].text, self.text)
q = TestQueryUpdateModel.objects.filter(text__like=like_clause)
self.assertEqual(q[0].text, self.text)
def test_context(self):
ss = SelectStatement('table')
ss.add_where(Column(db_field='a'), EqualsOperator(), 'b')
self.assertEqual(ss.get_context(), {'0': 'b'})
def test_count(self):
ss = SelectStatement('table', count=True, limit=10, order_by='d')
ss.add_where(Column(db_field='a'), EqualsOperator(), 'b')
self.assertEqual(six.text_type(ss), 'SELECT COUNT(*) FROM table WHERE "a" = %(0)s LIMIT 10', six.text_type(ss))
self.assertIn('LIMIT', six.text_type(ss))
self.assertNotIn('ORDER', six.text_type(ss))
def test_where_clause_rendering(self):
ss = SelectStatement('table')
ss.add_where(Column(db_field='a'), EqualsOperator(), 'b')
self.assertEqual(six.text_type(ss), 'SELECT * FROM table WHERE "a" = %(0)s', six.text_type(ss))
def test_context(self):
ss = SelectStatement('table')
ss.add_where(Column(db_field='a'), EqualsOperator(), 'b')
self.assertEqual(ss.get_context(), {'0': 'b'})
def test_where_clause_rendering(self):
ss = SelectStatement('table')
ss.add_where(Column(db_field='a'), EqualsOperator(), 'b')
self.assertEqual(six.text_type(ss),
'SELECT * FROM table WHERE "a" = %(0)s',
six.text_type(ss))
