def start_session(username="******",
password="******",
enterprise="enterprise",
api_url="https://*****:*****@enterprse.com"
user.enterprise_id = "<enterprise_id>"
user.enterprise_name = "enterprise"
user.firstname = "John",
user.id = "<user_id>"
user.lastname = "Doe"
user.role = "ROLE"
# Set API KEY
session._login_controller.api_key = user.api_key
with patch.object(NURESTTestSession, "_authenticate", return_value=True):
session.start()
return user
def test_to_dict(self):
""" Get object as dictionary """
user = User()
user.id = 3
user.user_name = "Christophe"
user.password = '******'
user.api_key = 'ABCD'
to_dict = user.to_dict()
self.assertEquals(to_dict['userName'], 'Christophe')
self.assertEquals(to_dict['password'], 'sorry')
self.assertEquals(to_dict['APIKey'], 'ABCD')
self.assertEquals(to_dict['parentID'], None)
self.assertEquals(to_dict['parentType'], None)
self.assertEquals(to_dict['owner'], None)
self.assertEquals(to_dict['creationDate'], None)
def test_to_dict(self):
""" Get object as dictionary """
user = User()
user.id = 3
user.user_name ="Christophe"
user.password = '******'
user.api_key = 'ABCD'
to_dict = user.to_dict()
self.assertEquals(to_dict['userName'], 'Christophe')
self.assertEquals(to_dict['password'], 'sorry')
self.assertEquals(to_dict['APIKey'], 'ABCD')
self.assertEquals(to_dict['parentID'], None)
self.assertEquals(to_dict['parentType'], None)
self.assertEquals(to_dict['owner'], None)
self.assertEquals(to_dict['creationDate'], None)
async def do_test():
t_model = TestTypesModel()
try:
await TestTypesModel.add(t_model)
assert False, 'Should be raise a AddEmptyInstanceError'
except AddEmptyInstanceError:
pass
test = 'hehe'
t_model.text = test
self.assertIsNone(t_model.id)
self.assertEqual(await TestTypesModel.add(t_model), 1)
self.assertEqual(t_model.id, 1)
t_model = await TestTypesModel.get(1)
self.assertEqual(t_model.text, test)
self.assertEqual(t_model.float_,
float(t_model._get_value('float_')))
user = User(name='test1',
num=133459,
password='******',
sex=1,
age=20,
date=datetime.now())
try:
await User.add(user)
assert False, 'Should be raise a MissingPKError'
except MissingPKError:
pass
user_id = 1
user.id = user_id
self.assertEqual(user.id, user_id)
query = await User.get(user_id)
self.assertIsNone(query)
self.assertEqual(await User.add(user), user_id)
query = await User.get(user_id)
self.assertEqual(query.id, 1)
self.assertEqual(query.num, 133459)
self.assertEqual(query.age, 20)
def start_session(username="******", password="******", enterprise="enterprise", api_url="https://*****:*****@enterprse.com"
user.enterprise_id ="<enterprise_id>"
user.enterprise_name ="enterprise"
user.firstname ="John",
user.id ="<user_id>"
user.lastname ="Doe"
user.role ="ROLE"
# Set API KEY
session._login_controller.api_key = user.api_key
# Activate session
_NURESTSessionCurrentContext.session = session
return user
async def do_test():
user = User(name='test8',
num=888888,
password='******',
sex=2,
age=80)
try:
await user.save()
assert False, 'Should be raise a MissingPKError'
except MissingPKError:
pass
user.id = 8
user_id = await user.save()
query_user = await user.get(user_id)
self.assertEqual(query_user.id, user.id)
self.assertEqual(query_user.num, user.num)
self.assertTrue(await query_user.delete())
query_user = await user.get(user_id)
self.assertIsNone(query_user)
