def test_onboard_pnf_load(mock_create, mock_submit, mock_certify, mock_load): getter_mock = mock.Mock(wraps=Pnf.status.fget) mock_status = Pnf.status.getter(getter_mock) with mock.patch.object(Pnf, 'status', mock_status): getter_mock.side_effect = [ const.CERTIFIED, const.CERTIFIED, const.CERTIFIED, const.CERTIFIED, const.APPROVED, const.APPROVED, const.APPROVED ] pnf = Pnf() pnf._time_wait = 0 pnf.onboard() mock_create.assert_not_called() mock_submit.assert_not_called() mock_certify.assert_not_called() mock_load.assert_called_once()
def test_onboard_whole_pnf_vendor(mock_certify, mock_create, mock_submit, mock_load): """Test onboarding with vendor""" getter_mock = mock.Mock(wraps=Pnf.status.fget) mock_status = Pnf.status.getter(getter_mock) with mock.patch.object(Pnf, 'status', mock_status): getter_mock.side_effect = [ None, const.DRAFT, const.DRAFT, const.CHECKED_IN, const.CHECKED_IN, const.CHECKED_IN, const.CERTIFIED, const.CERTIFIED, const.CERTIFIED, const.CERTIFIED, const.APPROVED, const.APPROVED, const.APPROVED, const.APPROVED ] vendor = Vendor() pnf = Pnf(vendor=vendor) pnf._time_wait = 0 pnf.onboard() mock_create.assert_called_once() mock_submit.assert_called_once() mock_load.assert_called_once() mock_certify.assert_called_once()