def test_w4_download(dummy_output_data, rf):
"""
test download output data from W4
"""
url_var = "/action/get_full_output_data?job_id=138thg-eg98198-sf98gy3-feh8h8"
response = get_test(url_var, w4_output_controller_download, rf)
assert response.status_code == 200
assert response.content.decode("utf-8") == SAMPLE_OUTPUT["Content"]
def test_render_logs_page(rf):
"""
:param rf:
:return:
"""
url_var = "logs_page/"
response = get_test(url_var, render_log_page, rf)
assert response.status_code == 200
def test_render_logs_data(rf):
"""
:param rf:
:return:
"""
url_var = "logs_data/"
response = get_test(url_var, log_data_controller, rf)
assert response.status_code == 200
def test_get_interfaces(rf):
"""
:param rf:
:return:
"""
url_var = "get_interfaces/"
response = get_test(url_var, get_interfaces, rf)
assert response.status_code == 200
def test_get_file(rf):
"""
This test imitates getting a file from
Brain.Files
:param rf:
:return:
"""
url_var = "file_download/{}/".format(SAMPLE_FILE_ID)
response = get_test(url_var, get_file, rf, target_id=SAMPLE_FILE_ID)
assert response.status_code == 200
assert SAMPLE_FILE_ID in response['Content-Disposition']
def test_desired_plugin_state_activate(rf):
desired_state = "activate"
url_var = "/desired_plugin_state/?plugin_id_list=%22{}%22&desired_state={}".format(
plugins[4]['id'], desired_state)
response = get_test(url_str=url_var,
function_obj=desired_plugin_state_controller,
rf=rf)
request_id = rf.get(url_var).GET['plugin_id_list']
assert "replaced" in str(response.content)
assert request_id.replace('"', '') == plugins[4]['id']
assert rf.get(url_var).GET['desired_state'] == desired_state
assert response.status_code == 200
def test_del_file_from_list(rf):
"""
This test imitates deleting file from file list
:param rf:
:return:
"""
url_var = "del_file_upload/{}/".format(SAMPLE_FILE_ID)
response = get_test(url_var,
del_file_from_list,
rf,
target_id=SAMPLE_FILE_ID)
assert response.status_code == 200
def test_get_file_list(rf):
"""
This test imitates populating a list of
files to the ui from Brain.Files
:param rf:
:return:
"""
binary.put_buffer(
SAMPLE_FILE_ID,
read_test_file(SAMPLE_FILE_ID, BACKEND_DIR + "/Backend_tests/"))
url_var = "file_listing/"
response = get_test(url_var, get_file_listing, rf)
db_file_list = ast.literal_eval(response.content.decode())
assert response.status_code == 200
assert SAMPLE_FILE_ID in db_file_list
def test_file_upload(rf):
"""
This test imitates uploading a file
:param rf: request factory
:return: status code
"""
url_var = "file_upload/"
response = get_test(url_var, file_upload_list, rf)
assert response.status_code == 200
with pytest.raises(json.JSONDecodeError):
post_data = json.loads(str(SAMPLE_FILE_ID))
response = post_test(url_var, post_data, file_upload_list, rf)
assert response.status_code == 200
response = post_test(url_var, {}, file_upload_list, rf)
assert response.status_code == 200
def test_edit_target_delete_get(rf):
"""
This test imitates deleting a target from the
edit target form and doing a request.GET
"""
target_key = ""
for target_item in get_brain_targets():
target_key = target_item["id"]
url_var = "delete_target_row/{}/".format(target_key)
response = get_test(url_var,
delete_specific_target,
rf,
target_id=target_key)
assert response.status_code == 302
assert response.url == "/"