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()