Python Mock.content方法代码示例

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
def mock_response(
        status=200,
        content="CONTENT",
        json_data=None,
        raise_for_status=None,
        url=None):
    """
    since we typically test a bunch of different
    requests calls for a service, we are going to do
    a lot of mock responses, so its usually a good idea
    to have a helper function that builds these things
    """
    mock_resp = Mock()
    # mock raise_for_status call w/optional error
    mock_resp.raise_for_status = Mock()
    if raise_for_status:
        mock_resp.raise_for_status.side_effect = raise_for_status
    # set status code and content
    mock_resp.status_code = status
    mock_resp.content = content
    mock_resp.url = url
    # add json data if provided
    if json_data:
        mock_resp.json = Mock(
            return_value=json_data
        )
    return mock_resp

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
 def setUp(self):
     response = Mock()
     response.content = json.dumps({'status': 'ok'})
     response.headers = {'Content-Type': 'application/json'}
     response.ok = True
     response.status_code = requests.codes.ok
     self.response = response

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
async def test_async_iterator():
    response = Mock()

    class TestAsyncIterator(AsyncIterator):
        def __init__(self):
            self.i = 0

        async def __anext__(self):
            self.i += 1

            if self.i > 5:
                raise StopAsyncIteration
            else:
                return '{}\r\n'.format(self.i).encode('utf-8')

    response.content = TestAsyncIterator()
    ait = _AsyncLineIterator(response, True)
    ait.get_encoding = lambda: 'utf-8'

    lines = list()
    async for line in ait:
        lines.append(line)

    assert lines == ['1', '2', '3', '4', '5']

    response = Mock()
    response.content.iter_chunked.return_value = TestAsyncIterator()
    ait = _AsyncChunkIterator(response, True)
    ait.get_encoding = lambda: 'utf-8'

    chunks = list()
    async for chunk in ait:
        chunks.append(chunk)

    assert chunks == ['1\n', '2\n', '3\n', '4\n', '5\n']

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
    def test_fetch_and_validate_source_when_source_does_not_contain_target(self, mock_get):
        mock_response = Mock()
        mock_response.status_code = 200
        mock_response.content = 'foo'
        mock_get.return_value = mock_response

        self.assertRaises(TargetNotFoundError, fetch_and_validate_source, self.source, self.target)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
    def test_fetch_and_validate_source_happy_path(self, mock_get):
        mock_response = Mock()
        mock_response.status_code = 200
        mock_response.content = '<a href="{href}">{href}</a>'.format(href=self.target)
        mock_get.return_value = mock_response

        self.assertEqual(mock_response.content, fetch_and_validate_source(self.source, self.target))

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
 def test_sucess(self, requests_mock):
     js = json.dumps(dct)
     response = Mock()
     response.content = js.encode('utf8')
     get_mock = Mock(return_value=response)
     requests_mock.get = get_mock
     resultado = usuario.acessar_dados('renzon')
     get_mock.assert_called_once_with('https://api.github.com/users/renzon')
     self.assertDictEqual(dct, resultado)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
    def test_complete_request(self):
        transport = t.APITransport(Mock())
        transport._http = Mock()

        http_result = Mock()
        http_result.content = b'{"stat":"ok","result":"bar"}'
        transport._http.post.return_value = http_result

        self.assertEqual(
            "bar", transport(t.APITransport.NO_ENCRYPT[0], foo="bar"))

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
 def assert_mock_search_results(self, search_parameter, page_num):
     # url to stub out
     url = SearchSong.URL_FORMAT % (search_parameter, page_num)
     # mock out hymnal.net response
     mock_response = Mock()
     mock_response.content = open('test_data/search_song_html_{}_{}.txt'.format(search_parameter, page_num), 'r')
     
     patcher = patch('requests.get', Mock(side_effect = lambda k: {url: mock_response}.get(k, 'unhandled request %s' % k)))
     patcher.start()
     self.assert_search_results(search_parameter, page_num)
     patcher.stop()

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
def listing_and_sumatra_record(request):
    listing = request.param
    output_data = Mock()
    output_data.content = listing.path_to_file.read_text().encode('utf-8')
    datastore = Mock(spec_set=DataStore)
    datastore.get_data_item.return_value = output_data
    record = Mock(spec=Record)
    record.datastore = datastore
    record.output_data = ['something']
    record.to_sumatra = Mock()
    record.to_sumatra.return_value = record
    return listing, record

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
    def test_make_http_request(self):
        # url, data, params
        http = Mock()
        retval = Mock()
        retval.content = "foo"
        http.post.return_value = retval

        self.transport._http = http
        res = self.transport._make_http_request(
            "/url", b"data", {"a": None, "b": "c"})

        http.post.assert_called_with("/url", data=b"data", params={"b": "c"})
        retval.raise_for_status.assert_called_with()

        self.assertEqual("foo", res)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
def requests_factory(text='some text',
                     location='some location',
                     content=b'some content'):
    requests = Mock()

    get_response = Mock()
    get_response.text = text
    get_response.content = content

    head_response = Mock()
    head_response.headers = {'Location': location}

    requests.get = Mock(return_value=get_response)
    requests.head = Mock(return_value=head_response)

    return requests

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
    def test_deliver_slice_csv_inline(self, send_email_smtp, get):
        response = Mock()
        get.return_value = response
        response.raise_for_status.return_value = None
        response.content = self.CSV

        schedule = db.session.query(SliceEmailSchedule).filter_by(
            id=self.slice_schedule).all()[0]

        schedule.email_format = SliceEmailReportFormat.data
        schedule.delivery_type = EmailDeliveryType.inline

        deliver_slice(schedule)
        send_email_smtp.assert_called_once()

        self.assertIsNone(send_email_smtp.call_args[1]['data'])
        self.assertTrue('<table ' in send_email_smtp.call_args[0][2])

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
    def test_get_or_create_from_username_create(self, mock_get):
        response = Mock()
        response.status_code = 200
        photo_bytes_array = bytearray(self.photo_bytes)
        expected_image = "data:image/jpg;base64,%s" % base64.b64encode(photo_bytes_array).decode('utf-8')
        response.content = photo_bytes_array
        mock_get.return_value = response

        with mock.patch('pmp_shield.employees.ldap_tools.LDAPTool.search_by_username',
                        TestACPEmployee.mock_ldap.search_by_username):
            employee, created = Employee.objects.get_or_create_from_username('lberrocal')
            self.assertEqual('Luis', employee.first_name)
            self.assertEqual('Berrocal Cordoba', employee.last_name)
            self.assertEqual('272-4149', employee.phones.first().phone_number)
            self.assertEqual(1, Employee.objects.count())
            self.assertEqual(expected_image, employee.photo_url)
            self.assertTrue(created)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
    def test_deliver_slice_csv_attachment(self, send_email_smtp, get):
        response = Mock()
        get.return_value = response
        response.raise_for_status.return_value = None
        response.content = self.CSV

        schedule = db.session.query(SliceEmailSchedule).filter_by(
            id=self.slice_schedule).all()[0]

        schedule.email_format = SliceEmailReportFormat.data
        schedule.delivery_type = EmailDeliveryType.attachment

        deliver_slice(schedule)
        send_email_smtp.assert_called_once()

        file_name = __('%(name)s.csv', name=schedule.slice.slice_name)

        self.assertEquals(
            send_email_smtp.call_args[1]['data'][file_name],
            self.CSV,
        )

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import content [as 别名]
 def message(self, last_vote):
     message = Mock()
     message.content = last_vote
     return message