def test_handle_post_request_file_unknwon(mock_server_result, tmpdir):
# videos
m = mock.MagicMock()
files = {
"file":
FileStorage(open(PATH_TO_TEST_VIDEO, "rb"), content_type='video/mkv')
}
m.files = files
m.form = {"fps": 2, "foo": "bar"}
split_frames_folder = os.path.join(
tmpdir, "{}_split".format(os.path.basename(PATH_TO_TEST_VIDEO)))
os.mkdir(split_frames_folder
) # this folder should be deleted by handle post request
with mock.patch("mot.serving.inference.request", m):
outputs = handle_post_request(upload_folder=str(tmpdir))
assert "error" in outputs
assert outputs["error"].endswith("{'unknown': '¯\\\\_(ツ)_/¯'}")
# images
data = np.ones((300, 200, 3))
filename = "test.jpg"
filepath = os.path.join(tmpdir, filename)
cv2.imwrite(filepath, data)
m = mock.MagicMock()
files = {
"file": FileStorage(open(filepath, "rb"), content_type='image/jpg')
}
m.files = files
with mock.patch("mot.serving.inference.request", m):
outputs = handle_post_request(upload_folder=str(tmpdir))
assert "error" in outputs
assert outputs["error"].endswith("{'unknown': '¯\\\\_(ツ)_/¯'}")
def test_handle_post_request_file_image(mock_server_result, tmpdir):
data = np.array([[[0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0]]])
filename = "test.jpg"
filepath = os.path.join(tmpdir, filename)
cv2.imwrite(filepath, data)
m = mock.MagicMock()
files = {
"file": FileStorage(open(filepath, "rb"), content_type='image/jpg')
}
m.files = files
upload_folder = os.path.join(tmpdir, "upload")
os.mkdir(upload_folder)
with open(os.path.join(upload_folder, filename), "w+") as f:
f.write(
"this file should be deleted by the handle post request to be replaced"
" by the image we are uploading.")
with mock.patch("mot.serving.inference.request", m):
output = handle_post_request(upload_folder=upload_folder)
expected_output = {
"image":
output["image"],
"detected_trash": [{
"box": [0.0, 0.0, 0.1, 0.1],
"label": "bottles",
"score": 0.71
}, {
"box": [0.0, 0.0, 0.2, 0.2],
"label": "fragments",
"score": 0.71
}]
}
assert output == expected_output
def test_handle_post_request_file_video(mock_server_result, tmpdir):
m = mock.MagicMock()
files = {
"file":
FileStorage(open(PATH_TO_TEST_VIDEO, "rb"), content_type='video/mkv')
}
m.files = files
m.form = {"fps": 2, "foo": "bar"}
split_frames_folder = os.path.join(
tmpdir, "{}_split".format(os.path.basename(PATH_TO_TEST_VIDEO)))
os.mkdir(split_frames_folder
) # this folder should be deleted by handle post request
with mock.patch("mot.serving.inference.request", m):
output = handle_post_request(upload_folder=str(tmpdir))
assert len(output["detected_trash"]) == 2
assert "id" in output["detected_trash"][0]
assert "frame_to_box" in output["detected_trash"][1]
for frame, box in output["detected_trash"][1]["frame_to_box"].items():
assert isinstance(frame, int)
assert isinstance(box, list)
assert len(box) == 4
assert output["video_length"] == 6 or output["video_length"] == 7
assert output["fps"] == 2
assert "video_id" in output
def test_handle_post_request_wrong_image():
data = "{}".format(
json.dumps(
{"image:0": [[[0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0]]]}))
data = data.encode('utf-8')
m = mock.MagicMock() # here we mock flask.request
m.data = data
with pytest.raises(ValueError):
with mock.patch("mot.serving.inference.request", m):
output = handle_post_request()
def test_handle_post_request_video():
# TODO test good behavior when implemented
data = "{}".format(
json.dumps({"video": [[[0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0,
0]]]}))
data = data.encode('utf-8')
m = mock.MagicMock() # here we mock flask.request
m.data = data
with pytest.raises(NotImplementedError):
with mock.patch("mot.serving.inference.request", m):
output = handle_post_request()
def test_handle_post_request_file_other(tmpdir):
filename = "test.pdf"
filepath = os.path.join(tmpdir, filename)
with open(filepath, "w") as f:
f.write("mock data")
m = mock.MagicMock()
files = {
"file": FileStorage(open(filepath, "rb"), content_type='poulet/pdf')
}
m.files = files
upload_folder = os.path.join(tmpdir, "upload_folder")
with pytest.raises(NotImplementedError):
with mock.patch("mot.serving.inference.request", m):
output = handle_post_request(upload_folder=upload_folder)
assert os.path.isdir(
upload_folder
) # the upload_folder should be created by handle post request
def test_handle_post_request_image(mock_server_result):
image = np.ones((300, 200, 3))
data = "{}".format(json.dumps({"image": image.tolist()}))
data = data.encode('utf-8')
m = mock.MagicMock() # here we mock flask.request
m.data = data
with mock.patch("mot.serving.inference.request", m):
output = handle_post_request()
expected_output = {
"detected_trash": [{
"box": [0.0, 0.0, 0.1, 0.05],
"label": "bottles",
"score": 0.71
}, {
"box": [0.0, 0.0, 0.1, 0.1],
"label": "fragments",
"score": 0.71
}]
}
assert output == expected_output
def test_handle_post_request_file_zip(mock_server_result, tmpdir):
m = mock.MagicMock()
files = {
"file":
FileStorage(open(PATH_TO_TEST_ZIP, "rb"),
content_type='application/zip')
}
m.files = files
m.form = {"fps": 2, "foo": "bar"}
with mock.patch("mot.serving.inference.request", m):
output = handle_post_request(upload_folder=str(tmpdir))
assert len(output["detected_trash"]) == 2
assert "id" in output["detected_trash"][0]
assert "frame_to_box" in output["detected_trash"][1]
for frame, box in output["detected_trash"][1]["frame_to_box"].items():
assert isinstance(frame, int)
assert isinstance(box, list)
assert len(box) == 4
assert output["video_length"] == 6 or output["video_length"] == 7
assert output["fps"] == 2
assert "video_id" in output
def index():
if request.method == 'POST':
return handle_post_request()
return render_template("upload.html")