def test_add_number_to_pool(self, uuid4):
uuid4.return_value = NUMBER_ID
add_number_to_pool(self.mock_session, SIP_URI)
self.mock_session.execute.assert_called_once_with(ANY,
{
"number_id": NUMBER_ID,
"number": SIP_URI,
"pstn": False
})
def test_add_number_to_pool(self, uuid4):
uuid4.return_value = NUMBER_ID
add_number_to_pool(self.mock_session, SIP_URI)
self.mock_session.execute.assert_called_once_with(
ANY, {
"number_id": NUMBER_ID,
"number": SIP_URI,
"pstn": False,
"specified": False
})
def standalone(start, num, pstn, realm):
connection.init_connection()
s = connection.Session()
create_count = 0
if not start:
start = 5108580271 if pstn else 6505550000
for x in xrange(num):
if pstn:
public_id = "sip:+1%d@%s" % (start + x, realm)
else:
public_id = "sip:%d@%s" % (start + x, realm)
try:
numbers.add_number_to_pool(s, public_id, pstn, False)
except IntegrityError:
# Entry already exists, not creating in db
pass
else:
create_count += 1
s.commit()
print "Created %d numbers, %d already present in database" % (create_count, num - create_count)
def post(self, username, sip_uri):
"""Allocate a phone number."""
_log.debug("Specific number allocation API call (%s)", sip_uri)
self.is_admin_request()
user_id = self.get_and_check_user_id(username)
db_sess = self.db_session()
pstn = self.get_argument("pstn", "false").lower() == "true"
private_id = self.get_argument("private_id", None)
new_private_id = self.get_argument("new_private_id", "false").lower() == "true"
try:
number_id = uuid.UUID(numbers.get_sip_uri_number_id(db_sess, sip_uri))
except NotFound:
# This SIP URI is not currently in the pool, so add it
number_id = numbers.add_number_to_pool(db_sess, sip_uri, False, True)
numbers.allocate_specific_number(db_sess, user_id, number_id)
self.sip_uri = sip_uri
db_sess.commit()
# Work out the response we'll send if the upstream requests
# are successful.
number = utils.sip_uri_to_phone_number(sip_uri)
pretty_number = format_phone_number(number)
self.__response = {
"sip_uri": sip_uri,
"sip_username": number,
"number": number,
"pstn": pstn,
"formatted_number": pretty_number,
"number_id": number_id.hex,
}
# Generate a random password and store it in Homestead.
_log.debug("Populating other servers...")
self._request_group = HTTPCallbackGroup(self._on_post_success, self._on_post_failure)
public_callback = self._request_group.callback()
if private_id == None:
# No private id was provided, so we need to create a new
# digest in Homestead
private_id = utils.sip_public_id_to_private(sip_uri)
new_private_id = True
if new_private_id:
sip_password = utils.generate_sip_password()
_log.debug("About to create private ID at Homestead")
homestead.create_private_id(
private_id, utils.sip_uri_to_domain(sip_uri), sip_password, self._request_group.callback()
)
_log.debug("Created private ID at Homestead")
self.__response["sip_password"] = sip_password
self.__response["sip_username"] = private_id
# Associate the new public identity with the private identity in Homestead
# and store the iFCs in homestead.
homestead.create_public_id(
private_id, sip_uri, ifcs.generate_ifcs(utils.sip_uri_to_domain(sip_uri)), public_callback
)
self.__response["private_id"] = private_id
# Concurrently, store the default simservs in XDM.
with open(settings.XDM_DEFAULT_SIMSERVS_FILE) as xml_file:
default_xml = xml_file.read()
xdm.put_simservs(sip_uri, default_xml, self._request_group.callback())
def post(self, username, sip_uri): # pragma: no cover
"""Allocate a phone number."""
_log.debug("Specific number allocation API call (%s)", sip_uri)
self.is_admin_request()
user_id = self.get_and_check_user_id(username)
db_sess = self.db_session()
pstn = self.get_argument('pstn', 'false').lower() == 'true'
private_id = self.get_argument('private_id', None)
new_private_id = self.get_argument('new_private_id',
'false').lower() == 'true'
try:
number_id = uuid.UUID(
numbers.get_sip_uri_number_id(db_sess, sip_uri))
except NotFound:
# This SIP URI is not currently in the pool, so add it
number_id = numbers.add_number_to_pool(db_sess, sip_uri, False,
True)
numbers.allocate_specific_number(db_sess, user_id, number_id)
self.sip_uri = sip_uri
db_sess.commit()
# Work out the response we'll send if the upstream requests
# are successful.
number = utils.sip_uri_to_phone_number(sip_uri)
pretty_number = format_phone_number(number)
self.__response = {
"sip_uri": sip_uri,
"sip_username": number,
"number": number,
"pstn": pstn,
"formatted_number": pretty_number,
"number_id": number_id.hex
}
# Generate a random password and store it in Homestead.
_log.debug("Populating other servers...")
self._request_group = HTTPCallbackGroup(self._on_post_success,
self._on_post_failure)
public_callback = self._request_group.callback()
if private_id == None:
# No private id was provided, so we need to create a new
# digest in Homestead
private_id = utils.sip_public_id_to_private(sip_uri)
new_private_id = True
if new_private_id:
sip_password = utils.generate_sip_password()
_log.debug("About to create private ID at Homestead")
homestead.create_private_id(private_id,
utils.sip_uri_to_domain(sip_uri),
sip_password,
self._request_group.callback())
_log.debug("Created private ID at Homestead")
self.__response["sip_password"] = sip_password
self.__response["sip_username"] = private_id
# Associate the new public identity with the private identity in Homestead
# and store the iFCs in homestead.
homestead.create_public_id(
private_id, sip_uri,
ifcs.generate_ifcs(utils.sip_uri_to_domain(sip_uri)),
public_callback)
self.__response["private_id"] = private_id
# Concurrently, store the default simservs in XDM.
xdm.put_simservs(sip_uri, simservs.default_simservs(),
self._request_group.callback())