class WebUserTests:
def setUp(self):
self.experiment = Experiment(name='backgroundcolor',
state=ENABLED_STATE)
self.experiment.save()
self.request = request_factory.get('/')
self.request.session = DatabaseSession()
def tearDown(self):
self.experiment.delete()
def confirm_human(self, experiment_user):
pass
def participants(self, alternative):
return self.experiment.participant_count(alternative)
def enrollment_initially_none(self, ):
experiment_user = create_user(self.request)
self.assertEqual(experiment_user.get_enrollment(self.experiment), None)
def test_user_enrolls(self):
experiment_user = create_user(self.request)
experiment_user.set_enrollment(self.experiment, TEST_ALTERNATIVE)
self.assertEqual(experiment_user.get_enrollment(self.experiment),
TEST_ALTERNATIVE)
def test_record_goal_increments_counts(self):
experiment_user = create_user(self.request)
self.confirm_human(experiment_user)
experiment_user.set_enrollment(self.experiment, TEST_ALTERNATIVE)
self.assertEqual(
self.experiment.goal_count(TEST_ALTERNATIVE, TEST_GOAL), 0)
experiment_user.record_goal(TEST_GOAL)
self.assertEqual(
self.experiment.goal_count(TEST_ALTERNATIVE, TEST_GOAL), 1)
def test_can_record_goal_multiple_times(self):
experiment_user = create_user(self.request)
self.confirm_human(experiment_user)
experiment_user.set_enrollment(self.experiment, TEST_ALTERNATIVE)
experiment_user.record_goal(TEST_GOAL)
experiment_user.record_goal(TEST_GOAL)
experiment_user.record_goal(TEST_GOAL)
self.assertEqual(
self.experiment.goal_count(TEST_ALTERNATIVE, TEST_GOAL), 1)
def test_counts_increment_immediately_once_confirmed_human(self):
experiment_user = create_user(self.request)
self.confirm_human(experiment_user)
experiment_user.set_enrollment(self.experiment, TEST_ALTERNATIVE)
self.assertEqual(self.participants(TEST_ALTERNATIVE), 1,
"Did not count participant after confirm human")
class WebUserTests:
def setUp(self):
self.experiment = Experiment(name=EXPERIMENT_NAME, state=ENABLED_STATE)
self.experiment.save()
self.request = request_factory.get('/')
self.request.session = DatabaseSession()
def tearDown(self):
self.experiment.delete()
def confirm_human(self, experiment_user):
pass
def participants(self, alternative):
return self.experiment.participant_count(alternative)
def enrollment_initially_none(self, ):
experiment_user = participant(self.request)
self.assertEqual(experiment_user.get_alternative(EXPERIMENT_NAME),
None)
def test_user_enrolls(self):
experiment_user = participant(self.request)
experiment_user.set_alternative(EXPERIMENT_NAME, TEST_ALTERNATIVE)
self.assertEqual(experiment_user.get_alternative(EXPERIMENT_NAME),
TEST_ALTERNATIVE)
def test_record_goal_increments_counts(self):
experiment_user = participant(self.request)
self.confirm_human(experiment_user)
experiment_user.set_alternative(EXPERIMENT_NAME, TEST_ALTERNATIVE)
self.assertEqual(
self.experiment.goal_count(TEST_ALTERNATIVE, TEST_GOAL), 0)
experiment_user.goal(TEST_GOAL)
self.assertEqual(
self.experiment.goal_count(TEST_ALTERNATIVE, TEST_GOAL), 1)
def test_can_record_goal_multiple_times(self):
experiment_user = participant(self.request)
self.confirm_human(experiment_user)
experiment_user.set_alternative(EXPERIMENT_NAME, TEST_ALTERNATIVE)
experiment_user.goal(TEST_GOAL)
experiment_user.goal(TEST_GOAL)
experiment_user.goal(TEST_GOAL)
self.assertEqual(
self.experiment.goal_count(TEST_ALTERNATIVE, TEST_GOAL), 1)
def test_counts_increment_immediately_once_confirmed_human(self):
experiment_user = participant(self.request)
self.confirm_human(experiment_user)
experiment_user.set_alternative(EXPERIMENT_NAME, TEST_ALTERNATIVE)
self.assertEqual(self.participants(TEST_ALTERNATIVE), 1,
"Did not count participant after confirm human")
def test_visit_increases_goal(self):
experiment_user = participant(self.request)
self.confirm_human(experiment_user)
experiment_user.set_alternative(EXPERIMENT_NAME, TEST_ALTERNATIVE)
experiment_user.visit()
self.assertEqual(
self.experiment.goal_distribution(TEST_ALTERNATIVE,
VISIT_COUNT_GOAL), {1: 1})
def test_visit_twice_increases_once(self):
experiment_user = participant(self.request)
self.confirm_human(experiment_user)
experiment_user.set_alternative(EXPERIMENT_NAME, TEST_ALTERNATIVE)
experiment_user.visit()
experiment_user.visit()
self.assertEqual(
self.experiment.goal_distribution(TEST_ALTERNATIVE,
VISIT_COUNT_GOAL), {1: 1})
