def test_get_cur_count(self):
"""
**test_get_cur_count** * Test that item number is returned
"""
with patch('cr_hunt.models.cr_Has.objects.filter') as filt:
with patch('cr_hunt.models.cr_Hunts.objects.filter') as filth:
filth.return_value.first.return_value = True
filt.return_value.count.return_value = 8
retval = models.get_cur_count('test')
self.assertEqual(retval, 8)
filt.reset_mock()
filth.return_value.first.return_value = False
retval = models.get_cur_count('test')
self.assertEqual(retval,0)
def render_main(request):
"""
:param: HttpRequest
:rtype: HttpRequest
Renders the welcome page for the create hunt app.
"""
if request.user.is_authenticated():
global hunt_id
hunt_id = request.user.username
hunt_id = gen_hunt_id(hunt_id)
global i_counter
i_counter = get_cur_count(hunt_id)
return render_to_response("cr_hunt/cr_hunt_main.html")
else:
return redirect('cr_error')
