Refactor tests

Author: aclark4life

tests/mql_panel/test_forms.py

@@ -7,7 +7,7 @@
 from django_mongodb_extensions.mql_panel.forms import MQLQueryForm
 
 
-class ConvertDocumentsToTableTests(TestCase):
+class MQLPanelTests(TestCase):
     def setUp(self):
         self.form = MQLQueryForm()
 
@@ -17,31 +17,6 @@ def test_empty_documents(self):
         self.assertEqual(rows, [])
         self.assertEqual(headers, [])
 
-    def test_handle_operation_error_format(self):
-        """Error return format matches convert_documents_to_table format."""
-        error = ValueError("Test error")
-        mql_string = "db.test.aggregate([])"
-        rows, headers = self.form._handle_operation_error(
-            error, mql_string, "aggregate"
-        )
-
-        # Should return one row with one cell
-        self.assertEqual(len(rows), 1)
-        self.assertEqual(len(rows[0]), 1)
-
-        # Cell should be a dict with 'value' and 'is_json' keys
-        cell = rows[0][0]
-        self.assertIsInstance(cell["value"], str)
-        self.assertIs(cell["is_json"], False)
-
-        # Should have one header
-        self.assertEqual(headers[0], "Query Parsing Error")
-
-
-class FormatCellValueTests(TestCase):
-    def setUp(self):
-        self.form = MQLQueryForm()
-
     def test_simple_fields(self):
         """Primitive field values like in model query output."""
         cases = [

tests/mql_panel/test_panel.py

@@ -4,7 +4,6 @@
 from django.test import RequestFactory, TestCase, override_settings
 
 from django_mongodb_extensions.mql_panel import MQLPanel
-from django_mongodb_extensions.mql_panel.forms import MQLQueryForm
 
 rf = RequestFactory()
 
@@ -139,32 +138,3 @@ def test_query_width_ratio(self):
         self.panel.generate_stats(self.request, response)
         total_width = sum(q["width_ratio"] for q in self.panel._queries)
         self.assertAlmostEqual(total_width, 100, places=5)
-
-
-class ConvertDocumentsToTableTests(TestCase):
-    def setUp(self):
-        self.form = MQLQueryForm()
-
-    def test_empty_documents(self):
-        """Empty document list returns empty rows and headers."""
-        rows, headers = self.form.convert_documents_to_table([])
-        self.assertEqual(rows, [])
-        self.assertEqual(headers, [])
-
-    def test_handle_operation_error_format(self):
-        """Error return format matches convert_documents_to_table format."""
-        error = ValueError("Test error")
-        mql_string = "db.test.aggregate([])"
-        rows, headers = self.form._handle_operation_error(error, mql_string, "query")
-
-        # Should return one row with one cell
-        self.assertEqual(len(rows), 1)
-        self.assertEqual(len(rows[0]), 1)
-
-        # Cell should be a dict with 'value' and 'is_json' keys
-        cell = rows[0][0]
-        self.assertIsInstance(cell["value"], str)
-        self.assertIs(cell["is_json"], False)
-
-        # Should have one header
-        self.assertEqual(headers[0], "Query Parsing Error")