[widgets][experimenteditor] Implement queue selection

The selection can be done for both remote and local
development.

beat/editor/test/conftest.py

@@ -95,8 +95,20 @@ def test_prefix():
             json.dumps(
                 {
                     "remote": [
-                        {"name": "Python 2.7", "version": "1.3.0"},
-                        {"name": "Python", "version": "2.0.0"},
+                        {
+                            "name": "Python 2.7",
+                            "version": "1.3.0",
+                            "queues": {"queue1": {}, "queue2": {}},
+                        },
+                        {
+                            "name": "Python",
+                            "version": "2.0.0",
+                            "queues": {
+                                "queue_extra_long": {},
+                                "queue_long": {},
+                                "queue_short": {},
+                            },
+                        },
                     ],
                     "docker": {
                         "Docker test env": {"name": "Pytorch", "version": "1.1.1"},

beat/editor/test/test_experimenteditor.py

@@ -27,7 +27,6 @@ import copy
 import pytest
 
 from PyQt5.QtCore import Qt
-from PyQt5.QtCore import QStringListModel
 
 from PyQt5.QtWidgets import QComboBox
 from PyQt5.QtWidgets import QCheckBox
@@ -62,6 +61,7 @@ from ..widgets.experimenteditor import typed_user_property
 
 from ..widgets.experimenteditor import ExperimentResources
 from ..widgets.experimenteditor import AlgorithmResourceModel
+from ..widgets.experimenteditor import QueueResourceModel
 
 from .conftest import prefix
 from .conftest import sync_prefix
@@ -287,6 +287,40 @@ class TestAlgorithmResourceModel:
         assert algorithm_model.rowCount() == query.value("cnt")
 
 
+class TestQueueResourceModel:
+    """Test the model used to generate suitable algorithm selections"""
+
+    @pytest.fixture
+    def prefix_model(self, beat_context):
+        return ExperimentResources(beat_context)
+
+    def test_default(self, prefix_model):
+        model = QueueResourceModel()
+
+        query = QSqlQuery()
+        assert query.exec_("SELECT COUNT(name) AS cnt FROM queues")
+
+        query.next()
+
+        assert model.rowCount() > 0
+        assert model.rowCount() == query.value("cnt")
+
+    def test_types(self, prefix_model):
+        model = QueueResourceModel()
+        for type_ in ["remote", "docker"]:
+            model.setType(type_)
+
+            query = QSqlQuery()
+            assert query.exec_(
+                f"SELECT COUNT(name) AS cnt FROM queues WHERE env_type='{type_}'"
+            )
+
+            query.next()
+
+            assert model.rowCount() > 0
+            assert model.rowCount() == query.value("cnt")
+
+
 class TestIOMapperDialog:
     """Test that the dialog used for input/output mapping works as expected"""
 
@@ -735,7 +769,6 @@ class TestExecutionPropertiesEditor(PropertiesEditorTestMixin, ParameterTestMixi
         editor = self.editor_klass(test_prefix)
         editor.setAlgorithmResourceModel(algorithm_model)
         editor.setEnvironmentModel(environment_model)
-        editor.setQueueModel(QStringListModel(["Test"]))
         return editor
 
 
@@ -752,7 +785,6 @@ class TestBlockEditor(TestExecutionPropertiesEditor):
 
         editor = self.editor_klass("block_name", test_prefix)
         editor.setEnvironmentModel(environment_model)
-        editor.setQueueModel(QStringListModel(["Test"]))
         return editor
 
 
@@ -769,7 +801,6 @@ class TestAnalyzerBlockEditor(PropertiesEditorTestMixin):
 
         editor = self.editor_klass("block_name", test_prefix)
         editor.setEnvironmentModel(environment_model)
-        editor.setQueueModel(QStringListModel(["Test"]))
         return editor
 
 
@@ -786,7 +817,6 @@ class TestLoopBlockEditor(TestExecutionPropertiesEditor):
 
         editor = self.editor_klass("block_name", test_prefix)
         editor.setEnvironmentModel(environment_model)
-        editor.setQueueModel(QStringListModel(["Test"]))
         return editor
 
     def test_edit_environment(
@@ -1030,7 +1060,6 @@ class TestGlobalParametersEditor:
         environment_model = EnvironmentModel()
         environment_model.setContext(beat_context)
         editor.setEnvironmentModel(environment_model)
-        editor.setQueueModel(QStringListModel(["Test", "Test2"]))
         return editor
 
     @pytest.mark.parametrize(
@@ -1080,7 +1109,8 @@ class TestGlobalParametersEditor:
     def test_change_queue(self, qtbot, gpe_editor, exp_globals):
         gpe_editor.load(exp_globals)
         with qtbot.waitSignal(gpe_editor.dataChanged):
-            gpe_editor.queue_combobox.setCurrentIndex(1)
+            index = gpe_editor.queue_combobox.currentIndex()
+            gpe_editor.queue_combobox.setCurrentIndex(change_index(index))
         assert gpe_editor.dump() != exp_globals