def test_ask_linking(self, sign, verify):
sign.return_value = "signed_parameters"
def user_conversation_id_callback(request):
payload = json.loads(request.body)
if payload["members"] == [{"id": "s4ur0n"}] and payload["channelData"] == {
"tenant": {"id": "7h3_gr347"}
}:
return (200, {}, json.dumps({"id": "d4rk_l0rd"}))
return (404, {}, json.dumps({}))
responses.add_callback(
method=responses.POST,
url="https://smba.trafficmanager.net/amer/v3/conversations",
callback=user_conversation_id_callback,
)
responses.add(
method=responses.POST,
url="https://smba.trafficmanager.net/amer/v3/conversations/d4rk_l0rd/activities",
status=200,
json={},
)
resp = self.post_webhook(user_id="s4ur0n", tenant_id="7h3_gr347")
linking_url = build_linking_url(
self.integration, self.org, "s4ur0n", "f3ll0wsh1p", "7h3_gr347"
)
data = json.loads(responses.calls[1].request.body)
assert resp.status_code == 201
assert "attachments" in data
assert data["attachments"][0]["content"] == build_linking_card(linking_url)
def test_overwrites_existing_identities(self, unsign):
Identity.objects.create(user=self.user1,
idp=self.idp,
external_id="h_p",
status=IdentityStatus.VALID)
Identity.objects.create(user=self.user2,
idp=self.idp,
external_id="g_w",
status=IdentityStatus.VALID)
unsign.return_value = {
"integration_id": self.integration.id,
"organization_id": self.org.id,
"teams_user_id": "g_w",
"team_id": "1_50l3mnly_5w34r",
"tenant_id": "th3_burr0w",
}
linking_url = build_linking_url(
self.integration,
self.org,
"g_w",
"1_50l3mnly_5w34r",
"th3_burr0w",
)
def user_conversation_id_callback(request):
payload = json.loads(request.body)
if payload["members"] == [{
"id": "g_w"
}] and payload["channelData"] == {
"tenant": {
"id": "th3_burr0w"
}
}:
return (200, {}, json.dumps({"id": "g1nny_w345l3y"}))
return (404, {}, json.dumps({}))
responses.add_callback(
method=responses.POST,
url="https://smba.trafficmanager.net/amer/v3/conversations",
callback=user_conversation_id_callback,
)
responses.add(
method=responses.POST,
url=
"https://smba.trafficmanager.net/amer/v3/conversations/g1nny_w345l3y/activities",
status=200,
json={},
)
self.client.post(linking_url)
Identity.objects.get(external_id="g_w", user=self.user1)
assert not Identity.objects.filter(external_id="h_p",
user=self.user1).exists()
assert not Identity.objects.filter(external_id="g_w",
user=self.user2).exists()
def test_basic_flow(self, unsign):
unsign.return_value = {
"integration_id": self.integration.id,
"organization_id": self.org.id,
"teams_user_id": "a_p_w_b_d",
"team_id": "1_50l3mnly_5w34r",
"tenant_id": "h0g5m34d3",
}
linking_url = build_linking_url(
self.integration,
self.org,
"a_p_w_b_d",
"1_50l3mnly_5w34r",
"h0g5m34d3",
)
resp = self.client.get(linking_url)
assert resp.status_code == 200
self.assertTemplateUsed(resp, "sentry/auth-link-identity.html")
def user_conversation_id_callback(request):
payload = json.loads(request.body)
if payload["members"] == [{
"id": "a_p_w_b_d"
}] and payload["channelData"] == {
"tenant": {
"id": "h0g5m34d3"
}
}:
return (200, {}, json.dumps({"id": "dumbl3d0r3"}))
responses.add_callback(
method=responses.POST,
url="https://smba.trafficmanager.net/amer/v3/conversations",
callback=user_conversation_id_callback,
)
responses.add(
method=responses.POST,
url=
"https://smba.trafficmanager.net/amer/v3/conversations/dumbl3d0r3/activities",
status=200,
json={},
)
resp = self.client.post(linking_url)
identity = Identity.objects.filter(external_id="a_p_w_b_d",
user=self.user1)
assert len(identity) == 1
assert identity[0].idp == self.idp
assert identity[0].status == IdentityStatus.VALID
assert len(responses.calls) == 2
