예제 #1
0
import copy
from django.test import override_settings

from rest_framework.test import APIClient
from users.models import User
from saef.models import DatasetSession
from saefportal.settings import MSG_ERROR_INVALID_INPUT, MSG_ERROR_REQUIRED_INPUT, MSG_ERROR_MISSING_OBJECT_INPUT, \
    MSG_ERROR_EXISTING
from utils.test_utils import load_test_json, load_test_db
from django.test import TransactionTestCase
from analyzer.celery_conf import app
from celery.contrib.testing.worker import start_worker

test_data = load_test_json('restapi')


def setUpTestDatabase():
    load_test_db('restapi', 'test_dataset_session')


class DatasetSessionStartTests(TransactionTestCase):
    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        cls.celery_worker = start_worker(app)
        cls.celery_worker.__enter__()

    @classmethod
    def tearDownClass(cls):
        super().tearDownClass()
        cls.celery_worker.__exit__(None, None, None)
예제 #2
0
import copy

from django.core.exceptions import ObjectDoesNotExist
from django.test import TestCase
from django.urls import reverse

from ..models import Dataset
from utils.test_utils import load_test_json, load_test_db
from users.models import User

test_data = load_test_json('saef')

EDIT_DATASET_TEMPLATE_NAME = 'dataset/dataset_detail.html'
PREVIEW_DATASET_TEMPLATE_NAME = 'dataset/dataset_preview.html'


class ManageEditDatasetViewTests(TestCase):
    def assertDatasetStructuralEquivalence(self, actual_dataset, expected_dataset, expected_sql,
                                           expected_dataset_access_method, expected_table):
        self.assertEqual(actual_dataset["connection"], str(expected_dataset.connection.pk))
        self.assertEqual(actual_dataset["job"], str(expected_dataset.job.pk))
        self.assertEqual(actual_dataset["sequence_in_job"], str(expected_dataset.sequence_in_job))
        self.assertEqual(actual_dataset["dataset_name"], str(expected_dataset.dataset_name))
        self.assertEqual(actual_dataset["dataset_type"], str(expected_dataset.dataset_type))
        self.assertEqual(actual_dataset["query_timeout"], str(expected_dataset.query_timeout))
        self.assertEqual(expected_dataset_access_method, expected_dataset.dataset_access_method)
        self.assertEqual(expected_sql, expected_dataset.dataset_extraction_sql)
        self.assertEqual(expected_table, expected_dataset.dataset_extraction_table)

    @classmethod
    def setUpTestData(cls):
예제 #3
0
import json
import os
import copy

from django.urls import reverse
from django.test import TestCase

from pathlib import Path
from ..models import User
from utils.test_utils import load_test_json, load_test_database

from saefportal.settings import MSG_SUCCESS_USER_ACTIVATED, MSG_SUCCESS_USER_DEACTIVATED

test_data = load_test_json('users')
    
class AdminViewTest(TestCase):
    """
    Testcases for admin system
    """
    def setUp(self):
        self.data = copy.deepcopy(test_data)
    
    def test_admin_view_get_no_access(self):
        response = self.client.get(reverse('admin'))
        self.assertRedirects(response, '/admin/login/?next=/user/admin/')
        
    def test_admin_view_get_success(self):
        # Creates an user to be rendered on the admin page
        self.client.post('/user/register/', self.data['user'], follow=True)