Initial support for or(|) operator in node state

Signed-off-by: Theofilos Manitaras <manitaras@cscs.ch>

Author: Theofilos Manitaras

reframe/core/schedulers/__init__.py

@@ -165,7 +165,14 @@ def filter_nodes_by_state(nodelist, state):
         state string.
     :returns: the filtered node list
     '''
-    if state == 'avail':
+    if '|' in state:
+        allowed_states = state.split('|')
+        final_nodelist = set()
+        for s in allowed_states:
+            final_nodelist.update(filter_nodes_by_state(nodelist, s))
+
+        nodelist = final_nodelist
+    elif state == 'avail':
         nodelist = {n for n in nodelist if n.is_avail()}
     elif state != 'all':
         if state.endswith('*'):

unittests/test_schedulers.py

@@ -1247,6 +1247,19 @@ def test_flex_alloc_not_enough_nodes_constraint_expr(make_flexible_job):
         prepare_job(job)
 
 
+def test_flex_alloc_allocated_or_idle_partition(make_flexible_job):
+    job = make_flexible_job('allocated|idle')
+    job.options = ['--partition=p3']
+    prepare_job(job)
+    assert job.num_tasks == 12
+
+
+def test_flex_alloc_maint_star_or_allocated(make_flexible_job):
+    job = make_flexible_job('maint*|idle')
+    prepare_job(job)
+    assert job.num_tasks == 16
+
+
 @pytest.fixture
 def slurm_node_allocated():
     return _SlurmNode(