def test_create_file(app, delete_temp_files): """Test file creation.""" assert app assert delete_temp_files home_path = current_app.config['LOCAL_STORAGE'] efs = EFS() efs.upload(TEST_FILE, RANDOM_DATA) assert os.path.exists(os.path.join(home_path, TEST_FILE))
def test_create_file(bucket): """Test file creation.""" bucket = bucket() efs = EFS(storage="s3") RANDOM_DATA.seek(0) efs.upload(TEST_FILE, RANDOM_DATA) key = bucket.Object(TEST_FILE) RANDOM_DATA.seek(0) assert key.get()["Body"].read() == RANDOM_DATA.read()
def test_read_file(app, delete_temp_files): """Test read file.""" assert app assert delete_temp_files efs = EFS() efs.upload(TEST_FILE, RANDOM_DATA) uploaded_content = efs.open(TEST_FILE) content = uploaded_content.read() assert content == RANDOM_DATA.read() uploaded_content.close()
def test_read_file(bucket): """Test read file.""" bucket = bucket() k = bucket.Object(TEST_FILE) RANDOM_DATA.seek(0) k.put(Body=RANDOM_DATA) RANDOM_DATA.seek(0) efs = EFS(storage="s3") uploaded_content = efs.open(TEST_FILE) content = uploaded_content.read() assert content == RANDOM_DATA.read()
def test_remove_file(bucket): """Test remove file.""" bucket = bucket() assert not list(bucket.objects.all()) k = bucket.Object(TEST_FILE) RANDOM_DATA.seek(0) k.put(Body=RANDOM_DATA) RANDOM_DATA.seek(0) assert list(bucket.objects.all()) efs = EFS(storage="s3") efs.remove(TEST_FILE) assert not list(bucket.objects.all())
def test_remove_folder(app, delete_temp_files): """Test remove folder with content.""" assert app assert delete_temp_files efs = EFS() home_path = current_app.config['LOCAL_STORAGE'] efs.upload('very/long/path/to/be/created/' + TEST_FILE, RANDOM_DATA) assert os.path.exists(os.path.join(home_path, 'very/long/path/to/be/created/', TEST_FILE)) efs.remove('very') assert not os.path.exists(os.path.join(home_path, 'very'))
def test_remove_file(app, delete_temp_files): """Test remove file.""" assert app assert delete_temp_files efs = EFS() home_path = current_app.config['LOCAL_STORAGE'] efs.upload(TEST_FILE, RANDOM_DATA) assert os.path.exists(os.path.join(home_path, TEST_FILE)) efs.remove(TEST_FILE) assert not os.path.exists(os.path.join(home_path, TEST_FILE))
def test_move_file(app, delete_temp_files): """Test move file to a new folder.""" assert app assert delete_temp_files efs = EFS() home_path = current_app.config['LOCAL_STORAGE'] efs.upload(TEST_FILE, RANDOM_DATA) assert os.path.exists(os.path.join(home_path, TEST_FILE)) efs.move(TEST_FILE, 'special_text/test_file.txt') assert not os.path.exists(os.path.join(home_path, TEST_FILE)) assert os.path.exists(os.path.join(home_path, 'special_text/' + TEST_FILE))
def test_upload_and_remove_folder(bucket): """Test remove folder with content.""" bucket = bucket() efs = EFS(storage="s3") RANDOM_DATA.seek(0) efs.upload("very/long/path/to/be/created/" + TEST_FILE, RANDOM_DATA) key = bucket.Object("very/long/path/to/be/created/" + TEST_FILE) assert key efs.remove("very/") key = bucket.Object("very/long/path/to/be/created/" + TEST_FILE) with pytest.raises(ClientError) as e: key.get() assert e.value.response["Error"]["Code"] == "NoSuchKey"
def test_rename_file(bucket): """Test rename file.""" bucket = bucket() efs = EFS(storage="s3") RANDOM_DATA.seek(0) efs.upload(TEST_FILE, RANDOM_DATA) key = bucket.Object(TEST_FILE) assert key efs.rename(TEST_FILE, "new_test_file.txt") key = bucket.Object(TEST_FILE) with pytest.raises(ClientError) as e: key.get() assert e.value.response["Error"]["Code"] == "NoSuchKey" key = bucket.Object("new_test_file.txt") assert key
def test_move_file(bucket): """Test move file to a new folder.""" bucket = bucket() efs = EFS(storage="s3") RANDOM_DATA.seek(0) efs.upload(TEST_FILE, RANDOM_DATA) key = bucket.Object(TEST_FILE) assert key.get() efs.move(TEST_FILE, "special_text/test_file.txt") key = bucket.Object(TEST_FILE) with pytest.raises(ClientError) as e: key.get() assert e.value.response["Error"]["Code"] == "NoSuchKey" key = bucket.Object("special_text/test_file.txt") assert key.get()