aboutsummaryrefslogtreecommitdiff
path: root/tests/integration/test_datasets.py
blob: 255d998a96c2001dd217478c5dea49032a5e23a7 (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
"""this module contains integration tests for datasets"""
from unittest import TestCase
from unittest import mock

from collections import namedtuple
from gn3.app import create_app


class DatasetIntegrationTests(TestCase):
    """class contains integration tests for datasets"""

    def setUp(self):
        self.app = create_app().test_client()

    @mock.patch("gn3.api.datasets.create_dataset")
    def test_create_dataset(self, mock_dataset):
        """test for creating dataset object"""
        mock_dataset_creator = namedtuple(
            'ProbeSet', ["dataset_name", "dataset_type"])
        new_dataset = mock_dataset_creator("HC_M2_0606_P", "ProbeSet")
        mock_dataset.return_value = new_dataset
        response = self.app.get(
            "/api/dataset/create/HC_M2_0606_P/", follow_redirects=True)
        mock_dataset.assert_called_once_with(
            dataset_type=None, dataset_name="HC_M2_0606_P")
        results = response.get_json()["dataset"]
        self.assertEqual(results[1], "ProbeSet")
        self.assertEqual(response.status_code, 200)