def test_login_api_no_user(self): func = Mock() decorat_func = login_api(func) resp = decorat_func(self.request) jsonres = json.loads(resp.content) self.assertEqual(jsonres['status'], 9) assert not func.called
def test_login_api_user(self): func = Mock(return_value='success') decorat_func = login_api(func) self.request.session['user'] = '******' resp = decorat_func(self.request) self.assertEqual(resp, 'success')