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)
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):
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)