milvus/tests/restful_client_v2/testcases/test_database_operation.py

import pytest
from base.testbase import TestBase
from utils.utils import gen_unique_str


@pytest.mark.L0
class TestDatabaseOperation(TestBase):
    """
    Test cases for database operations
    """

    def test_create_database_with_default_properties(self):
        """
        Test creating a database with default properties
        """
        db_name = f"test_db_{gen_unique_str()}"
        payload = {"dbName": db_name}
        rsp = self.database_client.database_create(payload)
        assert rsp["code"] == 0

        # Verify database exists
        list_rsp = self.database_client.database_list({})
        assert rsp["code"] == 0
        assert db_name in list_rsp["data"]

    def test_create_database_with_custom_properties(self):
        """
        Test creating a database with custom properties
        """
        db_name = f"test_db_{gen_unique_str()}"
        payload = {"dbName": db_name, "properties": {"mmap.enabled": True}}
        rsp = self.database_client.database_create(payload)
        assert rsp["code"] == 0

        # Verify properties
        describe_rsp = self.database_client.database_describe({"dbName": db_name})
        assert describe_rsp["code"] == 0
        assert any(
            prop["key"] == "mmap.enabled" and prop["value"] == "true"
            for prop in describe_rsp["data"]["properties"]
        )

    def test_alter_database_properties(self):
        """
        Test altering database properties
        """
        db_name = f"test_db_{gen_unique_str()}"

        # Create database with initial properties
        create_payload = {"dbName": db_name, "properties": {"mmap.enabled": True}}
        rsp = self.database_client.database_create(create_payload)
        assert rsp["code"] == 0
        # Verify properties
        describe_rsp = self.database_client.database_describe({"dbName": db_name})
        assert describe_rsp["code"] == 0
        assert any(
            prop["key"] == "mmap.enabled" and prop["value"] == "true"
            for prop in describe_rsp["data"]["properties"]
        )

        # Alter properties
        alter_payload = {"dbName": db_name, "properties": {"mmap.enabled": False}}
        alter_rsp = self.database_client.database_alter(alter_payload)
        assert alter_rsp["code"] == 0

        # Verify altered properties
        describe_rsp = self.database_client.database_describe({"dbName": db_name})
        assert describe_rsp["code"] == 0
        assert any(
            prop["key"] == "mmap.enabled" and prop["value"] == "false"
            for prop in describe_rsp["data"]["properties"]
        )

    def test_list_databases(self):
        """
        Test listing databases
        """
        # Create test database
        db_name = f"test_db_{gen_unique_str()}"
        self.database_client.database_create({"dbName": db_name})

        # List databases
        rsp = self.database_client.database_list({})
        assert rsp["code"] == 0
        assert "default" in rsp["data"]  # Default database should always exist
        assert db_name in rsp["data"]

    def test_describe_database(self):
        """
        Test describing database
        """
        db_name = f"test_db_{gen_unique_str()}"
        properties = {"mmap.enabled": True}

        # Create database
        self.database_client.database_create(
            {"dbName": db_name, "properties": properties}
        )

        # Describe database
        rsp = self.database_client.database_describe({"dbName": db_name})
        assert rsp["code"] == 0
        assert rsp["data"]["dbName"] == db_name
        assert "dbID" in rsp["data"]
        assert len(rsp["data"]["properties"]) > 0


@pytest.mark.L0
class TestDatabaseOperationNegative(TestBase):
    """
    Negative test cases for database operations
    """

    def test_create_database_with_invalid_name(self):
        """
        Test creating database with invalid name
        """
        invalid_names = ["", " ", "test db", "test/db", "test\\db"]
        for name in invalid_names:
            rsp = self.database_client.database_create({"dbName": name})
            assert rsp["code"] != 0

    def test_create_duplicate_database(self):
        """
        Test creating database with duplicate name
        """
        db_name = f"test_db_{gen_unique_str()}"

        # Create first database
        rsp1 = self.database_client.database_create({"dbName": db_name})
        assert rsp1["code"] == 0

        # Try to create duplicate
        rsp2 = self.database_client.database_create({"dbName": db_name})
        assert rsp2["code"] != 0

    def test_describe_non_existent_database(self):
        """
        Test describing non-existent database
        """
        rsp = self.database_client.database_describe({"dbName": "non_existent_db"})
        assert rsp["code"] != 0

    def test_alter_non_existent_database(self):
        """
        Test altering non-existent database
        """
        payload = {"dbName": "non_existent_db", "properties": {"mmap.enabled": False}}
        rsp = self.database_client.database_alter(payload)
        assert rsp["code"] != 0

    def test_drop_non_existent_database(self):
        """
        Test dropping non-existent database
        """
        rsp = self.database_client.database_drop({"dbName": "non_existent_db"})
        assert rsp["code"] == 0

    def test_drop_default_database(self):
        """
        Test dropping default database (should not be allowed)
        """
        rsp = self.database_client.database_drop({"dbName": "default"})
        assert rsp["code"] != 0