Exemple #1
0
    def setUp(self):
        """Prepare environment for test."""
        from resolwe.elastic.builder import index_builder

        super().setUp()

        # Prepare Elastic search indices.
        index_builder.discover_indexes()
        index_builder.create_mappings()
        index_builder.unregister_signals()
        index_builder.register_signals()

        # Directories need to be recreated here in case a previous
        # TestCase deleted them. Moving this logic into the test runner
        # and manager infrastructure would not work, because the manager
        # and listener can't know where the testcase boundaries are,
        # they just see a series of data objects; deleting too soon
        # might cause problems for some tests. The runner does not have
        # any code between tests, so could only remove data at the very
        # end, by which time it's already too late, since some tests may
        # deal specifically with the purging functionality and should
        # start in a clean environment, without the sediment
        # (e.g. jsonout.txt, stdout.txt) from previous tests.
        os.makedirs(FLOW_EXECUTOR_SETTINGS['DATA_DIR'], exist_ok=True)
        os.makedirs(FLOW_EXECUTOR_SETTINGS['UPLOAD_DIR'], exist_ok=True)
        os.makedirs(FLOW_EXECUTOR_SETTINGS['RUNTIME_DIR'], exist_ok=True)

        self._keep_data = settings.FLOW_MANAGER_KEEP_DATA

        self.addCleanup(self._clean_up)
Exemple #2
0
    def setUp(self):
        from .test_app.elastic_indexes import TestSearchIndex, TestAnalyzerSearchIndex

        super(IndexTest, self).setUp()

        apps.clear_cache()
        call_command('migrate', verbosity=0, interactive=False, load_initial_data=False)

        index_builder.indexes = [
            TestSearchIndex(),
            TestAnalyzerSearchIndex(),
        ]

        index_builder.register_signals()
Exemple #3
0
    def setUp(self):
        from .test_app.models import TestModel
        from .test_app.elastic_indexes import TestSearchIndex
        from .test_app.viewsets import TestViewSet

        super().setUp()

        apps.clear_cache()
        call_command("migrate", verbosity=0, interactive=False)

        index_builder.indexes = [TestSearchIndex()]
        index_builder.register_signals()

        # Prepare users and groups
        user_model = get_user_model()
        self.user_1 = user_model.objects.create(username="******")
        self.user_2 = user_model.objects.create(username="******")
        group = Group.objects.create(name="group")
        group.user_set.add(self.user_2)

        tzone = get_current_timezone()
        # Prepare test data
        test_obj_1 = TestModel.objects.create(
            name="Object name 1",
            number=43,
            date=datetime.datetime(2018, 1, 1, 0, 0, tzinfo=tzone),
        )
        test_obj_2 = TestModel.objects.create(
            name="Object name 2",
            number=44,
            date=datetime.datetime(2017, 1, 1, 0, 0, tzinfo=tzone),
        )
        test_obj_3 = TestModel.objects.create(
            name="Object name 3",
            number=45,
            date=datetime.datetime(2016, 1, 1, 0, 0, tzinfo=tzone),
        )

        # Assing permissions
        assign_perm("view_testmodel", self.user_1, test_obj_1)
        assign_perm("view_testmodel", group, test_obj_2)
        assign_perm("view_testmodel", AnonymousUser(), test_obj_3)

        # Prepare test viewset
        self.test_viewset = TestViewSet.as_view(actions={
            "post": "list_with_post",
        })
Exemple #4
0
    def setUp(self):
        from .test_app.models import TestModel
        from .test_app.elastic_indexes import TestSearchIndex
        from .test_app.viewsets import TestViewSet

        super().setUp()

        apps.clear_cache()
        call_command('migrate',
                     verbosity=0,
                     interactive=False,
                     load_initial_data=False)

        index_builder.indexes = [TestSearchIndex()]
        index_builder.register_signals()

        # Prepare users and groups
        user_model = get_user_model()
        self.user_1 = user_model.objects.create(username='******')
        self.user_2 = user_model.objects.create(username='******')
        group = Group.objects.create(name='group')
        group.user_set.add(self.user_2)

        # Prepare test data
        test_obj_1 = TestModel.objects.create(name='Object name 1',
                                              number=43,
                                              date=datetime.datetime(
                                                  2018, 1, 1, 0, 0))
        test_obj_2 = TestModel.objects.create(name='Object name 2',
                                              number=44,
                                              date=datetime.datetime(
                                                  2017, 1, 1, 0, 0))
        test_obj_3 = TestModel.objects.create(name='Object name 3',
                                              number=45,
                                              date=datetime.datetime(
                                                  2016, 1, 1, 0, 0))

        # Assing permissions
        assign_perm('view_testmodel', self.user_1, test_obj_1)
        assign_perm('view_testmodel', group, test_obj_2)
        assign_perm('view_testmodel', AnonymousUser(), test_obj_3)

        # Prepare test viewset
        self.test_viewset = TestViewSet.as_view(actions={
            'post': 'list_with_post',
        })
    def setUp(self):
        from .test_app.models import TestModel
        from .test_app.elastic_indexes import TestSearchIndex
        from .test_app.viewsets import TestViewSet

        super().setUp()

        apps.clear_cache()
        call_command('migrate', verbosity=0, interactive=False, load_initial_data=False)

        index_builder.indexes = [TestSearchIndex()]
        index_builder.register_signals()

        # Prepare users and groups
        user_model = get_user_model()
        self.user_1 = user_model.objects.create(username='******')
        self.user_2 = user_model.objects.create(username='******')
        group = Group.objects.create(name='group')
        group.user_set.add(self.user_2)

        tzone = get_current_timezone()
        # Prepare test data
        test_obj_1 = TestModel.objects.create(name='Object name 1', number=43,
                                              date=datetime.datetime(2018, 1, 1, 0, 0, tzinfo=tzone))
        test_obj_2 = TestModel.objects.create(name='Object name 2', number=44,
                                              date=datetime.datetime(2017, 1, 1, 0, 0, tzinfo=tzone))
        test_obj_3 = TestModel.objects.create(name='Object name 3', number=45,
                                              date=datetime.datetime(2016, 1, 1, 0, 0, tzinfo=tzone))

        # Assing permissions
        assign_perm('view_testmodel', self.user_1, test_obj_1)
        assign_perm('view_testmodel', group, test_obj_2)
        assign_perm('view_testmodel', AnonymousUser(), test_obj_3)

        # Prepare test viewset
        self.test_viewset = TestViewSet.as_view(actions={
            'post': 'list_with_post',
        })