def test_assignment(self):
b = UserBadge.create(u'all users - ' + tt_make_str(), u'', False, u'')
t = Treatment.create(tt_make_str(), source_badges=[b], variant_count=2)
sb, assignments, unassigned = get_assignments_by_source_badge(t)[0]
assert sb == b
ass1, ass2 = assignments
assert len(ass1) == 0
assert len(ass2) == 0
assert len(unassigned) == 0
first_users = [tt_make_user() for _ in range(5)]
for u in first_users:
b.assign(u, u)
# sanity check
assert set(u.id for u in b.users) == set(u.id for u in first_users)
sb, assignments, unassigned = get_assignments_by_source_badge(t)[0]
assert sb == b
ass1, ass2 = assignments
assert len(ass1) == 0
assert len(ass2) == 0
assert len(unassigned) == len(first_users)
assert assign_users(t)
sb, assignments, unassigned = get_assignments_by_source_badge(t)[0]
assert sb == b
ass1, ass2 = assignments
assert len(ass1) + len(ass2) == len(first_users)
assert (len(ass1) == len(ass2) - 1) or (len(ass1) == len(ass2) + 1)
assert len(unassigned) == 0
assert not assign_users(t)
new_users = [tt_make_user() for _ in range(5)]
for u in new_users:
b.assign(u, u)
sb, assignments, unassigned = get_assignments_by_source_badge(t)[0]
assert sb == b
ass1, ass2 = assignments
assert len(ass1) + len(ass2) == len(first_users)
assert (len(ass1) == len(ass2) - 1) or (len(ass1) == len(ass2) + 1)
assert len(unassigned) == len(new_users)
assert assign_users(t)
sb, assignments, unassigned = get_assignments_by_source_badge(t)[0]
assert sb == b
ass1, ass2 = assignments
assert len(ass1) + len(ass2) == len(first_users) + len(new_users)
# The following assert is disabled as it fails non-deterministically
# (see #601). It can be reenabled once this issue is fixed.
#assert len(ass1) == len(ass2)
assert len(unassigned) == 0
assert not assign_users(t)
def assigned(self, key):
treatment = model.Treatment.find(key)
if not treatment:
return ret_abort(_("Could not find the entity '%s'") % id,
code=404)
assignments = [{
'source_badge': source_badge,
'variants': current_assignment,
} for source_badge, current_assignment, unassigned in (
get_assignments_by_source_badge(treatment))]
data = {
'assignments': assignments,
'treatment': treatment,
}
return render('treatment/assigned.html', data)
def assigned(self, key):
treatment = model.Treatment.find(key)
if not treatment:
return ret_abort(_("Could not find the entity '%s'") % id,
code=404)
assignments = [
{
'source_badge': source_badge,
'variants': current_assignment,
}
for source_badge, current_assignment, unassigned in (
get_assignments_by_source_badge(treatment))]
data = {
'assignments': assignments,
'treatment': treatment,
}
return render('treatment/assigned.html', data)