def test_map_for_scenario(self, mock_choice):
users_ = []
tenants = {}
for i in range(2):
tenants[str(i)] = {"name": str(i)}
for j in range(3):
users_.append({
"id": "%s_%s" % (i, j),
"tenant_id": str(i),
"endpoint": "endpoint"
})
context = {
"admin": mock.MagicMock(),
"users": users_,
"tenants": tenants,
"some_random_key": {
"nested": mock.MagicMock(),
"one_more": 10
}
}
chosen_tenant = context["tenants"][context["users"][1]["tenant_id"]]
expected_context = {
"admin": context["admin"],
"user": context["users"][1],
"tenant": chosen_tenant,
"some_random_key": context["some_random_key"]
}
self.assertEqual(expected_context,
users.UserContextMixin().map_for_scenario(context))
def setUp(self):
super(self.__class__, self).setUp()
self.mixin = users.UserContextMixin()
self.mixin.context = {
"config": {
"users": {
"user_choice_method": "random",
},
},
}