Files
scylladb/test/cqlpy/test_use.py
Avi Kivity f3eade2f62 treewide: relicense to ScyllaDB-Source-Available-1.0
Drop the AGPL license in favor of a source-available license.
See the blog post [1] for details.

[1] https://www.scylladb.com/2024/12/18/why-were-moving-to-a-source-available-license/
2024-12-18 17:45:13 +02:00

35 lines
1.5 KiB
Python

# Copyright 2022-present ScyllaDB
#
# SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
#############################################################################
# Tests for the "USE" statement, which modifies the default keyspace used
# by subsequent statements.
#
# Note that because the "USE" statement modifies the state of the current
# connection, and there is no way to undo its effects (there is no "UNUSE"
# or way to do an empty "USE"), the following tests should all use a new_cql
# wrapper over the cql fixture, instead of the cql fixture directly. This
# wrapper creates a new connection, with its own default USE.
#############################################################################
import pytest
from cassandra.protocol import InvalidRequest
from .util import unique_name, new_cql
# Check that CREATE TABLE and DROP TABLE work without an explicit keyspace
# name if a default keyspace name is specified with "USE".
def test_create_table_use_keyspace(cql, test_keyspace):
with new_cql(cql) as ncql:
ncql.execute(f'USE {test_keyspace}')
table = unique_name()
ncql.execute(f'CREATE TABLE {table} (k int PRIMARY KEY)')
ncql.execute(f'DROP TABLE {table}')
# Check that without a USE, one cannot CREATE TABLE if the keyspace is not
# explicitly specified.
def test_create_table_no_keyspace(cql, test_keyspace):
with new_cql(cql) as ncql:
with pytest.raises(InvalidRequest, match='No keyspace'):
ncql.execute(f'CREATE TABLE {unique_name()} (k int PRIMARY KEY)')