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
