def test_create_access(self): access_data = {} access = MagicMock() process_data = {} process = MagicMock() with patch('audit_tools.audit.models.models_factory._access_factory', return_value=access) as af_mock, \ patch('audit_tools.audit.models.models_factory.cache') as cache_mock: cache_mock.get_process.return_value = process result = create_access(access_data, process_data) self.assertIn('process', access_data) self.assertEqual(cache_mock.get_process.call_count, 1) self.assertEqual(af_mock.call_count, 1) self.assertEqual(result, access)
def process_view(self, request, view_func, view_args, view_kwargs): """Preprocess request. :param request: Http request. :type request: django.http.HttpRequest :param view_func: View. :type view_func: callable :param view_args: View arguments. :type view_args: list :param view_kwargs: View keyword arguments. :type view_kwargs: dict :return: None """ self._disabled = getattr(view_func, 'disable_audit', False) if not self._disabled: try: view_data = self._extract_view_data(view_func, view_args, view_kwargs) self._blacklisted = self._check_blacklist(request.path, view_data['app']) msg = "<Process View> View:%s %s", view_data['full_name'], 'BlackList' if self._blacklisted else '' logger.debug(msg) if not self._blacklisted: self._view = view_data user = self._extract_user_data(request) # Time self._time = { 'request': datetime.datetime.now(), 'response': None } # Interlink ID interlink_id = get_access_interlink_id() # Create access access = { 'interlink_id': interlink_id, 'request': request_to_dict(request), 'response': None, 'time': self._time, 'view': self._view, 'user': user, 'custom': None, } # Extract process data self._process = extract_process_data() # Save Access self._access = create_access(access, self._process) cache.set_last_access(self._access) if not settings.RUN_ASYNC: save_access(self._access) else: save_access.apply_async((self._access, )) logger.info("<Process View> View:%s", self._view['full_name']) logger.debug("View:%s", str(self._view)) except Exception: logger.exception("<Process View>") return None