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_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))