aboutsummaryrefslogtreecommitdiff
path: root/tests/integration/test_general.py
blob: 76cc754a20b0fbe66429b7b44e9d5146c6e1d54d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
"""Integration tests for some 'general' API endpoints"""
import os
import unittest

from unittest import mock
from gn3.app import create_app


class GeneralAPITest(unittest.TestCase):
    """Test cases for 'general' API endpoints"""
    def setUp(self):
        self.app = create_app().test_client()

    def test_metadata_endpoint_exists(self):
        """Test that /metadata/upload exists"""
        response = self.app.post("/metadata/upload/d41d86-e4ceEo")
        self.assertEqual(response.status_code, 400)
        self.assertEqual(response.get_json(),
                         {"status": 128,
                          "error": "Please provide a file!"})

    @mock.patch("gn3.api.general.extract_uploaded_file")
    def test_metadata_file_upload(self, mock_extract_upload):
        """Test correct upload of file"""
        mock_extract_upload.return_value = {
            "status": 0,
            "token": "d41d86-e4ceEo",
        }
        gzip_file = os.path.abspath(os.path.join(
            os.path.dirname(__file__),
            "../unit/upload-data.tar.gz"))
        response = self.app.post("/metadata/upload/d41d86-e4ceEo",
                                 data={"file": (gzip_file,
                                                "upload-data.tar.gz")})
        self.assertEqual(response.status_code, 201)
        self.assertEqual(response.get_json(),
                         {"status": 0,
                          "token": "d41d86-e4ceEo"})

    def test_metadata_file_wrong_upload(self):
        """Test that incorrect upload return correct status code"""
        response = self.app.post("/metadata/upload/d41d86-e4ceEo",
                                 data={"file": (__file__,
                                                "my_file")})
        self.assertEqual(response.status_code, 500)
        self.assertEqual(response.get_json(),
                         {"status": 128,
                          "error": "gzip failed to unpack file"})