def test_datapump_nr_requires_consent_flag(app, mocker, consent_flag,
state_cd):
# create minimal NR to send to NRO
nr = Request()
nr.nrNum = 'NR 0000001'
nr.stateCd = state_cd
nr.consentFlag = consent_flag
nr.lastUpdate = datetime(1970,
1,
1,
00,
00,
tzinfo=timezone('US/Pacific', ))
# requires the username
user = User('idir/bob', 'bob', 'last', 'idir', 'localhost')
nr.activeUser = user
# add name(s) to the NR - max 3
for i in range(1, 4):
name = Name()
name.state = Name.APPROVED if i == 1 else Name.NOT_EXAMINED
name.name = 'sample name {}'.format(i)
name.choice = i
name.decision_text = 'All good to go {}'.format(i)
nr.names.append(name)
# mock the oracle cursor
oc = mocker.MagicMock()
# make the real call
nro_data_pump_update(nr, ora_cursor=oc, expires_days=60)
oc.callfunc.assert_called_with(
'NRO_DATAPUMP_PKG.name_examination_func', # package.func_name
str,
[
'NR 0000001', # p_nr_number
'A', # p_status
'19700302', # p_expiry_date (length=8)
'Y', # p_consent_flag
'bob', # p_examiner_id (anything length <=7)
'A****All good to go 1', # p_choice1
None, # p_choice2
None, # p_choice3
None, # p_exam_comment
'', # p_add_info - not used in proc anymore
None, # p_confname1A
None, # p_confname1B
None, # p_confname1C
None, # p_confname2A
None, # p_confname2B
None, # p_confname2C
None, # p_confname3A
None, # p_confname3B
None
]) # p_confname3C
def test_datapump(app, mocker, start_date, expected_date):
# create minimal NR to send to NRO
nr = Request()
nr.nrNum = 'NR 0000001'
nr.stateCd = State.REJECTED
nr.consentFlag = 'N'
nr.lastUpdate = start_date
# requires the username
user = User('idir/bob', 'bob', 'last', 'idir', 'localhost')
nr.activeUser = user
# add name(s) to the NR - max 3
for i in range(1, 4):
name = Name()
name.state = Name.REJECTED
name.name = 'sample name {}'.format(i)
name.choice = i
name.decision_text = 'No Distinctive Term {}'.format(i)
nr.names.append(name)
# mock the oracle cursor
oc = mocker.MagicMock()
# make the real call
nro_data_pump_update(nr, ora_cursor=oc, expires_days=56)
oc.callfunc.assert_called_with(
'NRO_DATAPUMP_PKG.name_examination_func', # package.func_name
str,
[
'NR 0000001', # p_nr_number
'R', # p_status
expected_date.strftime('%Y%m%d'), # p_expiry_date (length=8)
'N', # p_consent_flag
'bob', # p_examiner_id (anything length <=7)
'R****No Distinctive Term 1', # p_choice1
'R****No Distinctive Term 2', # p_choice2
'R****No Distinctive Term 3', # p_choice3
None, # p_exam_comment
'', # p_add_info - not used in proc anymore
None, # p_confname1A
None, # p_confname1B
None, # p_confname1C
None, # p_confname2A
None, # p_confname2B
None, # p_confname2C
None, # p_confname3A
None, # p_confname3B
None
]) # p_confname3C
# leaving this debug statement here as there were some translation and image caching issues
# that are easier to see from the raw SQL in the log
#
current_app.logger.debug(str(q.statement.compile(
dialect=postgresql.dialect(),
compile_kwargs={"literal_binds": True}))
)
for r in q.all():
row_count += 1
current_app.logger.debug('processing: {}'.format(r.nrNum))
try:
nro_data_pump_update(r, ora_cursor, expires_days)
db.session.add(r)
EventRecorder.record(user, Event.NRO_UPDATE, r, r.json(), save_to_session=True)
ora_con.commit()
db.session.commit()
JobTracker.job_detail(db, job_id, r.nrNum)
except Exception as err:
current_app.logger.error(err)
current_app.logger.error('ERROR: {}'.format(r.nrNum))
db.session.rollback()
ora_con.rollback()
JobTracker.job_detail_error(db, job_id, r.nrNum, str(err))
JobTracker.end_job(db, job_id, datetime.utcnow(), 'success')
