def test_onboard_vf_load(mock_create, mock_submit, mock_load): getter_mock = mock.Mock(wraps=Vf.status.fget) mock_status = Vf.status.getter(getter_mock) with mock.patch.object(Vf, 'status', mock_status): getter_mock.side_effect = [ const.CERTIFIED, const.CERTIFIED, const.CERTIFIED, const.APPROVED, const.APPROVED, const.APPROVED ] vf = Vf() vf._time_wait = 0 vf.onboard() mock_create.assert_not_called() mock_submit.assert_not_called() mock_load.assert_called_once()
def test_onboard_vf_submit(mock_certify, mock_add_resource, mock_create, mock_submit, mock_load): getter_mock = mock.Mock(wraps=Vf.status.fget) mock_status = Vf.status.getter(getter_mock) with mock.patch.object(Vf, 'status', mock_status): getter_mock.side_effect = [ const.DRAFT, const.DRAFT, const.CHECKED_IN, const.CHECKED_IN, const.CHECKED_IN, const.APPROVED, const.APPROVED, const.APPROVED, const.APPROVED ] vf = Vf() vf._time_wait = 0 vf.onboard() mock_create.assert_not_called() mock_add_resource.assert_not_called() mock_submit.assert_called_once() mock_load.assert_not_called() mock_certify.assert_called_once()