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())