def auto_enter(deviceid='test_device_id'):
'全自动登录进入世界，返回世界socket'
rsp = ensure_login(deviceid)
# rsp = ensure_sdk_login(deviceid)
assert rsp.servers, '服务器为空'
# if settings.MODE == 'qq':
# ip = settings.SESSION['client'][0]
# else:
ip = rsp.servers[0].ip
sock = socket.create_connection((ip, int(rsp.servers[0].port)))
set_peer(getcurrent(), sock)
set_expect_sock(sock)
rsp = login_world(sock, rsp.verify_code)
newRole = False
if not rsp.roles:
rand_sleep(5, 3)
from role import create_role
print 'create role'
rsp = create_role(sock, random.choice([0, 1]), random.choice([1, 2,
3]))
if isinstance(rsp, int):
print rsp
rsp = create_role(sock, random.choice([0, 1]),
random.choice([1, 2, 3]))
print 'retry create role'
roleID = rsp.roleId
newRole = True
else:
roleID = rsp.roles[0].id
enter(sock, roleID, newRole)
return sock
def test_create_role_duplicate(self):
"""Makes sure roles can't be created with the same role name."""
create_role('tester')
from peewee import IntegrityError
with self.assertRaises(IntegrityError):
create_role('tester')
t = Role.delete().where(Role.role_name == 'test')
t.execute()
def test_create_role(self):
"""Create a role for users."""
create_role('tester')
t = Role.select().where(Role.role_name == 'tester')
d = Role.delete().where(Role.role_name == 'tester')
d.execute()
self.assertEqual(str(t), ("<class 'models.Role'> SELECT `t1`.`id`,"
" `t1`.`role_name` FROM `role` AS t1 "
"WHERE (`t1`.`role_name` = %s) ['tester']"))
def test_create_role_no_arguments(self):
"""Attempt to create a role, but fail if not nullable armaments are missing."""
with self.assertRaises(TypeError):
create_role()
