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
