def test__using_with_valid_zip_and_target_folder(self): test_zip = Files.current_folder() target_folder = '/tmp/unzip_test' with Zip_Folder(test_zip) as (zip_file): with Unzip_File(zip_file, target_folder, True) as temp_folder: assert Files.exists(temp_folder) is True assert Files.exists(temp_folder) is False
def test__using_with__no_params(self): with Temp_File() as temp: assert Files.file_extension(temp.file_path) == '.tmp' assert Files.exists(temp.file_path) assert Files.contents(temp.file_path) == '...' assert Files.not_exists(temp.file_path) with Temp_File('abc', 'txt') as temp: assert Files.file_extension(temp.file_path) == '.txt' assert Files.exists(temp.file_path) assert Files.contents(temp.file_path) == 'abc' assert Files.not_exists(temp.file_path)
def test__init__(self): temp_file = Temp_File() assert Files.exists(temp_file.tmp_folder) assert Files.not_exists(temp_file.tmp_file) assert Files.not_exists(temp_file.file_path) assert temp_file.tmp_folder in temp_file.file_path assert '/' == temp_file.file_path.replace(temp_file.tmp_folder, '').replace( temp_file.tmp_file, '')
def get_oauth_token(self, desired_scope): secret_data = json.loads(Secrets(self.gsuite_secret_id).value() ) # load secret from AWS Secrets store token_file = '/tmp/gmail_credential_{0}.json'.format( desired_scope ) # this is the tmp file with the token value for the desired scope if not Files.exists(token_file): # if the file does not exist if os.getenv('AWS_REGION') is not None or os.getenv( 'SYNC_SERVER' ): # check if we are running in AWS or in the sync server Files.write( token_file, secret_data['token.json'] ) # if we are, use the token.json value from the AWS secret_data else: secret_data = json.loads(Secrets( 'gsuite_token').value()) # BUG, need to refactor this credentials_file = '/tmp/gsuite_credentials.json' # file to hold the credentials.json value Files.write(credentials_file, secret_data['credentials.json'] ) # save value received from AWS into file store = file.Storage( token_file) # create a gsuite Storage object scopes = 'https://www.googleapis.com/auth/{0}'.format( desired_scope ) # full qualified name for the desired scopes flow = client.flow_from_clientsecrets( credentials_file, scopes) # create a gsuite flow object flags = argparser.parse_args( '--auth_host_name localhost --logging_level INFO'.split() ) # configure the use of a localhost server to received the oauth response run_flow( flow, store, flags ) # open browser and prompt user to follow the OAuth flow Files.delete( credentials_file ) # delete main gsuite credentials file (since we don't want it hanging around) return token_file # return file with token credentials
def test__using_with_valid_zip_no_target_folder(self): test_zip = Files.current_folder() with Zip_Folder(test_zip) as (zip_file): with Unzip_File(zip_file, None, True) as temp_folder: assert Files.exists(temp_folder) is True assert Files.exists(temp_folder) is False
def get_last_chrome_session(self): if Files.exists(self.file_tmp_last_chrome_session): return Json.load_json(self.file_tmp_last_chrome_session) return {}
def __exit__(self, type, value, traceback): if Files.exists(self.zip_file) and self.delete_zip_file: Files.delete(self.zip_file)
def __enter__(self): if Files.exists(self.target_folder): self.zip_file = Files.zip_folder(self.target_folder) return self.zip_file
async def test_screenshot(self): await self.api.open('https://news.bbc.co.uk') file = await self.api.screenshot() assert Files.exists(file)
def test_exists(self): assert Files.exists(Files.current_folder()) is True assert Files.exists('aaaa_bbb_ccc') is False assert Files.exists(None) is False
def test__using_with_params(self): target_folder = Files.current_folder() with Zip_Folder(target_folder) as (zip_file): assert Files.exists(zip_file) is True assert Files.exists(zip_file) is False
def test_screenshot_file(self): with Temp_File(self.html, 'html') as temp_file: tmp_img = '/tmp/test_screenshot_html.png' clip = {'x': 1, 'y': 1, 'width': 180, 'height': 30} img_file = self.render_page.screenshot_file(temp_file.file_path,tmp_img,clip=clip) assert Files.exists(img_file)
def test_screenshot_html(self): tmp_img = '/tmp/test_screenshot_html.png' img_file = self.render_page.screenshot_html(self.html,tmp_img) assert Files.exists(img_file)