def new_task(project_id): """Return a new task for a project.""" # Check if the request has an arg: try: tasks, timeout, cookie_handler = _retrieve_new_task(project_id) if type(tasks) is Response: return tasks user_id_or_ip = get_user_id_or_ip() # If there is a task for the user, return it if tasks is not None: guard = ContributionsGuard(sentinel.master, timeout=timeout) for task in tasks: guard.stamp(task, user_id_or_ip) if not guard.check_task_presented_timestamp( task, user_id_or_ip): guard.stamp_presented_time(task, user_id_or_ip) else: # user returning back for the same task # original presented time has not expired yet # to continue original presented time, extend expiry guard.extend_task_presented_timestamp_expiry( task, user_id_or_ip) data = [ TaskAuth.dictize_with_access_control(task) for task in tasks ] add_task_signature(data) if len(data) == 0: response = make_response(json.dumps({})) elif len(data) == 1: response = make_response(json.dumps(data[0])) else: response = make_response(json.dumps(data)) response.mimetype = "application/json" cookie_handler(response) return response return Response(json.dumps({}), mimetype="application/json") except Exception as e: return error.format_exception(e, target='project', action='GET')
class TestContributionsGuard(object): def setUp(self): sentinel = Sentinel(app=FakeApp()) self.connection = sentinel.master self.connection.flushall() self.guard = ContributionsGuard(self.connection) self.anon_user = {'user_id': None, 'user_ip': '127.0.0.1'} self.auth_user = {'user_id': 33, 'user_ip': None} self.task = Task(id=22) # Task requested guard tests def test_stamp_registers_specific_user_id_and_task(self): key = 'pybossa:task_requested:user:33:task:22' self.guard.stamp(self.task, self.auth_user) assert key in self.connection.keys(), self.connection.keys() def test_stamp_registers_specific_user_ip_and_task_if_no_id_provided(self): key = 'pybossa:task_requested:user:127.0.0.1:task:22' self.guard.stamp(self.task, self.anon_user) assert key in self.connection.keys(), self.connection.keys() def test_stamp_expires_in_one_hour(self): key = 'pybossa:task_requested:user:33:task:22' ONE_HOUR = 60 * 60 self.guard.stamp(self.task, self.auth_user) assert self.connection.ttl(key) == ONE_HOUR, self.connection.ttl(key) @patch('pybossa.contributions_guard.make_timestamp') def test_stamp_adds_a_timestamp_when_the_task_is_stamped( self, make_timestamp): make_timestamp.return_value = "now" key = 'pybossa:task_requested:user:127.0.0.1:task:22' self.guard.stamp(self.task, self.anon_user) assert self.connection.get(key) == 'now' def test_check_task_stamped_returns_False_for_non_stamped_task(self): assert self.guard.check_task_stamped(self.task, self.auth_user) is False def test_check_task_stamped_returns_True_for_auth_user_who_requested_task( self): self.guard.stamp(self.task, self.auth_user) assert self.guard.check_task_stamped(self.task, self.auth_user) is True def test_check_task_stamped_returns_True_for_anon_user_who_requested_task( self): self.guard.stamp(self.task, self.anon_user) assert self.guard.check_task_stamped(self.task, self.anon_user) is True def test_retrieve_timestamp_returns_None_for_non_stamped_task(self): assert self.guard.retrieve_timestamp(self.task, self.auth_user) is None @patch('pybossa.contributions_guard.make_timestamp') def test_retrieve_timestamp_returs_the_timestamp_for_stamped_task( self, make_timestamp): make_timestamp.return_value = "now" self.guard.stamp(self.task, self.auth_user) assert self.guard.retrieve_timestamp(self.task, self.auth_user) == 'now' # Task presented guard tests def test_stamp_presented_time_registers_specific_user_id_and_task(self): key = 'pybossa:task_presented:user:33:task:22' self.guard.stamp_presented_time(self.task, self.auth_user) assert key in self.connection.keys(), self.connection.keys() def test_stamp_presented_time_registers_user_as_None_and_task_if_no_id_provided( self): key = 'pybossa:task_presented:user:None:task:22' self.guard.stamp_presented_time(self.task, self.anon_user) assert key in self.connection.keys(), self.connection.keys() def test_stamp_presented_time_expires_in_one_hour(self): key = 'pybossa:task_presented:user:33:task:22' ONE_HOUR = 60 * 60 self.guard.stamp_presented_time(self.task, self.auth_user) assert self.connection.ttl(key) == ONE_HOUR, self.connection.ttl(key) @patch('pybossa.contributions_guard.make_timestamp') def test_stamp_presented_time_adds_a_timestamp_when_the_task_is_stamped( self, make_timestamp): make_timestamp.return_value = "now" key = 'pybossa:task_presented:user:33:task:22' self.guard.stamp_presented_time(self.task, self.auth_user) assert self.connection.get(key) == 'now' def test_check_task_presented_stamped_returns_False_for_non_stamped_task( self): assert self.guard.check_task_presented_timestamp( self.task, self.auth_user) is False def test_check_task_presented_stamped_returns_True_for_auth_user_who_was_presented_task( self): self.guard.stamp_presented_time(self.task, self.auth_user) assert self.guard.check_task_presented_timestamp( self.task, self.auth_user) is True def test_check_task_presented_stamped_returns_True_for_anon_user_who_was_presented_task( self): self.guard.stamp_presented_time(self.task, self.anon_user) assert self.guard.check_task_presented_timestamp( self.task, self.anon_user) is True def test_retrieve_presented_timestamp_returns_None_for_non_stamped_task( self): assert self.guard.retrieve_presented_timestamp(self.task, self.anon_user) is None @patch('pybossa.contributions_guard.make_timestamp') def test_retrieve_presented_timestamp_returs_the_timestamp_for_stamped_task( self, make_timestamp): make_timestamp.return_value = "now" self.guard.stamp_presented_time(self.task, self.auth_user) assert self.guard.retrieve_presented_timestamp(self.task, self.auth_user) == 'now'