From 0a70905e5b16e372bea9fd3e13a16fbd9152a416 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Tue, 16 Feb 2021 22:17:41 +0300
Subject: Add procedure for uploading gzipped file

* gn3/file_utils.py (extract_uploaded_file): New procedure.
* tests/unit/test_file_utils.py: Test cases for ^^.
* tests/unit/upload-data.tar.gz: New test data.
---
 tests/unit/test_file_utils.py |  47 +++++++++++++++++++++++++++++++++++++++++-
 tests/unit/upload-data.tar.gz | Bin 0 -> 445 bytes
 2 files changed, 46 insertions(+), 1 deletion(-)
 create mode 100644 tests/unit/upload-data.tar.gz

(limited to 'tests/unit')

diff --git a/tests/unit/test_file_utils.py b/tests/unit/test_file_utils.py
index f5efa02..166e576 100644
--- a/tests/unit/test_file_utils.py
+++ b/tests/unit/test_file_utils.py
@@ -1,11 +1,22 @@
 """Test cases for procedures defined in file_utils.py"""
 import os
+import shutil
 import unittest
 
+from dataclasses import dataclass
+from typing import Callable
 from unittest import mock
+from gn3.file_utils import extract_uploaded_file
+from gn3.file_utils import get_dir_hash
 from gn3.file_utils import jsonfile_to_dict
 from gn3.file_utils import lookup_file
-from gn3.file_utils import get_dir_hash
+
+
+@dataclass
+class MockFile:
+    """Mock File object returned by request.files"""
+    filename: str
+    save: Callable
 
 
 class TestFileUtils(unittest.TestCase):
@@ -63,3 +74,37 @@ non-existent"""
         self.assertRaises(FileNotFoundError,
                           jsonfile_to_dict,
                           "/non-existent-dir")
+
+    @mock.patch("gn3.file_utils.secure_filename")
+    def test_extract_uploaded_file(self, mock_file):
+        """Test that the gzip file is extracted to the right location"""
+        file_loc = os.path.join(
+            os.path.dirname(__file__),
+            "upload-data.tar.gz")
+        mock_file.return_value = file_loc
+        mock_fileobj = MockFile(save=mock.MagicMock(),
+                                filename="upload-data.tar.gz")
+        result = extract_uploaded_file(mock_fileobj, "/tmp")
+        mock_fileobj.save.assert_called_once_with(file_loc)
+        mock_file.assert_called_once_with("upload-data.tar.gz")
+        # Clean up!
+        shutil.rmtree(os.path.join("/tmp",
+                                   "d41d8cd98f00b204e9800998ecf8427e"))
+        self.assertEqual(result,
+                         {"status": 0,
+                          "token": "d41d8cd98f00b204e9800998ecf8427e"})
+
+    @mock.patch("gn3.file_utils.secure_filename")
+    def test_extract_uploaded_file_non_existent_gzip(self, mock_file):
+        """Test that the right error message is returned when there is a problem
+extracting the file"""
+        file_loc = os.path.join(
+            os.path.dirname(__file__),
+            "CTtyodSTh5")  # Does not exist!
+        mock_file.return_value = file_loc
+        mock_fileobj = MockFile(save=mock.MagicMock(),
+                                filename="")
+        result = extract_uploaded_file(mock_fileobj, "/tmp")
+        self.assertEqual(result,
+                         {"status": 128,
+                          "error": "gzip failed to unpack file"})
diff --git a/tests/unit/upload-data.tar.gz b/tests/unit/upload-data.tar.gz
new file mode 100644
index 0000000..86077d9
Binary files /dev/null and b/tests/unit/upload-data.tar.gz differ
-- 
cgit v1.2.3