def addUser(req):
response_data = {"result": "fail", "method": req.method}
if (req.method == "POST"):
data = req.POST
username = data["username"]
password = data["password"]
user1 = User(username=username, password=password)
user1.save()
response_data = {"result": "ok", "method": req.method}
return HttpResponse(json.dumps(response_data),
content_type="application/json")
def persis_test_user():
user = User()
user.username = '******'
user.email = '*****@*****.**'
user.set_password(password)
user.save()
return user
def test_user_crud(self):
"""
Test following 5 methods
- camunda_api.create_user
- camunda_api.delete_user
- camunda_api.get_user
- camunda_api.get_user_list
- camunda_api.get_user_count
"""
user = User()
user.email = '*****@*****.**'
user.first_name = 'test_first_name'
user.last_name = 'test_last_name'
user_count_initial = UserApi.get_count(self.session)
# Test camunda_api.create_user
response = UserApi.create_user(self.session, user)
self.assertTrue(
response.ok,
"Response should be OK. (status_code = {})\n{}".format(
response.status_code, response.text))
# Check user count / Test camunda_api.user_list and camunda_api.user_count
user_count_after_creation = UserApi.get_count(self.session)
self.assertEqual(user_count_initial + 1, user_count_after_creation)
# Test camunda_api.get_user
response = UserApi.get_one(self.session, user)
self.assertTrue(
response.ok,
"Response should be OK. (status_code = {})\n{}".format(
response.status_code, response.text))
response_json = response.json()
self.assertEqual(response_json['id'], user.email)
self.assertEqual(response_json['firstName'], user.first_name)
self.assertEqual(response_json['lastName'], user.last_name)
self.assertEqual(response_json['email'], user.email)
# Test camunda_api.delete_user
response = UserApi.delete(self.session, user)
self.assertTrue(
response.ok,
"Response should be OK. (status_code = {})\n{}".format(
response.status_code, response.text))
# Check user count / Test camunda_api.user_list and camunda_api.user_count
user_count_after_deletion = UserApi.get_count(self.session)
self.assertEqual(user_count_after_creation - 1,
user_count_after_deletion)
def create(self, validated_data):
password = validated_data.pop('password')
user = User(**validated_data)
user.username = validated_data.get('email').lower()
user.set_password(password)
user.save()
return user
def test_register_should_save_user(self):
u = User()
u.username = '******'
u.set_password('123')
u.email = '*****@*****.**'
u.active = True
register_user(u, 'captcha', fake_session)
self.assertIsNotNone(authenticate('ali', '123'))
def test_create_user_member(self):
"""
Test following 3 methods
- camunda_api.TenantApi.create_user_member
- camunda_api.TenantApi.get_list
- camunda_api.UserApi.get_list
"""
tenant = Tenant()
tenant.name = 'test_create_user_member_tenant_name'
user = User()
user.email = '*****@*****.**'
user.first_name = 'test_create_user_member_first_name'
user.last_name = 'test_create_user_member_last_name'
user.tenant = tenant
UserApi.create_user(self.session, user)
TenantApi.create(self.session, tenant)
TenantApi.create_user_member(self.session, user)
tenant_list = TenantApi.get_list(self.session,
userMember=user.email).json()
user_list = UserApi.get_list(self.session,
memberOfTenant=tenant.name).json()
self.assertEqual(tenant.name, tenant_list[0]['id'])
self.assertEqual(tenant.name, tenant_list[0]['name'])
self.assertEqual(user.email, user_list[0]['id'])
self.assertEqual(user.first_name, user_list[0]['firstName'])
self.assertEqual(user.last_name, user_list[0]['lastName'])
self.assertEqual(user.email, user_list[0]['email'])
UserApi.delete(self.session, user)
TenantApi.delete(self.session, tenant)
def post(self, request):
username = request.POST['username']
password = request.POST['password']
email = request.POST['email']
captcha = request.POST['captcha']
user = User()
user.username = username
user.email = email
user.set_password(password)
try:
register_user(user, captcha, request.session)
return HttpResponse("You registered")
except CaptchaException as e:
return HttpResponse("Captcha is wrong")
except UserAlreadyExistException as e:
return HttpResponse("User already exists")
except Exception as e:
return HttpResponse(e)
def test_user_confirmed_callback(self):
tenant = Tenant()
tenant.name = 'test_tenant_name_SignalCallbacksTestCase'
user = User()
user.email = '*****@*****.**'
user.first_name = 'test_first_name_SignalCallbacksTestCase'
user.last_name = 'test_last_name_SignalCallbacksTestCase'
user.tenant = tenant
user_confirmed_callback(self.app, user)
response_json = UserApi.get_one(self.session, user).json()
self.assertEqual(response_json['id'], user.email)
self.assertEqual(response_json['firstName'], user.first_name)
self.assertEqual(response_json['lastName'], user.last_name)
self.assertEqual(response_json['email'], user.email)
response_json = TenantApi.get_one(self.session, user.tenant).json()
self.assertEqual(response_json['id'], user.tenant.name)
self.assertEqual(response_json['name'], user.tenant.name)
UserApi.delete(self.session, user)
TenantApi.delete(self.session, tenant)
def test_register_should_return_Exception(self):
u = User()
u.username = '******'
u.set_password('123')
u.email = '*****@*****.**'
self.assertRaises(CaptchaException, register_user, u, 'captch', fake_session)