def test_remove_row_failure(client):
#get data of row to remove
csv_data = get_csv(csv_path, test_dialect)
row_data = csv_data[0]
#block file to prevent write
os.chmod(csv_path, S_IREAD | S_IRGRP | S_IROTH)
#invoke removation of row
response = client.post('/remove_row',
data=dict(row_index='0'),
follow_redirects=True)
#check if error message has appeared
assert b'ERROR! Cannot write to file!' in response.data
#confirm that row to remove still exist
csv_data_after_remove = get_csv(csv_path, test_dialect)
assert row_data[0] == csv_data_after_remove[0][0]
assert row_data[1] == csv_data_after_remove[0][1]
assert row_data[2] == csv_data_after_remove[0][2]
#check if response returned code 200
assert response.status_code == 200
#unlock file at the end of test
os.chmod(csv_path, S_IWUSR | S_IREAD)
def test_create_backup_successful(client):
#get current state of csv
csv_data = get_csv(csv_path, test_dialect)
#do backup
response = client.post('/do_bak', follow_redirects=True)
#get list of backups to find path to the latest created backup
backup_dir = backup_directory
backup_list = list(backup_dir.glob("*.bak"))
backups = []
for path in backup_list:
backups.append([str(path), path.name])
backups = sorted(backups, key=lambda tup: tup[1], reverse=True)
bak_path = backups[0][0]
#check if created backup trully exist
assert os.path.exists(bak_path)
#load backup to memory
backup_data = get_csv(bak_path, test_dialect)
#compare staged state and last backup to confirm that backup is correct
assert csv_data == backup_data
#check if response returned code 200
assert response.status_code == 200
def test_columns_mismatch():
#prepare other dialect and file
other_dialect = {
'CSV_DELIMITER': '\t',
'CSV_QUOTING': csv.QUOTE_NONE,
'CSV_DOUBLEQUOTE': False
}
other_file_path = "C:\\Users\\grzes\\Desktop\\python_dev\\pet_projects\\flatfile_ed\\tests\\other_format.csv"
save_data_set = [["Row1Column1", "Row1Column2", "Row1Column3"],
["Row2Column1", "Row2Column2", "Row2Column3"]]
save_csv(save_data_set, other_file_path, other_dialect)
#assert get_columns_number(other_file_path,test_dialect) != 3
with pytest.raises(csv.Error) as err_info:
get_csv(other_file_path, test_dialect)
assert 'Wrong format of provied CSV! Check file or review configuration' in str(
err_info.value)
def test_remove_row_success(client):
#get data of row to remove
csv_data = get_csv(csv_path, test_dialect)
row_data = csv_data[0]
#invoke removation of row
response = client.post('/remove_row',
data=dict(row_index='0'),
follow_redirects=True)
#confirm that removed row doesnt exist
csv_data_after_remove = get_csv(csv_path, test_dialect)
assert row_data[0] != csv_data_after_remove[0][0]
assert row_data[1] != csv_data_after_remove[0][1]
assert row_data[2] != csv_data_after_remove[0][2]
#check if response returned code 200
assert response.status_code == 200
def test_return_backup_failure(client):
#remove all backups
backup_dir = backup_directory
filelist = list(backup_dir.glob("*.bak"))
if len(filelist) > 0:
for f in filelist:
os.remove(f)
#do backup
resp1 = client.post('/do_bak', follow_redirects=True)
assert resp1.status_code == 200
#get list of .bak files in backup folder
backup_list = list(backup_dir.glob("*.bak"))
backup_path = backup_list[0]
#data to write
new_row = ['AAAWrite1', 'AAAWrite2', 'AAAkWrite3']
#apply change to csv file
resp2 = client.post('/add_row',
data=dict(col1=new_row[0],
col2=new_row[1],
col3=new_row[2],
sort='on'),
follow_redirects=True)
assert resp2.status_code == 200
#block file to prevent write
os.chmod(csv_path, S_IREAD | S_IRGRP | S_IROTH)
#retrun from backup
resp3 = client.post('/return_from_backup',
data=dict(backup_path=backup_path),
follow_redirects=True)
assert resp3.status_code == 200
#check if error message has appeared
assert b'ERROR! Cannot write to file!' in resp3.data
#get data from csv and check if new row exist
csv_data = get_csv(csv_path, test_dialect)
assert new_row[0] == csv_data[0][0]
assert new_row[1] == csv_data[0][1]
assert new_row[2] == csv_data[0][2]
#unlock file at the end of test
os.chmod(csv_path, S_IWUSR | S_IREAD)
def test_add_row_post_not_sort(client):
#ensure that file is readble
os.chmod(csv_path, S_IWUSR | S_IREAD)
#post new row into file
response = client.post('/add_row',
data=dict(col1='TestWrite1',
col2='TestWrite2',
col3='TestWrite3'),
follow_redirects=True)
#check if new row appeared in csv file
csv_data = get_csv(csv_path, test_dialect)
assert any("TestWrite1" in s for s in csv_data)
#check if response returned code 200
assert response.status_code == 200
def test_mod_row(client):
#get first row from CSV
csv_data = get_csv(csv_path, test_dialect)
row_data = csv_data[0]
response = client.post('/modify_row',
data=dict(row_index=0),
follow_redirects=True)
#check if modify row form has loaded
assert b'Modify row of csv file' in response.data
#check if requested value of CSV file has been loaded correctly
for cell in row_data:
assert cell.encode() in response.data
#check if response returned code 200
assert response.status_code == 200
def test_as_new_success(client):
#Send row to modify
response = client.post('/save_mod',
data=dict(submit='save_as_new',
col1='NewWrite1',
col2='NewWrite2',
col3='NewWrite3'),
follow_redirects=True)
#check if modifcation have completed successful
#get modified CSV row
csv_data = get_csv(csv_path, test_dialect)
row_data = csv_data[-1]
assert row_data[0] == 'NewWrite1'
assert row_data[1] == 'NewWrite2'
assert row_data[2] == 'NewWrite3'
#check if response returned code 200
assert response.status_code == 200
def test_return_backup_success(client):
#remove all backups
backup_dir = backup_directory
filelist = list(backup_dir.glob("*.bak"))
if len(filelist) > 0:
for f in filelist:
os.remove(f)
#do backup
resp1 = client.post('/do_bak', follow_redirects=True)
assert resp1.status_code == 200
#get list of .bak files in backup folder
backup_list = list(backup_dir.glob("*.bak"))
backup_path = backup_list[0]
#data to write
new_row = ['AAAWrite1', 'AAAWrite2', 'AAAkWrite3']
#apply change to csv file
resp2 = client.post('/add_row',
data=dict(col1=new_row[0],
col2=new_row[1],
col3=new_row[2],
sort='on'),
follow_redirects=True)
assert resp2.status_code == 200
#retrun from backup
resp3 = client.post('/return_from_backup',
data=dict(backup_path=backup_path),
follow_redirects=True)
assert resp3.status_code == 200
#get data from csv and check if new row exist
csv_data = get_csv(csv_path, test_dialect)
assert new_row[0] != csv_data[0][0]
assert new_row[1] != csv_data[0][1]
assert new_row[2] != csv_data[0][2]
def test_add_row_post_not_sort_failed(client):
#block file to prevent write
os.chmod(csv_path, S_IREAD | S_IRGRP | S_IROTH)
#perform write to capture error
response = client.post('/add_row',
data=dict(col1='ErrorWrite1',
col2='ErrorWrite2',
col3='ErrorWrite3'),
follow_redirects=True)
#check if error message has appeared
assert b'ERROR! Cannot write to file!' in response.data
#confirm that entered row does not exist
csv_data = get_csv(csv_path, test_dialect)
assert not (any("ErrorWrite1" in s for s in csv_data))
#check if response returned code 200
assert response.status_code == 200
#unlock file at the end of test
os.chmod(csv_path, S_IWUSR | S_IREAD)
def test_get_csv_exception():
with pytest.raises(FileNotFoundError) as err_info:
get_csv(test_path_wrong, test_dialect)
assert 'CSV File not provided!' in str(err_info.value)
def test_get_csv():
assert get_csv(test_path_correct, test_dialect) is not None