Revision control
Copy as Markdown
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
import json
from cirrus import CirrusClient, NimbusError
from conftest import TestMetricsHandler
def test_one_experiment_should_enroll(client, req):
response = json.loads(client.handle_enrollment(req()))
assert len(response["enrollments"]) == 1
assert "Enrolled" in response["enrollments"][0]["status"].keys()
assert len(list(response["enrolledFeatureConfigMap"].values())) == 1
assert len(response["events"]) == 1
def test_one_experiment_should_not_enroll(client, req, request_context):
request_context["username"] = "not test"
request = req(request_context=request_context)
response = json.loads(client.handle_enrollment(request))
assert len(response["enrollments"]) == 1
assert "NotEnrolled" in response["enrollments"][0]["status"].keys()
assert len(list(response["enrolledFeatureConfigMap"].values())) == 0
assert len(response["events"]) == 0
def test_one_experiment_should_enroll_1000_times(client, req):
response = None
for i in range(0, 1000):
response = json.loads(client.handle_enrollment(req()))
assert len(response["enrollments"]) == 1
assert "Enrolled" in response["enrollments"][0]["status"].keys()
assert len(list(response["enrolledFeatureConfigMap"].values())) == 1
assert len(response["events"]) == 1
def test_failure_case_no_client_id(client, req):
request = req(client_id=None)
try:
client.handle_enrollment(request)
except NimbusError:
pass
else:
assert False, "client.handle_enrollment did not throw an error"
def test_metrics_handler(app_context, experiment, req):
test_metrics = TestMetricsHandler()
client = CirrusClient(app_context, test_metrics, [])
data = json.dumps({"data": [experiment]})
client.set_experiments(data)
json.loads(client.handle_enrollment(req()))
assert len(test_metrics.recordings) == 1