SQL: Test AUTO_DL_LIMIT

Author: VesnaT

Orange/widgets/data/tests/test_owsql.py

@@ -9,6 +9,10 @@
 from Orange.widgets.tests.base import WidgetTest, simulate
 from Orange.tests.sql.base import DataBaseTest as dbt
 
+mock_msgbox = mock.MagicMock()
+mock_msgbox().addButton.return_value = "NO"
+mock_msgbox().clickedButton.return_value = "NO"
+
 
 def mock_sqltable(*args, **_):
     table = Table(args[1])
@@ -173,6 +177,36 @@ def test_table_text(self, mocked_backends: mock.Mock):
         self.assertIsInstance(output, Table)
         self.assertEqual(len(output), 101)
 
+    @mock.patch('Orange.widgets.data.owsql.AUTO_DL_LIMIT', 120)
+    @mock.patch('Orange.widgets.data.owsql.is_postgres',
+                mock.Mock(return_value=True))
+    @mock.patch('Orange.widgets.data.owsql.QMessageBox', mock_msgbox)
+    @mock.patch('Orange.widgets.data.owsql.SqlTable',
+                mock.Mock(side_effect=mock_sqltable))
+    @mock.patch('Orange.widgets.data.owsql.Backend')
+    def test_auto_dl_limit(self, mocked_backends: mock.Mock):
+        backend = mock.Mock()
+        backend().display_name = "Dummy Backend"
+        backend().list_tables.return_value = ["iris", "zoo", "titanic"]
+        backend().n_tables.return_value = 3
+        mocked_backends.available_backends.return_value = [backend]
+
+        settings = {"selected_backend": "Dummy Backend",
+                    "host": "host", "port": "port", "database": "DB",
+                    "schema": "", "username": "username",
+                    "password": "password"}
+        widget: OWSql = self.create_widget(OWSql, stored_settings=settings)
+        widget.tablecombo.setCurrentIndex(2)
+        widget.select_table()
+        output = self.get_output(widget.Outputs.data, widget=widget)
+        self.assertIsInstance(output, Table)
+        self.assertEqual(len(output), 101)
+
+        widget.tablecombo.setCurrentIndex(1)
+        widget.select_table()
+        output = self.get_output(widget.Outputs.data, widget=widget)
+        self.assertIsNone(output)
+
 
 if __name__ == "__main__":
     unittest.main()