def test_get_numbers(self): self.mock_cursor.fetchall.return_value = [(NUMBER_ID.hex, SIP_URI, False, GAB_LISTED)] numbers = get_numbers(self.mock_session, OWNER_ID) self.assertEqual(numbers, [{ "number": SIP_URI, "number_id": NUMBER_ID, "pstn": False, "gab_listed": GAB_LISTED }])
def test_get_numbers(self): self.mock_cursor.fetchall.return_value = [(NUMBER_ID.hex, SIP_URI, False, GAB_LISTED)] numbers = get_numbers(self.mock_session, OWNER_ID) self.assertEqual(numbers, [ { "number": SIP_URI, "number_id": NUMBER_ID, "pstn": False, "gab_listed": GAB_LISTED } ])
def get(self, username): """Retrieve list of phone numbers.""" user_id = self.get_and_check_user_id(username) self._numbers = numbers.get_numbers(self.db_session(), user_id) if len(self._numbers) == 0: self.finish({"numbers": []}) return for number in self._numbers: number["number_id"] = number["number_id"].hex number["sip_uri"] = number["number"] number["sip_username"] = utils.sip_uri_to_phone_number(number["number"]) number["domain"] = utils.sip_uri_to_domain(number["number"]) number["number"] = utils.sip_uri_to_phone_number(number["number"]) number["formatted_number"] = format_phone_number(number["number"]) _request_group = HTTPCallbackGroup(partial(self._on_get_success, number["sip_uri"]), self._on_get_failure) # We only store the public identities in Ellis, and must query # Homestead for the associated private identities homestead.get_associated_privates(number["sip_uri"], _request_group.callback())
def get(self, username): """Retrieve list of phone numbers.""" user_id = self.get_and_check_user_id(username) self._numbers = numbers.get_numbers(self.db_session(), user_id) if len(self._numbers) == 0: self.finish({"numbers": []}) return for number in self._numbers: number["number_id"] = number["number_id"].hex number["sip_uri"] = number["number"] number["sip_username"] = utils.sip_uri_to_phone_number( number["number"]) number["domain"] = utils.sip_uri_to_domain(number["number"]) number["number"] = utils.sip_uri_to_phone_number(number["number"]) number["formatted_number"] = format_phone_number(number["number"]) _request_group = HTTPCallbackGroup( partial(self._on_get_success, number["sip_uri"]), partial(self._on_get_failure, number["sip_uri"])) # We only store the public identities in Ellis, and must query # Homestead for the associated private identities homestead.get_associated_privates(number["sip_uri"], _request_group.callback())
def delete(self, email): _log.info("Request to delete account") self._user_id = self.get_and_check_user_id(email) self._numbers = numbers.get_numbers(self.db_session(), self._user_id) self._do_delete_work()