from django.test import SimpleTestCase

from apps.bi.serializers import (
    DataExecSerializer,
    DatasetListResponseSerializer,
    DatasetSerializer,
)
from apps.bi.views import DatasetViewSet


class DatasetAgentDiscoveryTests(SimpleTestCase):
    def test_dataset_catalog_searches_description(self):
        self.assertEqual(
            DatasetViewSet.search_fields,
            ["name", "code", "description"],
        )

    def test_dataset_schema_explains_discovery_and_exec_parameters(self):
        dataset = DatasetSerializer()
        execute = DataExecSerializer()

        self.assertIn("适合回答的问题", dataset.fields["description"].help_text)
        self.assertIn("业务参数", execute.fields["query"].help_text)
        self.assertTrue(execute.fields["raise_exception"].default)
        self.assertIn("results", DatasetListResponseSerializer().fields)