def check_number_ownership(self, sip_uri, user_id): try: owner_id = numbers.get_sip_uri_owner_id(self.db_session(), sip_uri) except NotFound: raise HTTPError(404, "Number not found") else: if owner_id != user_id: raise HTTPError(404, "User doesn't own that number")
def test_get_sip_uri_owner_id(self): self.mock_cursor.fetchone.return_value = (OWNER_ID,) id = get_sip_uri_owner_id(self.mock_session, SIP_URI) self.mock_session.execute.assert_called_once_with(ANY, {"number": SIP_URI}) self.assertEqual(id, OWNER_ID)
def test_get_sip_uri_owner_id(self): self.mock_cursor.fetchone.return_value = (OWNER_ID, ) id = get_sip_uri_owner_id(self.mock_session, SIP_URI) self.mock_session.execute.assert_called_once_with( ANY, {"number": SIP_URI}) self.assertEqual(id, OWNER_ID)