def test_refresh(self): new_boto_assignment = BotoAssignmentFactory(AssignmentStatus="NEW STATUS") get_assignment = mturk.connection.get_assignment = MagicMock(return_value=[new_boto_assignment]) assignment = AssignmentFactory(boto_assignment__AssignmentId="theID") assignment.refresh() get_assignment.assert_called_with("theID") assert assignment.status == "NEW STATUS"
def test_grant_bonus(self): grant_bonus = mturk.connection.grant_bonus = MagicMock() Price = mturk.Price = MagicMock(side_effect=lambda a, c: "{0}{1}".format(a, c)) assignment = AssignmentFactory() assignment.grant_bonus(0.25, "the reason") Price.assert_called_with(0.25, "USD") grant_bonus.assert_called_with( assignment_id=assignment.id, worker_id=assignment.worker.id, bonus_price = "0.25USD", reason = "the reason" )
def test_reject_reviewable(self): assignment = AssignmentFactory(boto_assignment__AssignmentStatus="Reviewable") self.setForUpApproveReject(assignment) assignment.reject("no good") self.reject.assert_called_with(assignment.id, feedback="no good") assert assignment.refresh.called
def test_approve_rejected(self): assignment = AssignmentFactory(boto_assignment__AssignmentStatus="Rejected") self.setForUpApproveReject(assignment) assignment.approve("thanks") self.approve_rejected.assert_called_with(assignment.id, feedback="thanks") assert assignment.refresh.called