def test_add_hunt_has(self):
"""
**test_add_hunt_has** * Test that correct items are added
"""
with patch('cr_hunt.models.cr_Has.objects.create') as cr_obj:
with patch('cr_hunt.models.cr_Hunts.objects.get') as geth:
with patch('cr_hunt.models.Items.objects.get') as geti:
id_hunt = '123'
nitem = '1122'
nnum = 3
nclue = 'clue'
cr_obj = MagicMock(number=3, clue='clue')
cr_obj.save = MagicMock(return_value=None)
models.add_hunt_has(id_hunt, nitem, nnum, nclue)
self.assertEqual(cr_obj.number, nnum)
self.assertEqual(cr_obj.clue, nclue)
def render_proc_it(request):
"""
:param: HttpRequest
:rtype: HttpRequest
Renders the "Next" add item page, where the user is prompt to select items from our database,
(the Met's database, ultimately) and add it to the hunt that they're creating.
"""
if request.user.is_authenticated():
global i_counter
if i_counter == 10:
return redirect('cr_end')
if request.method == "POST":
global hunt_id
global i_counter
clue = str(request.POST.get('clue', ''))
item_id = str(request.POST.get('item', ''))
if item_id == '':
return redirect ('cr_aitem')
i_counter = i_counter + 1
add_hunt_has(hunt_id, item_id, i_counter, clue)
return redirect('cr_aitem')
else:
return redirect('cr_error')
