1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import os
import logging
import sys
import pytest
from devtools_testutils.fake_credentials import FakeTokenCredential
from devtools_testutils import get_credential, is_live, recorded_by_proxy, set_bodiless_matcher
from _shared.utils import get_http_logging_policy
from azure.core.exceptions import HttpResponseError
from acs_sms_test_case import ACSSMSTestCase
from azure.communication.sms import SmsClient
class TestClient(ACSSMSTestCase):
def setup_method(self):
super().setUp()
set_bodiless_matcher()
@recorded_by_proxy
def test_send_sms_single(self):
sms_client = self.create_client_from_connection_string()
# calling send() with sms values
sms_responses = sms_client.send(from_=self.phone_number, to=self.phone_number, message="Hello World via SMS")
assert len(sms_responses) == 1
self.verify_successful_sms_response(sms_responses[0])
@recorded_by_proxy
def test_send_sms_multiple_with_options(self):
sms_client = self.create_client_from_connection_string()
# calling send() with sms values
sms_responses = sms_client.send(
from_=self.phone_number,
to=[self.phone_number, self.phone_number],
message="Hello World via SMS",
enable_delivery_report=True, # optional property
tag="custom-tag",
) # optional property
assert len(sms_responses) == 2
self.verify_successful_sms_response(sms_responses[0])
self.verify_successful_sms_response(sms_responses[1])
@recorded_by_proxy
def test_send_sms_from_managed_identity(self):
if not is_live():
credential = FakeTokenCredential()
else:
credential = get_credential()
sms_client = SmsClient(self.endpoint, credential, http_logging_policy=get_http_logging_policy())
# calling send() with sms values
sms_responses = sms_client.send(from_=self.phone_number, to=[self.phone_number], message="Hello World via SMS")
assert len(sms_responses) == 1
self.verify_successful_sms_response(sms_responses[0])
@recorded_by_proxy
def test_send_sms_fake_from_phone_number(self):
sms_client = self.create_client_from_connection_string()
with pytest.raises(HttpResponseError) as ex:
# calling send() with sms values
sms_client.send(from_="+15550000000", to=[self.phone_number], message="Hello World via SMS")
assert str(ex.value.status_code) == "401"
assert ex.value.message is not None
@recorded_by_proxy
def test_send_sms_fake_to_phone_number(self):
sms_client = self.create_client_from_connection_string()
with pytest.raises(HttpResponseError) as ex:
sms_responses = sms_client.send(
from_=self.phone_number, to=["Ad155500000000000"], message="Hello World via SMS"
)
assert str(ex.value.status_code == "400")
@recorded_by_proxy
def test_send_sms_unauthorized_from_phone_number(self):
sms_client = self.create_client_from_connection_string()
with pytest.raises(HttpResponseError) as ex:
# calling send() with sms values
sms_client.send(from_="+14255550123", to=[self.phone_number], message="Hello World via SMS")
assert str(ex.value.status_code) == "401"
assert ex.value.message is not None
@pytest.mark.live_test_only
@recorded_by_proxy
def test_send_sms_unique_message_ids(self):
sms_client = self.create_client_from_connection_string()
# calling send() with sms values
sms_responses_1 = sms_client.send(
from_=self.phone_number, to=[self.phone_number], message="Hello World via SMS"
)
# calling send() again with the same sms values
sms_responses_2 = sms_client.send(
from_=self.phone_number, to=[self.phone_number], message="Hello World via SMS"
)
self.verify_successful_sms_response(sms_responses_1[0])
self.verify_successful_sms_response(sms_responses_2[0])
# message ids should be unique due to having a different idempotency key
assert sms_responses_1[0].message_id != sms_responses_2[0].message_id
def verify_successful_sms_response(self, sms_response):
if self.is_live:
assert sms_response.to == self.phone_number
assert sms_response.message_id is not None
assert sms_response.http_status_code == 202
assert sms_response.error_message is None
assert sms_response.successful
def create_client_from_connection_string(self):
return SmsClient.from_connection_string(self.connection_str, http_logging_policy=get_http_logging_policy())
|
