Skip to content

Commit

Permalink
Merge pull request #458 from tutorcruncher/change-originator-to-our-n…
Browse files Browse the repository at this point in the history
…umber
  • Loading branch information
tomhamiltonstubber committed Aug 29, 2023
2 parents 2706a28 + 194a508 commit 234ccdc
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 3 deletions.
1 change: 1 addition & 0 deletions src/settings.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@ class Settings(BaseSettings):
# Have to use a US number as the originator to send to the US
# https://support.messagebird.com/hc/en-us/articles/208747865-United-States
us_send_number = '15744445663'
gb_send_number = '12048170659'

@validator('pg_dsn')
def heroku_ready_pg_dsn(cls, v):
Expand Down
2 changes: 1 addition & 1 deletion src/worker/sms.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ def __init__(self, ctx: dict, group_id: int, company_id: int, recipient: SmsReci
self.m: SmsSendModel = m
self.tags = list(set(self.recipient.tags + self.m.tags + [str(self.m.uid)]))
self.messagebird: MessageBird = ctx['messagebird']
self.from_name = self.m.from_name if self.m.country_code != 'US' else self.settings.us_send_number
self.from_name = self.settings.gb_send_number if self.m.country_code != 'US' else self.settings.us_send_number

async def run(self):
sms_data = await self._sms_prep()
Expand Down
6 changes: 4 additions & 2 deletions tests/test_sms.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
from urllib.parse import urlencode
from uuid import uuid4

from src.main import settings


def test_send_message(cli, tmpdir, worker, loop):
data = {
Expand All @@ -26,7 +28,7 @@ def test_send_message(cli, tmpdir, worker, loop):
"to: Number(number='+447891123856', country_code='44', "
"number_formatted='+44 7891 123856', descr=None, is_mobile=True)"
) in msg_file
assert '\nfrom_name: foobar send\n' in msg_file
assert f'\nfrom_name: {settings.gb_send_number}\n' in msg_file
assert '\nmessage:\nthis is a message bar\n' in msg_file
assert '\nlength: SmsLength(length=21, parts=1)\n' in msg_file

Expand Down Expand Up @@ -323,7 +325,7 @@ def test_link_shortening(cli, tmpdir, sync_db: SyncDb, worker, loop):
f = '69eb85e8-1504-40aa-94ff-75bb65fd8d75-447891123856.txt'
assert str(tmpdir.listdir()[0]).endswith(f)
msg_file = tmpdir.join(f).read()
assert '\nfrom_name: Morpheus\n' in msg_file
assert f'\nfrom_name: {settings.gb_send_number}\n' in msg_file
assert '\nmessage:\nthis is a message click.example.com/l' in msg_file
token = re.search('message click.example.com/l(.+?)\n', msg_file).groups()[0]
assert len(token) == 12
Expand Down

0 comments on commit 234ccdc

Please sign in to comment.