def test_invoice_get_rates_inconsistent_scorelog_rates(member, store0): USER_RATE_ONE = 0.5 USER_RATE_TWO = 0.2 # Set some rate member.rate = USER_RATE_ONE member.review_rate = USER_RATE_ONE member.save() month = timezone.datetime(2014, 04, 01) submission_kwargs = { 'store': store0, 'unit': store0.units[0], 'field': SubmissionFields.TARGET, 'type': SubmissionTypes.NORMAL, 'old_value': 'foo', 'new_value': 'bar', 'submitter': member, 'translation_project': store0.translation_project, 'creation_time': month, } scorelog_kwargs = { 'wordcount': 1, 'similarity': 0, 'action_code': TranslationActionCodes.NEW, 'creation_time': month, 'user': member, 'submission': SubmissionFactory(**submission_kwargs), } ScoreLogFactory(**scorelog_kwargs) # Alter rates, producing an inconsistent state when recording the ScoreLog member.rate = USER_RATE_TWO member.review_rate = USER_RATE_TWO member.save() submission_kwargs['unit'] = store0.units[1] scorelog_kwargs['submission'] = SubmissionFactory(**submission_kwargs) ScoreLogFactory(**scorelog_kwargs) invoice = Invoice(member, FAKE_CONFIG, month=month) with pytest.raises(ValueError) as e: invoice.get_rates() assert ( 'Multiple rate values recorded for user %s' % (member.username, ) in e.value.message )
def test_is_similarity_taken_from_mt(similarity, mt_similarity): submission = SubmissionFactory.build(similarity=similarity, mt_similarity=mt_similarity) score_log = ScoreLogFactory.build(submission=submission) if submission.similarity < submission.mt_similarity: assert score_log.is_similarity_taken_from_mt() else: assert not score_log.is_similarity_taken_from_mt()
def test_invoice_get_rates_inconsistent_paidtask_rates(member, task_type, task_type_name, user_rate_attr_name, store0): USER_RATE = 0.5 PAID_TASK_RATE = 0.2 # Set some user rate setattr(member, user_rate_attr_name, USER_RATE) member.save() month = timezone.datetime(2014, 04, 01) submission_kwargs = { 'store': store0, 'unit': store0.units[0], 'field': SubmissionFields.TARGET, 'type': SubmissionTypes.NORMAL, 'old_value': 'foo', 'new_value': 'bar', 'submitter': member, 'translation_project': store0.translation_project, 'creation_time': month, } scorelog_kwargs = { 'wordcount': 1, 'similarity': 0, 'action_code': TranslationActionCodes.NEW, 'creation_time': month, 'user': member, 'submission': SubmissionFactory(**submission_kwargs), } paid_task_kwargs = { 'rate': PAID_TASK_RATE, # Note how this doesn't match user's rate 'datetime': month, 'user': member, 'task_type': task_type, } ScoreLogFactory(**scorelog_kwargs) PaidTaskFactory(**paid_task_kwargs) invoice = Invoice(member, FAKE_CONFIG, month=month) with pytest.raises(ValueError) as e: invoice.get_rates() assert ( 'Multiple %s rate values for user %s' % (task_type_name, member.username) in e.value.message )
def test_invoice_generate_add_correction(member, invoice_directory): """Tests that generating invoices multiple times for the same month + user will add corrections only once. """ from pootle_statistics.models import Submission EVENT_COUNT = 5 WORDCOUNT = 5 TRANSLATION_RATE = 0.5 INITIAL_SUBTOTAL = EVENT_COUNT * WORDCOUNT * TRANSLATION_RATE MINIMAL_PAYMENT = 20 month = get_previous_month() config = dict({ 'minimal_payment': MINIMAL_PAYMENT, }, **FAKE_CONFIG) invoice = Invoice(member, config, month=month, add_correction=True) # Set some rates member.rate = TRANSLATION_RATE member.save() # Fake some activity that will leave amounts below the minimum bar: # EVENT_COUNT * WORDCOUNT * TRANSLATION_RATE < MINIMAL_PAYMENT for i in range(EVENT_COUNT): scorelog_kwargs = { 'wordcount': WORDCOUNT, 'similarity': 0, 'action_code': TranslationActionCodes.NEW, 'creation_time': month, 'user': member, 'submission': Submission.objects.all()[i], } ScoreLogFactory(**scorelog_kwargs) # Generate an invoice first amounts = invoice.get_total_amounts() assert amounts['subtotal'] == INITIAL_SUBTOTAL assert invoice.should_add_correction(amounts['subtotal']) invoice.generate() _check_single_paidtask(invoice, INITIAL_SUBTOTAL) # Subsequent invoice generations must not add any corrections for i in range(5): invoice.get_total_amounts.cache_clear() # clears the LRU cache amounts = invoice.get_total_amounts() assert amounts['subtotal'] == 0 assert not invoice.should_add_correction(amounts['subtotal']) invoice.generate() _check_single_paidtask(invoice, INITIAL_SUBTOTAL)
def test_invoice_get_rates_scorelog_rates(member, task_type, task_type_name, user_rate_attr_name, store0): """Tests that `Invoice.get_rates()` returns the rates set for users in their `ScoreLog` entries. """ USER_RATE_ONE = 0.5 USER_RATE_TWO = 0.2 # Set some user rate setattr(member, user_rate_attr_name, USER_RATE_ONE) member.save() month = timezone.datetime(2014, 04, 01) submission_kwargs = { 'store': store0, 'unit': store0.units[0], 'field': SubmissionFields.TARGET, 'type': SubmissionTypes.NORMAL, 'old_value': 'foo', 'new_value': 'bar', 'submitter': member, 'translation_project': store0.translation_project, 'creation_time': month, } scorelog_kwargs = { 'wordcount': 1, 'similarity': 0, 'action_code': TranslationActionCodes.NEW, 'creation_time': month, 'user': member, 'submission': SubmissionFactory(**submission_kwargs), } ScoreLogFactory(**scorelog_kwargs) invoice = Invoice(member, FAKE_CONFIG, month=month) # Set user rate to something else to ensure we get the recorded rates setattr(member, user_rate_attr_name, USER_RATE_TWO) member.save() rate, review_rate, hourly_rate = invoice.get_rates() assert locals()[user_rate_attr_name] == USER_RATE_ONE
def test_get_similarity(similarity): score_log = ScoreLogFactory.build(similarity=similarity) if similarity >= SIMILARITY_THRESHOLD: assert score_log.get_similarity() == similarity else: assert score_log.get_similarity() == 0
def test_invoice_get_user_amounts(member, action_code, task_type): """Tests that `Invoice._get_user_amounts()` returns the total amount of work performed for the given user when their activities were recorded via both score logs and paid tasks. """ from pootle_statistics.models import Submission EVENT_COUNT = 5 WORDCOUNT = 5 TASK_COUNT = 5 PAID_TASK_AMOUNT = 22 month = timezone.datetime(2014, 04, 01) for i in range(EVENT_COUNT): scorelog_kwargs = { 'wordcount': WORDCOUNT, 'similarity': 0, 'action_code': action_code, 'creation_time': month, 'user': member, 'submission': Submission.objects.all()[i], } ScoreLogFactory(**scorelog_kwargs) for i in range(TASK_COUNT): paid_task_kwargs = { 'amount': PAID_TASK_AMOUNT, 'datetime': month, 'user': member, 'task_type': task_type, } PaidTaskFactory(**paid_task_kwargs) invoice = Invoice(member, FAKE_CONFIG, month=month) translated, reviewed, hours, correction = invoice._get_user_amounts(member) if (action_code == TranslationActionCodes.NEW and task_type == PaidTaskTypes.TRANSLATION): assert translated == (EVENT_COUNT * WORDCOUNT + TASK_COUNT * PAID_TASK_AMOUNT) elif action_code == TranslationActionCodes.NEW: assert translated == EVENT_COUNT * WORDCOUNT elif task_type == PaidTaskTypes.TRANSLATION: assert translated == TASK_COUNT * PAID_TASK_AMOUNT else: assert translated == 0 if (action_code == TranslationActionCodes.REVIEWED and task_type == PaidTaskTypes.REVIEW): assert reviewed == (EVENT_COUNT * WORDCOUNT + TASK_COUNT * PAID_TASK_AMOUNT) elif action_code == TranslationActionCodes.REVIEWED: assert reviewed == EVENT_COUNT * WORDCOUNT elif task_type == PaidTaskTypes.REVIEW: assert reviewed == TASK_COUNT * PAID_TASK_AMOUNT else: assert reviewed == 0 if task_type == PaidTaskTypes.HOURLY_WORK: assert hours == TASK_COUNT * PAID_TASK_AMOUNT else: assert hours == 0 if task_type == PaidTaskTypes.CORRECTION: assert correction == TASK_COUNT * PAID_TASK_AMOUNT else: assert correction == 0