Merge pull request #125 from simonsobs/dev

Update tests

tests/schema/mutations/auto_mode/test_change_mode.py

@@ -28,4 +28,4 @@ async def test_schema(mode: str, schema: Schema) -> None:
     expected = {'schedule': {'autoMode': {'changeMode': valid}}}
     assert DeepDiff(data, expected) == {}
     if valid:
-        auto_mode.change_mode.assert_called_once_with(mode)
+        auto_mode.change_mode.assert_awaited_once_with(mode)

tests/schema/mutations/auto_mode/test_turn_off.py

@@ -0,0 +1,16 @@
+from unittest.mock import Mock
+
+from nextline_schedule.auto import AutoMode
+from nextline_schedule.graphql import MUTATE_AUTO_MODE_TURN_OFF
+from tests.schema.conftest import Schema
+
+
+async def test_schema(schema: Schema) -> None:
+    auto_mode = Mock(spec=AutoMode)
+    context_auto_mode = {'auto_mode': auto_mode}
+    context_value = {'schedule': context_auto_mode}
+    resp = await schema.execute(MUTATE_AUTO_MODE_TURN_OFF, context_value=context_value)
+    assert (data := resp.data)
+    expected = {'schedule': {'autoMode': {'turnOff': True}}}
+    assert data == expected
+    auto_mode.turn_off.assert_awaited_once()

tests/schema/mutations/auto_mode/test_turn_on.py

@@ -0,0 +1,16 @@
+from unittest.mock import Mock
+
+from nextline_schedule.auto import AutoMode
+from nextline_schedule.graphql import MUTATE_AUTO_MODE_TURN_ON
+from tests.schema.conftest import Schema
+
+
+async def test_schema(schema: Schema) -> None:
+    auto_mode = Mock(spec=AutoMode)
+    context_auto_mode = {'auto_mode': auto_mode}
+    context_value = {'schedule': context_auto_mode}
+    resp = await schema.execute(MUTATE_AUTO_MODE_TURN_ON, context_value=context_value)
+    assert (data := resp.data)
+    expected = {'schedule': {'autoMode': {'turnOn': True}}}
+    assert data == expected
+    auto_mode.turn_on.assert_awaited_once()

tests/schema/mutations/queue/move/test_one_backward.py

@@ -31,4 +31,4 @@ async def test_schema(d: st.DataObject, schema: Schema) -> None:
     expected_data = {'schedule': {'queue': {'move': {'oneBackward': return_value}}}}
     assert DeepDiff(data, expected_data) == {}
 
-    queue.move_one_backward.assert_called_once_with(id)
+    queue.move_one_backward.assert_awaited_once_with(id)

tests/schema/mutations/queue/move/test_one_forward.py

@@ -31,4 +31,4 @@ async def test_schema(d: st.DataObject, schema: Schema) -> None:
     expected_data = {'schedule': {'queue': {'move': {'oneForward': return_value}}}}
     assert DeepDiff(data, expected_data) == {}
 
-    queue.move_one_forward.assert_called_once_with(id)
+    queue.move_one_forward.assert_awaited_once_with(id)

tests/schema/mutations/queue/move/test_to_first.py

@@ -31,4 +31,4 @@ async def test_schema(d: st.DataObject, schema: Schema) -> None:
     expected_data = {'schedule': {'queue': {'move': {'toFirst': return_value}}}}
     assert DeepDiff(data, expected_data) == {}
 
-    queue.move_to_first.assert_called_once_with(id)
+    queue.move_to_first.assert_awaited_once_with(id)

tests/schema/mutations/queue/move/test_to_last.py

@@ -31,4 +31,4 @@ async def test_schema(d: st.DataObject, schema: Schema) -> None:
     expected_data = {'schedule': {'queue': {'move': {'toLast': return_value}}}}
     assert DeepDiff(data, expected_data) == {}
 
-    queue.move_to_last.assert_called_once_with(id)
+    queue.move_to_last.assert_awaited_once_with(id)

tests/schema/mutations/queue/test_push.py

@@ -60,4 +60,4 @@ async def test_schema(d: st.DataObject, schema: Schema) -> None:
     }
     assert DeepDiff(data, expected_data) == {}
 
-    queue.push.assert_called_once_with(PushArg(**input))
+    queue.push.assert_awaited_once_with(PushArg(**input))

tests/schema/mutations/queue/test_remove.py

@@ -31,4 +31,4 @@ async def test_schema(d: st.DataObject, schema: Schema) -> None:
     expected_data = {'schedule': {'queue': {'remove': return_value}}}
     assert DeepDiff(data, expected_data) == {}
 
-    queue.remove.assert_called_once_with(id)
+    queue.remove.assert_awaited_once_with(id)

tests/schema/queries/auto_mode/test_mode.py

@@ -12,5 +12,6 @@ async def test_schema(schema: Schema) -> None:
     context_auto_mode = {'auto_mode': auto_mode}
     context = {'schedule': context_auto_mode}
     resp = await schema.execute(QUERY_AUTO_MODE_MODE, context_value=context)
-    assert resp.data
-    assert resp.data['schedule']['autoMode']['mode'] == mode
+    assert (data := resp.data)
+    expected = {'schedule': {'autoMode': {'mode': mode}}}
+    assert data == expected

tests/schema/queries/auto_mode/test_state.py

@@ -12,5 +12,6 @@ async def test_schema(schema: Schema) -> None:
     context_auto_mode = {'auto_mode': auto_mode}
     context = {'schedule': context_auto_mode}
     resp = await schema.execute(QUERY_AUTO_MODE_STATE, context_value=context)
-    assert resp.data
-    assert resp.data['schedule']['autoMode']['state'] == state
+    assert (data := resp.data)
+    expected = {'schedule': {'autoMode': {'state': state}}}
+    assert data == expected

tests/schema/queries/scheduler/test_scheduler.py

@@ -0,0 +1,37 @@
+from string import ascii_letters
+from unittest.mock import Mock
+
+from hypothesis import given, provisional
+from hypothesis import strategies as st
+
+from nextline_schedule.graphql import QUERY_SCHEDULER
+from nextline_schedule.scheduler import Scheduler
+from nextline_test_utils.strategies import st_graphql_ints
+from tests.schema.conftest import Schema
+
+
+@given(d=st.data())
+async def test_schema(d: st.DataObject, schema: Schema) -> None:
+    api_url = d.draw(provisional.urls())
+    length_minutes = d.draw(st_graphql_ints(min_value=1))
+    policy = d.draw(st.text(ascii_letters, min_size=1))
+
+    scheduler = Mock(spec=Scheduler)
+    scheduler._api_url = api_url
+    scheduler._length_minutes = length_minutes
+    scheduler._policy = policy
+
+    context = {'schedule': {'scheduler': scheduler}}
+    resp = await schema.execute(QUERY_SCHEDULER, context_value=context)
+    assert (data := resp.data)
+
+    expected = {
+        'schedule': {
+            'scheduler': {
+                'apiUrl': api_url,
+                'lengthMinutes': length_minutes,
+                'policy': policy,
+            }
+        }
+    }
+    assert data == expected