26 lines
885 B
Python
26 lines
885 B
Python
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)
|