async def test_finish_deployment_cb(gstate: GlobalState, mocker: MockerFixture, nodemgr: NodeMgr) -> None: from gravel.controllers.nodes.mgr import NodeInitStage nodemgr._init_stage = NodeInitStage.NONE assert await expect_assertion(nodemgr._finish_deployment(True, None)) nodemgr._init_stage = NodeInitStage.PREPARE assert await expect_assertion(nodemgr._finish_deployment(True, None)) nodemgr._init_stage = NodeInitStage.STARTED assert await expect_assertion(nodemgr._finish_deployment(True, None)) nodemgr._init_stage = NodeInitStage.AVAILABLE nodemgr._deployment.finish_deployment = mocker.MagicMock() nodemgr._load = mocker.AsyncMock() nodemgr._node_start = mocker.AsyncMock() await nodemgr._finish_deployment(True, None) nodemgr._deployment.finish_deployment.assert_called_once() # type: ignore nodemgr._load.assert_called_once() # type: ignore nodemgr._node_start.assert_called_once() # type: ignore
async def test_node_start( gstate: GlobalState, mocker: MockerFixture, fs: fake_filesystem.FakeFilesystem, nodemgr: NodeMgr, ) -> None: from gravel.controllers.nodes.deployment import NodeStageEnum from gravel.controllers.nodes.mgr import NodeInitStage nodemgr._deployment._state._stage = NodeStageEnum.READY nodemgr._obtain_state = mocker.AsyncMock() nodemgr._load = mocker.AsyncMock() nodemgr._incoming_msg_task = mocker.AsyncMock() nodemgr._connmgr.start_receiving = mocker.MagicMock() await nodemgr._node_start() assert nodemgr._init_stage == NodeInitStage.STARTED nodemgr._obtain_state.assert_called_once() # type: ignore nodemgr._load.assert_called_once() # type: ignore nodemgr._incoming_msg_task.assert_called_once() # type: ignore nodemgr._connmgr.start_receiving.assert_called_once() # type: ignore