def test_rest(): reset_all_users() username = pwd.getpwuid(os.getuid()).pw_name client = make_logged_in_client(username=username, is_superuser=True) d = {'eventlet.input': None} response = client.get('/shell/', follow=True, **d) shell_types_available = [] for item in shell.conf.SHELL_TYPES.keys(): nice_name = shell.conf.SHELL_TYPES[item].nice_name.get().strip() assert nice_name in response.content command = shell.conf.SHELL_TYPES[item].command.get().strip().split() if utils.executable_exists(command): shell_types_available.append(item) if not shell_types_available: return response = client.get('/shell/create?keyName=%s' % (time.strftime("%s"), ), follow=True, **d) assert "There is no shell with that name." in response.content nonexistent_username = "******" % (time.strftime("%s"), ) client2 = make_logged_in_client(username=nonexistent_username, is_superuser=True) response = client2.get("/shell/create?keyName=%s" % (shell_types_available[0], ), follow=True, **d) assert "There is no Unix user account for you." in response.content response = client.get("/shell/create?keyName=%s" % (shell_types_available[0], ), follow=True, **d) fragment = '<span id="shell_id" class="hide">' assert fragment in response.content shell_id_start = response.content.index(fragment) + len(fragment) shell_id_end = response.content.index('</span>', shell_id_start) shell_id = response.content[shell_id_start:shell_id_end] assert re.match(r"^\s*\d+\s*$", shell_id) response = client.post("/shell/kill_shell", follow=True, data={constants.SHELL_ID: shell_id}, **d) assert 'Shell successfully marked for cleanup' in response.content
def test_unix_user_account_check(): reset_all_users() nonexistent_username = "******" % (time.strftime("%s"),) client = make_logged_in_client(username=nonexistent_username, is_superuser=True) d = {"eventlet.input": None} response = client.get("/shell/", follow=True, **d) assert "The Shell app requires a Unix user account for every user of Hue" in response.content
def test_unix_user_account_check(): reset_all_users() nonexistent_username = "******" % (time.strftime("%s"), ) client = make_logged_in_client(username=nonexistent_username, is_superuser=True) d = {'eventlet.input': None} response = client.get('/shell/', follow=True, **d) assert "The Shell application requires a Unix user account for every user of Hue" in response.content
def log_msg(self, msg, **kwargs): if kwargs and kwargs.get("cat"): print time.strftime("[%H:%M:%S] ") + "[" + kwargs.get("cat") + "] " + str(msg) elif self.get_options() == None: print time.strftime("[%H:%M:%S] ") + str(msg) sys.stdout.flush() elif self.get_options().debug == True: print time.strftime("[%H:%M:%S] ") + str(msg) sys.stdout.flush() else: print time.strftime("[%H:%M:%S] ") + str(msg) sys.stdout.flush()
def test_rest(): reset_all_users() username = pwd.getpwuid(os.getuid()).pw_name client = make_logged_in_client(username=username, is_superuser=True) d = {"eventlet.input": None} response = client.get("/shell/", follow=True, **d) shell_types_available = [] for item in shell.conf.SHELL_TYPES.keys(): nice_name = shell.conf.SHELL_TYPES[item].nice_name.get().strip() assert nice_name in response.content command = shell.conf.SHELL_TYPES[item].command.get().strip().split() if utils.executable_exists(command): shell_types_available.append(item) if not shell_types_available: return response = client.get("/shell/create?keyName=%s" % (time.strftime("%s"),), follow=True, **d) assert "There is no shell with that name." in response.content nonexistent_username = "******" % (time.strftime("%s"),) client2 = make_logged_in_client(username=nonexistent_username, is_superuser=True) response = client2.get("/shell/create?keyName=%s" % (shell_types_available[0],), follow=True, **d) assert "There is no Unix user account for you." in response.content response = client.get("/shell/create?keyName=%s" % (shell_types_available[0],), follow=True, **d) fragment = '<span id="shell_id" class="hide">' assert fragment in response.content shell_id_start = response.content.index(fragment) + len(fragment) shell_id_end = response.content.index("</span>", shell_id_start) shell_id = response.content[shell_id_start:shell_id_end] assert re.match(r"^\s*\d+\s*$", shell_id) response = client.post("/shell/kill_shell", follow=True, data={constants.SHELL_ID: shell_id}, **d) assert "Shell successfully marked for cleanup" in response.content
def on_block(self, LP, block, work, server): """ Recieves the current block and server and dumps it to file """ fd = None try: # determine if application is a script file or frozen exe if hasattr(sys, 'frozen'): application_path = os.path.dirname(sys.executable) elif __file__: application_path = os.path.dirname(__file__) fd = open(os.path.join(application_path, self.file_name), 'brwa+') except: fd = open(self.file_name, 'brwa+') fd.write(time.strftime('%H%M%S') + ":" + block + ":" + server + "\n" ) fd.close()
def on_block(self, LP, block, work, server): """ Recieves the current block and server and dumps it to file """ fd = None try: # determine if application is a script file or frozen exe if hasattr(sys, 'frozen'): application_path = os.path.dirname(sys.executable) elif __file__: application_path = os.path.dirname(__file__) fd = open(os.path.join(application_path, self.file_name), 'brwa+') except: fd = open(self.file_name, 'brwa+') fd.write(time.strftime('%H%M%S') + ":" + block + ":" + server + "\n") fd.close()