def testSetSubProviderResultList(self):
provider = create_user(get_random_username(),
"provider",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider.save()
nppa = NP_PA(user=provider)
nppa.save()
provider1 = create_user(get_random_username(),
"physician",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider1.save()
phy = Physician(user=provider1)
phy.specialty = 'AC'
phy.save()
self.assertEqual(0, len(setSubProviderResultList([], phy)),
'test failed for setSubProviderResultList')
self.assertEqual(1, len(setSubProviderResultList([nppa], phy)),
'test failed for setSubProviderResultList')
self.assertEqual(2, len(setSubProviderResultList([nppa, phy])),
'test failed for setSubProviderResultList')
def testSetSubProviderResultList(self): provider = create_user(get_random_username(), "provider", "thj", "demo", "555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider) provider.save() nppa = NP_PA(user=provider) nppa.save() provider1 = create_user(get_random_username(), "physician", "thj", "demo", "555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider) provider1.save() phy = Physician(user=provider1) phy.specialty = 'AC' phy.save() self.assertEqual(0, len(setSubProviderResultList([], phy)), 'test failed for setSubProviderResultList') self.assertEqual(1, len(setSubProviderResultList([nppa], phy)), 'test failed for setSubProviderResultList') self.assertEqual(2, len(setSubProviderResultList([nppa, phy])), 'test failed for setSubProviderResultList')
def testGetUserInfo(self):
mhlu = create_user(get_random_username(), "mhluser", "thj", "demo", "555 Bryant St.",
"Palo Alto", "CA", "")
mhlu.mdcom_phone = '9002000001'
mhlu.save()
mhlu_id = mhlu.id
return_data1 = {'last_name': u'thj', 'office_address1': u'555 Bryant St.',
'office_address2': u'', 'photo': '/media/images/photos/generic_128.png',
'specialty': '', 'mdcom_phone': '', 'office_city': u'Palo Alto',
'accepting_patients': False, 'id': mhlu_id, 'custom_logos': get_custom_logos(mhlu_id),
'first_name': u'mhluser', 'office_state': u'CA', 'office_zip': u'', 'staff_type': ''}
self.assertEqual(Http404, getUserInfo(mhlu_id + 1, None), 'test failed for getUserInfo')
self.assertEqual(return_data1, getUserInfo(mhlu_id), 'test failed for getUserInfo mhluser')
# create provider for test
provider = create_user(get_random_username(), "provider", "thj", "demo",
"555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider)
provider.mdcom_phone = '9002000002'
provider.save()
return_data2 = {'last_name': u'thj', 'office_address1': u'555 Bryant St.',
'office_address2': u'', 'photo': '/media/images/photos/avatar2.png',
'specialty': 'NP/PA/Midwife', 'mdcom_phone': u'9002000002', 'office_city': u'Palo Alto',
'accepting_patients': False, 'id': provider.user.id,
'custom_logos': get_custom_logos(mhlu_id), 'first_name': u'provider',
'office_state': u'CA', 'office_zip': u'', 'staff_type': ''}
self.assertEqual(return_data2, getUserInfo(provider.user.id), 'test failed for getUserInfo provider')
# create physician
provider1 = create_user(get_random_username(), "physician", "thj", "demo", "555 Bryant St.",
"Palo Alto", "CA", "", uklass=Provider)
provider1.mdcom_phone = '9002000002'
provider1.save()
phy = Physician(user=provider1)
phy.specialty = 'AC'
phy.save()
return_data3 = {'last_name': u'thj', 'office_address1': u'555 Bryant St.',
'office_address2': u'', 'photo': '/media/images/photos/avatar2.png',
'specialty': u'Acupuncture', 'mdcom_phone': u'9002000002',
'office_city': u'Palo Alto', 'accepting_patients': True,
'id': phy.user.user.id, 'custom_logos': get_custom_logos(mhlu_id),
'first_name': u'physician', 'office_state': u'CA', 'office_zip': u'',
'staff_type': ''}
self.assertEqual(return_data3, getUserInfo(phy.user.user.id), 'test failed for getUserInfo physician')
# create office staff
staff = create_office_staff(get_random_username(), "staff", "thj", "demo", "555 Bryant St.",
"Palo Alto", "CA", "", uklass=OfficeStaff)
return_data4 = {'last_name': u'thj', 'office_address1': u'555 Bryant St.',
'office_address2': u'', 'photo': '/media/images/photos/staff_icon.jpg',
'specialty': '', 'mdcom_phone': '', 'office_city': u'Palo Alto',
'accepting_patients': False, 'id': staff.user.id,
'custom_logos': get_custom_logos(mhlu_id), 'first_name': u'staff',
'office_state': u'CA', 'office_zip': u'', 'staff_type': 'Office Staff'}
self.assertEqual(return_data4, getUserInfo(staff.user.id), 'test failed for getUserInfo staff')
# create nurse
staff2 = create_office_staff(get_random_username(), "nurse", "thj", "demo",
"555 Bryant St.", "Palo Alto", "CA", "", uklass=OfficeStaff)
nurse = Nurse(user=staff2)
nurse.save()
return_data5 = {'last_name': u'thj', 'office_address1': u'555 Bryant St.',
'office_address2': u'', 'photo': '/media/images/photos/nurse.jpg',
'specialty': '', 'mdcom_phone': '', 'office_city': u'Palo Alto',
'accepting_patients': False, 'id': nurse.user.user.id,
'custom_logos': get_custom_logos(mhlu_id), 'first_name': u'nurse',
'office_state': u'CA', 'office_zip': u'', 'staff_type': 'Office Staff'}
self.assertEqual(return_data5, getUserInfo(nurse.user.user.id), 'test failed for getUserInfo nurse')
# create office staff
staff3 = create_office_staff(get_random_username(), "maneger", "thj", "demo",
"555 Bryant St.", "Palo Alto", "CA", "", uklass=OfficeStaff)
mgr = Office_Manager(user=staff3)
practice = PracticeLocation(
practice_address1='555 Pleasant Pioneer Grove',
practice_address2='Trailer Q615',
practice_city='Mountain View',
practice_state='CA',
practice_zip='94040-4104',
practice_lat=37.36876,
practice_longit=-122.081864)
practice.save()
mgr.manager_role = 1
mgr.practice = practice
mgr.save()
return_data5 = {'last_name': u'thj', 'office_address1': u'555 Bryant St.',
'office_address2': u'', 'photo': '/media/images/photos/staff_icon.jpg',
'specialty': '', 'mdcom_phone': '', 'office_city': u'Palo Alto',
'accepting_patients': False, 'id': mgr.user.user.id,
'custom_logos': get_custom_logos(mhlu_id), 'first_name': u'maneger',
'office_state': u'CA', 'office_zip': u'', 'staff_type': 'Office Manager'}
self.assertEqual(return_data5, getUserInfo(mgr.user.user.id), 'test failed for getUserInfo maneger')
def testGetProviderList(self, mockgeo):
practice1 = PracticeLocation(
practice_name='USA practice',
practice_address1='555 Pleasant Pioneer Grove',
practice_address2='Trailer Q615',
practice_city='Mountain View',
practice_state='CA',
practice_zip='94040-4104',
practice_lat=37.36876,
practice_longit=-122.081864)
practice1.save()
practice2 = PracticeLocation(
practice_name='China practice',
practice_address1='jiangsu',
practice_address2='beijing',
practice_city='suzhou',
practice_state='JS',
practice_zip='94040-4104',
practice_lat=37.36876,
practice_longit=-122.081864)
practice2.save()
site1 = Site(
name='mysite',
address1='555 Pleasant Pioneer Grove',
address2='Trailer Q615',
city='Mountain View',
state='CA',
zip='94040-4104',
lat=37.36876,
longit=-122.081864,
short_name='MSite'
)
site1.save()
site2 = Site(
name='doctorcom',
address1='555 Pleasant Pioneer Grove',
address2='Trailer Q615',
city='Mountain View',
state='CA',
zip='94040-4104',
lat=37.36876,
longit=-122.081864,
short_name='MSite'
)
site2.save()
provider1 = create_user("prov1", "provider_first",
"provider_last", "demo", "555 Bryant St.", "Palo Alto", "CA",
"94306", uklass=Provider)
provider1.address2 = 'suzhou china'
provider1.user.save()
provider1.sites.add(site1)
provider1.sites.add(site2)
provider1.current_site = site1
provider1.practices.add(practice1)
provider1.practices.add(practice2)
provider1.current_practice = practice1
provider1.save()
provider2 = create_user("prov2", "provider_first1", "provider_last1",
"demo", "suzhou china", "suzhou", "AB", "94307", uklass=Provider)
provider2.sites.add(site2)
provider2.current_site = site2
provider2.practices.add(practice2)
provider2.current_practice = practice2
provider2.save()
condition_dicts = [
{'name':u'provider', 'result':2, 'result_failed':'name failed'},
{'name':u'last1 first1', 'result':1, 'result_failed':'name failed'},
{'name':u'abc', 'result':0, 'result_failed':'name failed'},
{'address':u'Bryant', 'result':1, 'result_failed':'address failed'},
{'address':u'suzhou china', 'result':2, 'result_failed':'address failed'},
{'address':u'abc', 'result':0, 'result_failed':'address failed'},
{'city':u'Palo', 'result':1, 'result_failed':'city failed'},
{'city':u'suzhou', 'result':1, 'result_failed':'city failed'},
{'city':u'abc', 'result':0, 'result_failed':'city failed'},
{'state':u'CA', 'result':1, 'result_failed':'state failed'},
{'state':u'AB', 'result':1, 'result_failed':'state failed'},
{'state':u'abc', 'result':0, 'result_failed':'state failed'},
{'zip':provider1.zip, 'result':1, 'result_failed':'zip failed1'},
{'zip':provider2.zip, 'result':1, 'result_failed':'zip failed2'},
{'zip':provider2.zip, 'distance':5000, 'result':1, 'result_failed':'zip failed3'},
{'zip':u'43523', 'result':0, 'result_failed':'zip failed'},
{'current_hospital':u'doctorcom', 'result':1, 'result_failed':'current_hospital failed'},
{'current_hospital':u'mysite', 'result':1, 'result_failed':'current_hospital failed'},
{'current_hospital':u'22222', 'result':0, 'result_failed':'current_hospital failed'},
{'hospital':u'mysite', 'result':1, 'result_failed':'hospital failed'},
{'hospital':u'doctorcom', 'result':2, 'result_failed':'hospital failed'},
{'hospital':u'abc', 'result':0, 'result_failed':'hospital failed'},
{'current_practice':u'USA practice', 'result':1, 'result_failed':'current_practice failed'},
{'current_practice':u'China practice', 'result':1, 'result_failed':'current_practice failed'},
{'current_practice':u'abc', 'result':0, 'result_failed':'current_practice failed'},
{'practice':u'USA practice', 'result':1, 'result_failed':'practice failed'},
{'practice':u'China practice', 'result':2, 'result_failed':'practice failed'},
{'practice':u'abc', 'result':0, 'result_failed':'practice failed'},
{'limit':0, 'result':2, 'result_total':2, 'result_failed':'limit failed'},
{'limit':1, 'result':1, 'result_total':2, 'result_failed':'limit failed'},
{'limit':2, 'result':2, 'result_total':2, 'result_failed':'limit failed'},
{
'name':u'provider', 'address':u'suzhou china', 'city':u'suzhou', 'state':u'AB',
'current_hospital':u'doctorcom',
'hospital':u'doctorcom', 'current_practice':u'China practice',
'limit':1, 'result':1, 'result_total':1, 'result_failed':'all failed'
},
{
'name':u'abc', 'address':u'suzhou china', 'city':u'suzhou', 'state':u'AB',
'zip':provider2.zip, 'current_hospital':u'doctorcom',
'hospital':u'doctorcom', 'current_practice':u'China practice',
'limit':1, 'result':0, 'result_total':0, 'result_failed':'all failed'
},
]
with mock.patch('MHLogin.api.v1.utils_users.geocode2',
new_callable=lambda: UtilsUsersTest.mock_geocode):
for d in condition_dicts:
result = getProviderList(d)
if 'result_total' not in d:
d['result_total'] = d['result']
self.assertEqual(d['result_total'], result['total_count'], d['result_failed'])
self.assertEqual(d['result'], len(result['results']), d['result_failed'])
phy2 = Physician(user=provider1)
phy2.specialty = 'AC'
phy2.save()
nppa = NP_PA(user=provider2)
nppa.save()
condition_dicts = [
{'specialty':u'AC', 'result':1, 'result_failed':'specialty failed'},
{'specialty':u'NP/PA/Midwife', 'result':1, 'result_failed':'specialty failed'},
{'specialty':u'abc', 'result':2, 'result_failed':'specialty failed'},
]
for d in condition_dicts:
result = getProviderList(d)
if 'result_total' not in d:
d['result_total'] = d['result']
self.assertEqual(d['result_total'], result['total_count'], d['result_failed'])
self.assertEqual(d['result'], len(result['results']), d['result_failed'])
def testSetProviderResultList(self): provider = create_user(get_random_username(), "nppa", "thj", "demo", "555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider) provider.save() nppa = NP_PA(user=provider) nppa.save() provider1 = create_user(get_random_username(), "physician", "thj", "demo", "555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider) provider1.save() phy = Physician(user=provider1) phy.specialty = 'AC' phy.save() practice = PracticeLocation( practice_address1='555 Pleasant Pioneer Grove', practice_address2='Trailer Q615', practice_city='Mountain View', practice_state='CA', practice_zip='94040-4104', practice_lat=37.36876, practice_longit=-122.081864) practice.save() provider2 = create_user(get_random_username(), "nppa", "thj", "demo", "555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider) provider2.practices.add(practice) provider2.current_practice = practice provider2.save() nppa2 = NP_PA(user=provider2) nppa2.save() provider3 = create_user(get_random_username(), "physician", "thj", "demo", "555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider) provider3.practices.add(practice) provider3.current_practice = practice provider3.save() phy2 = Physician(user=provider3) phy2.specialty = 'AC' phy2.save() site = Site( name='mysite', address1='555 Pleasant Pioneer Grove', address2='Trailer Q615', city='Mountain View', state='CA', zip='94040-4104', lat=37.36876, longit=-122.081864, short_name='MSite' ) site.save() provider4 = create_user(get_random_username(), "nppa", "thj", "demo", "555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider) provider4.sites.add(site) provider4.current_site = site provider4.save() nppa3 = NP_PA(user=provider4) nppa3.save() provider5 = create_user(get_random_username(), "physician", "thj", "demo", "555 Bryant St.", "Palo Alto", "CA", "", uklass=Provider) provider5.sites.add(site) provider5.current_site = site provider5.save() phy3 = Physician(user=provider5) phy3.specialty = 'AC' phy3.save() self.assertEqual(0, len(setProviderResultList([])), 'test failed for setProviderResultList') self.assertEqual(1, len(setProviderResultList([nppa.user], phy.user)), 'test failed for setProviderResultList') self.assertEqual(1, len(setProviderResultList([phy.user])), 'test failed for setProviderResultList') self.assertEqual(1, len(setProviderResultList([nppa2.user])), 'test failed for setProviderResultList') self.assertEqual(1, len(setProviderResultList([phy2.user])), 'test failed for setProviderResultList') self.assertEqual(1, len(setProviderResultList([nppa3.user])), 'test failed for setProviderResultList') self.assertEqual(1, len(setProviderResultList([phy3.user])), 'test failed for setProviderResultList') self.assertEqual(6, len(setProviderResultList([nppa.user, phy.user, nppa2.user, phy2.user, nppa3.user, phy3.user])), 'test failed for setProviderResultList')
def testGetUserInfo(self):
mhlu = create_user(get_random_username(), "mhluser", "thj", "demo",
"555 Bryant St.", "Palo Alto", "CA", "")
mhlu.mdcom_phone = '9002000001'
mhlu.save()
mhlu_id = mhlu.id
return_data1 = {
'last_name': u'thj',
'office_address1': u'555 Bryant St.',
'office_address2': u'',
'photo': '/media/images/photos/generic_128.png',
'specialty': '',
'mdcom_phone': '',
'office_city': u'Palo Alto',
'accepting_patients': False,
'id': mhlu_id,
'custom_logos': get_custom_logos(mhlu_id),
'first_name': u'mhluser',
'office_state': u'CA',
'office_zip': u'',
'staff_type': ''
}
self.assertEqual(Http404, getUserInfo(mhlu_id + 1, None),
'test failed for getUserInfo')
self.assertEqual(return_data1, getUserInfo(mhlu_id),
'test failed for getUserInfo mhluser')
# create provider for test
provider = create_user(get_random_username(),
"provider",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider.mdcom_phone = '9002000002'
provider.save()
return_data2 = {
'last_name': u'thj',
'office_address1': u'555 Bryant St.',
'office_address2': u'',
'photo': '/media/images/photos/avatar2.png',
'specialty': 'NP/PA/Midwife',
'mdcom_phone': u'9002000002',
'office_city': u'Palo Alto',
'accepting_patients': False,
'id': provider.user.id,
'custom_logos': get_custom_logos(mhlu_id),
'first_name': u'provider',
'office_state': u'CA',
'office_zip': u'',
'staff_type': ''
}
self.assertEqual(return_data2, getUserInfo(provider.user.id),
'test failed for getUserInfo provider')
# create physician
provider1 = create_user(get_random_username(),
"physician",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider1.mdcom_phone = '9002000002'
provider1.save()
phy = Physician(user=provider1)
phy.specialty = 'AC'
phy.save()
return_data3 = {
'last_name': u'thj',
'office_address1': u'555 Bryant St.',
'office_address2': u'',
'photo': '/media/images/photos/avatar2.png',
'specialty': u'Acupuncture',
'mdcom_phone': u'9002000002',
'office_city': u'Palo Alto',
'accepting_patients': True,
'id': phy.user.user.id,
'custom_logos': get_custom_logos(mhlu_id),
'first_name': u'physician',
'office_state': u'CA',
'office_zip': u'',
'staff_type': ''
}
self.assertEqual(return_data3, getUserInfo(phy.user.user.id),
'test failed for getUserInfo physician')
# create office staff
staff = create_office_staff(get_random_username(),
"staff",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=OfficeStaff)
return_data4 = {
'last_name': u'thj',
'office_address1': u'555 Bryant St.',
'office_address2': u'',
'photo': '/media/images/photos/staff_icon.jpg',
'specialty': '',
'mdcom_phone': '',
'office_city': u'Palo Alto',
'accepting_patients': False,
'id': staff.user.id,
'custom_logos': get_custom_logos(mhlu_id),
'first_name': u'staff',
'office_state': u'CA',
'office_zip': u'',
'staff_type': 'Office Staff'
}
self.assertEqual(return_data4, getUserInfo(staff.user.id),
'test failed for getUserInfo staff')
# create nurse
staff2 = create_office_staff(get_random_username(),
"nurse",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=OfficeStaff)
nurse = Nurse(user=staff2)
nurse.save()
return_data5 = {
'last_name': u'thj',
'office_address1': u'555 Bryant St.',
'office_address2': u'',
'photo': '/media/images/photos/nurse.jpg',
'specialty': '',
'mdcom_phone': '',
'office_city': u'Palo Alto',
'accepting_patients': False,
'id': nurse.user.user.id,
'custom_logos': get_custom_logos(mhlu_id),
'first_name': u'nurse',
'office_state': u'CA',
'office_zip': u'',
'staff_type': 'Office Staff'
}
self.assertEqual(return_data5, getUserInfo(nurse.user.user.id),
'test failed for getUserInfo nurse')
# create office staff
staff3 = create_office_staff(get_random_username(),
"maneger",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=OfficeStaff)
mgr = Office_Manager(user=staff3)
practice = PracticeLocation(
practice_address1='555 Pleasant Pioneer Grove',
practice_address2='Trailer Q615',
practice_city='Mountain View',
practice_state='CA',
practice_zip='94040-4104',
practice_lat=37.36876,
practice_longit=-122.081864)
practice.save()
mgr.manager_role = 1
mgr.practice = practice
mgr.save()
return_data5 = {
'last_name': u'thj',
'office_address1': u'555 Bryant St.',
'office_address2': u'',
'photo': '/media/images/photos/staff_icon.jpg',
'specialty': '',
'mdcom_phone': '',
'office_city': u'Palo Alto',
'accepting_patients': False,
'id': mgr.user.user.id,
'custom_logos': get_custom_logos(mhlu_id),
'first_name': u'maneger',
'office_state': u'CA',
'office_zip': u'',
'staff_type': 'Office Manager'
}
self.assertEqual(return_data5, getUserInfo(mgr.user.user.id),
'test failed for getUserInfo maneger')
def testSetProviderResultList(self):
provider = create_user(get_random_username(),
"nppa",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider.save()
nppa = NP_PA(user=provider)
nppa.save()
provider1 = create_user(get_random_username(),
"physician",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider1.save()
phy = Physician(user=provider1)
phy.specialty = 'AC'
phy.save()
practice = PracticeLocation(
practice_address1='555 Pleasant Pioneer Grove',
practice_address2='Trailer Q615',
practice_city='Mountain View',
practice_state='CA',
practice_zip='94040-4104',
practice_lat=37.36876,
practice_longit=-122.081864)
practice.save()
provider2 = create_user(get_random_username(),
"nppa",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider2.practices.add(practice)
provider2.current_practice = practice
provider2.save()
nppa2 = NP_PA(user=provider2)
nppa2.save()
provider3 = create_user(get_random_username(),
"physician",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider3.practices.add(practice)
provider3.current_practice = practice
provider3.save()
phy2 = Physician(user=provider3)
phy2.specialty = 'AC'
phy2.save()
site = Site(name='mysite',
address1='555 Pleasant Pioneer Grove',
address2='Trailer Q615',
city='Mountain View',
state='CA',
zip='94040-4104',
lat=37.36876,
longit=-122.081864,
short_name='MSite')
site.save()
provider4 = create_user(get_random_username(),
"nppa",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider4.sites.add(site)
provider4.current_site = site
provider4.save()
nppa3 = NP_PA(user=provider4)
nppa3.save()
provider5 = create_user(get_random_username(),
"physician",
"thj",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"",
uklass=Provider)
provider5.sites.add(site)
provider5.current_site = site
provider5.save()
phy3 = Physician(user=provider5)
phy3.specialty = 'AC'
phy3.save()
self.assertEqual(0, len(setProviderResultList([])),
'test failed for setProviderResultList')
self.assertEqual(1, len(setProviderResultList([nppa.user], phy.user)),
'test failed for setProviderResultList')
self.assertEqual(1, len(setProviderResultList([phy.user])),
'test failed for setProviderResultList')
self.assertEqual(1, len(setProviderResultList([nppa2.user])),
'test failed for setProviderResultList')
self.assertEqual(1, len(setProviderResultList([phy2.user])),
'test failed for setProviderResultList')
self.assertEqual(1, len(setProviderResultList([nppa3.user])),
'test failed for setProviderResultList')
self.assertEqual(1, len(setProviderResultList([phy3.user])),
'test failed for setProviderResultList')
self.assertEqual(
6,
len(
setProviderResultList([
nppa.user, phy.user, nppa2.user, phy2.user, nppa3.user,
phy3.user
])), 'test failed for setProviderResultList')
def testGetProviderList(self, mockgeo):
practice1 = PracticeLocation(
practice_name='USA practice',
practice_address1='555 Pleasant Pioneer Grove',
practice_address2='Trailer Q615',
practice_city='Mountain View',
practice_state='CA',
practice_zip='94040-4104',
practice_lat=37.36876,
practice_longit=-122.081864)
practice1.save()
practice2 = PracticeLocation(practice_name='China practice',
practice_address1='jiangsu',
practice_address2='beijing',
practice_city='suzhou',
practice_state='JS',
practice_zip='94040-4104',
practice_lat=37.36876,
practice_longit=-122.081864)
practice2.save()
site1 = Site(name='mysite',
address1='555 Pleasant Pioneer Grove',
address2='Trailer Q615',
city='Mountain View',
state='CA',
zip='94040-4104',
lat=37.36876,
longit=-122.081864,
short_name='MSite')
site1.save()
site2 = Site(name='doctorcom',
address1='555 Pleasant Pioneer Grove',
address2='Trailer Q615',
city='Mountain View',
state='CA',
zip='94040-4104',
lat=37.36876,
longit=-122.081864,
short_name='MSite')
site2.save()
provider1 = create_user("prov1",
"provider_first",
"provider_last",
"demo",
"555 Bryant St.",
"Palo Alto",
"CA",
"94306",
uklass=Provider)
provider1.address2 = 'suzhou china'
provider1.user.save()
provider1.sites.add(site1)
provider1.sites.add(site2)
provider1.current_site = site1
provider1.practices.add(practice1)
provider1.practices.add(practice2)
provider1.current_practice = practice1
provider1.save()
provider2 = create_user("prov2",
"provider_first1",
"provider_last1",
"demo",
"suzhou china",
"suzhou",
"AB",
"94307",
uklass=Provider)
provider2.sites.add(site2)
provider2.current_site = site2
provider2.practices.add(practice2)
provider2.current_practice = practice2
provider2.save()
condition_dicts = [
{
'name': u'provider',
'result': 2,
'result_failed': 'name failed'
},
{
'name': u'last1 first1',
'result': 1,
'result_failed': 'name failed'
},
{
'name': u'abc',
'result': 0,
'result_failed': 'name failed'
},
{
'address': u'Bryant',
'result': 1,
'result_failed': 'address failed'
},
{
'address': u'suzhou china',
'result': 2,
'result_failed': 'address failed'
},
{
'address': u'abc',
'result': 0,
'result_failed': 'address failed'
},
{
'city': u'Palo',
'result': 1,
'result_failed': 'city failed'
},
{
'city': u'suzhou',
'result': 1,
'result_failed': 'city failed'
},
{
'city': u'abc',
'result': 0,
'result_failed': 'city failed'
},
{
'state': u'CA',
'result': 1,
'result_failed': 'state failed'
},
{
'state': u'AB',
'result': 1,
'result_failed': 'state failed'
},
{
'state': u'abc',
'result': 0,
'result_failed': 'state failed'
},
{
'zip': provider1.zip,
'result': 1,
'result_failed': 'zip failed1'
},
{
'zip': provider2.zip,
'result': 1,
'result_failed': 'zip failed2'
},
{
'zip': provider2.zip,
'distance': 5000,
'result': 1,
'result_failed': 'zip failed3'
},
{
'zip': u'43523',
'result': 0,
'result_failed': 'zip failed'
},
{
'current_hospital': u'doctorcom',
'result': 1,
'result_failed': 'current_hospital failed'
},
{
'current_hospital': u'mysite',
'result': 1,
'result_failed': 'current_hospital failed'
},
{
'current_hospital': u'22222',
'result': 0,
'result_failed': 'current_hospital failed'
},
{
'hospital': u'mysite',
'result': 1,
'result_failed': 'hospital failed'
},
{
'hospital': u'doctorcom',
'result': 2,
'result_failed': 'hospital failed'
},
{
'hospital': u'abc',
'result': 0,
'result_failed': 'hospital failed'
},
{
'current_practice': u'USA practice',
'result': 1,
'result_failed': 'current_practice failed'
},
{
'current_practice': u'China practice',
'result': 1,
'result_failed': 'current_practice failed'
},
{
'current_practice': u'abc',
'result': 0,
'result_failed': 'current_practice failed'
},
{
'practice': u'USA practice',
'result': 1,
'result_failed': 'practice failed'
},
{
'practice': u'China practice',
'result': 2,
'result_failed': 'practice failed'
},
{
'practice': u'abc',
'result': 0,
'result_failed': 'practice failed'
},
{
'limit': 0,
'result': 2,
'result_total': 2,
'result_failed': 'limit failed'
},
{
'limit': 1,
'result': 1,
'result_total': 2,
'result_failed': 'limit failed'
},
{
'limit': 2,
'result': 2,
'result_total': 2,
'result_failed': 'limit failed'
},
{
'name': u'provider',
'address': u'suzhou china',
'city': u'suzhou',
'state': u'AB',
'current_hospital': u'doctorcom',
'hospital': u'doctorcom',
'current_practice': u'China practice',
'limit': 1,
'result': 1,
'result_total': 1,
'result_failed': 'all failed'
},
{
'name': u'abc',
'address': u'suzhou china',
'city': u'suzhou',
'state': u'AB',
'zip': provider2.zip,
'current_hospital': u'doctorcom',
'hospital': u'doctorcom',
'current_practice': u'China practice',
'limit': 1,
'result': 0,
'result_total': 0,
'result_failed': 'all failed'
},
]
with mock.patch('MHLogin.api.v1.utils_users.geocode2',
new_callable=lambda: UtilsUsersTest.mock_geocode):
for d in condition_dicts:
result = getProviderList(d)
if 'result_total' not in d:
d['result_total'] = d['result']
self.assertEqual(d['result_total'], result['total_count'],
d['result_failed'])
self.assertEqual(d['result'], len(result['results']),
d['result_failed'])
phy2 = Physician(user=provider1)
phy2.specialty = 'AC'
phy2.save()
nppa = NP_PA(user=provider2)
nppa.save()
condition_dicts = [
{
'specialty': u'AC',
'result': 1,
'result_failed': 'specialty failed'
},
{
'specialty': u'NP/PA/Midwife',
'result': 1,
'result_failed': 'specialty failed'
},
{
'specialty': u'abc',
'result': 2,
'result_failed': 'specialty failed'
},
]
for d in condition_dicts:
result = getProviderList(d)
if 'result_total' not in d:
d['result_total'] = d['result']
self.assertEqual(d['result_total'], result['total_count'],
d['result_failed'])
self.assertEqual(d['result'], len(result['results']),
d['result_failed'])