def test_post_image_json_utf16(module_scoped_container_getter):
multipart_form_data = build_image_multipart("Pictures/",
abstract_test.utf16_json)
json_response = abstract_test.post_multipart(abstract_test.image,
multipart_form_data, 400)
assert json_response["error"] == "Invalid encoding in file " \
+ abstract_test.utf16_json + ", valid encodings are UTF-8 and US-ASCII"
def test_post_categorize_invalid_categories_file(module_scoped_container_getter):
multipart_form_data = abstract_test.build_base_multipart_images(abstract_test.categorize, files, "Pictures/")
multipart_form_data.append(('categories', (str("invalid2.pdf")
, open('InvalidFiles/' + "Invalid2.pdf", 'rb')
, 'text/plain')))
json_response = abstract_test.post_multipart(abstract_test.categorize, multipart_form_data, 400)
assert json_response["error"] == "Invalid extension, allowed extensions are: ['csv', 'json']"
def test_post_categorize_json(module_scoped_container_getter):
multipart_form_data = build_categorize_multipart("Pictures/", abstract_test.example_json)
json_response = abstract_test.post_multipart(abstract_test.categorize, multipart_form_data, 201)
for f in files:
if f != "sources.txt":
assert ((json_response[f] == "an apple") or (json_response[f] == "a dog"))
assert len(json_response) == len(files) - 1
def test_post_video_csv(module_scoped_container_getter):
multipart_form_data = build_video_multipart(file, abstract_test.example_csv, "Videos/")
json_response = abstract_test.post_multipart(abstract_test.video, multipart_form_data, 201)
with open('CategoryFiles/' + abstract_test.example_csv, 'r') as f:
reader = csv.reader(f)
categories = next(reader)
is_valid_response(categories, json_response)
def test_post_video_json(module_scoped_container_getter):
multipart_form_data = build_video_multipart(file, abstract_test.example_json, "Videos/")
json_response = abstract_test.post_multipart(abstract_test.video, multipart_form_data, 201)
with open('CategoryFiles/' + abstract_test.example_json) as json_file:
categories = json.load(json_file)['categories']
is_valid_response(categories, json_response)
assert len(json_response) == len(categories)
def test_post_image_invalid_image_files(module_scoped_container_getter):
multipart_form_data = abstract_test\
.build_base_multipart_images(abstract_test.image, os.listdir("InvalidFiles"), "InvalidFiles/")
multipart_form_data.append(
('categories', (str(abstract_test.example_json),
open("CategoryFiles/" + abstract_test.example_json,
'rb'), 'text/plain')))
json_response = abstract_test.post_multipart(abstract_test.image,
multipart_form_data, 400)
assert json_response[
"error"] == "Invalid extension, allowed extensions are: ['png', 'jpg', 'jpeg']"
def test_post_image_json(module_scoped_container_getter):
multipart_form_data = build_image_multipart("Pictures/",
abstract_test.example_json)
json_response = abstract_test.post_multipart(abstract_test.image,
multipart_form_data, 201)
with open('CategoryFiles/' + abstract_test.example_json) as json_file:
categories = json.load(json_file)['categories']
matches = 0
for element in categories:
if json_response[element] in files:
matches += 1
assert len(json_response) == len(categories)
assert len(categories) == matches
def test_post_image_csv(module_scoped_container_getter):
multipart_form_data = build_image_multipart("Pictures/",
abstract_test.example_csv)
json_response = abstract_test.post_multipart(abstract_test.image,
multipart_form_data, 201)
with open('CategoryFiles/' + abstract_test.example_csv, 'r') as f:
reader = csv.reader(f)
categories = next(reader)
matches = 0
for element in categories:
if json_response[element] in files:
matches += 1
assert len(json_response) == len(categories)
assert len(categories) == matches
def test_post_categorize_invalid_image_files(module_scoped_container_getter):
multipart_form_data = abstract_test.build_base_multipart_images(abstract_test.categorize, os.listdir('InvalidFiles'), "InvalidFiles/")
json_response = abstract_test.post_multipart(abstract_test.categorize, multipart_form_data, 400)
assert json_response["error"] == "Invalid extension, allowed extensions are: ['png', 'jpg', 'jpeg']"
def test_post_video_invalid_video_file(module_scoped_container_getter):
multipart_form_data = build_video_multipart('Invalid2.pdf', abstract_test.example_json, "InvalidFiles/")
json_response = abstract_test.post_multipart(abstract_test.video, multipart_form_data, 400)
assert json_response["error"] == "Invalid extension, allowed extensions are: {'mp4'}"
def test_post_video_json_utf16(module_scoped_container_getter):
multipart_form_data = build_video_multipart(file, abstract_test.utf16_json, "Videos/")
json_response = abstract_test.post_multipart(abstract_test.video, multipart_form_data, 400)
assert json_response['error'] == 'Invalid encoding in file ' \
+ abstract_test.utf16_json + ', valid encodings are UTF-8 and US-ASCII'
