EDI Viewer: Checking files doesn't work

chellow/views.py

@@ -621,16 +621,16 @@ def edi_viewer_get():
 @home.route("/edi_viewer", methods=["POST"])
 def edi_viewer_post():
     segments = []
+    file_name = None
     try:
-        if "edi_file" in request.values:
+        if "edi_file" in request.files:
             file_item = req_file("edi_file")
             edi_str = str(
                 file_item.stream.read(), encoding="utf-8-sig", errors="ignore"
             )
             file_name = file_item.filename
         else:
             edi_str = req_str("edi_fragment")
-            file_name = None
 
         f = StringIO(edi_str)
         f.seek(0)
@@ -723,8 +723,9 @@ def edi_viewer_post():
         )
     except BadRequest as e:
         flash(e.description)
-        return render_template(
-            "edi_viewer.html", segments=segments, file_name=file_name
+        return make_response(
+            render_template("edi_viewer.html", segments=segments, file_name=file_name),
+            400,
         )
 
 

test/test_edi_viewer.py

@@ -0,0 +1,13 @@
+from io import BytesIO
+from utils import match
+
+
+def test_edit_viewer_post(client, mocker):
+    file_name = "sample.edi"
+    file_bytes = "MHD=1+UTLHDR:3'".encode("utf8")
+    f = BytesIO(file_bytes)
+
+    data = {"edi_file": (f, file_name)}
+    response = client.post("/edi_viewer", data=data)
+
+    match(response, 200)