def test__prepare_request_non_bytes_data(self): upload = _upload.SimpleUpload(SIMPLE_URL) assert not upload.finished with pytest.raises(TypeError) as exc_info: upload._prepare_request(u"", None) exc_info.match(u"must be bytes")
def test__prepare_request_already_finished(self): upload = _upload.SimpleUpload(SIMPLE_URL) upload._finished = True with pytest.raises(ValueError) as exc_info: upload._prepare_request(b"", None) exc_info.match(u"An upload can only be used once.")
def test__prepare_request(self): upload = _upload.SimpleUpload(SIMPLE_URL) content_type = u"image/jpeg" data = b"cheetos and eetos" method, url, payload, headers = upload._prepare_request(data, content_type) assert method == u"POST" assert url == SIMPLE_URL assert payload == data assert headers == {u"content-type": content_type}
def test__prepare_request(self): upload = _upload.SimpleUpload(SIMPLE_URL) content_type = u'image/jpeg' data = b'cheetos and eetos' method, url, payload, headers = upload._prepare_request( data, content_type) assert method == u'POST' assert url == SIMPLE_URL assert payload == data assert headers == {u'content-type': content_type}
def test__prepare_request_with_headers(self): headers = {u"x-goog-cheetos": u"spicy"} upload = _upload.SimpleUpload(SIMPLE_URL, headers=headers) content_type = u"image/jpeg" data = b"some stuff" method, url, payload, new_headers = upload._prepare_request(data, content_type) assert method == u"POST" assert url == SIMPLE_URL assert payload == data assert new_headers is headers expected = {u"content-type": content_type, u"x-goog-cheetos": u"spicy"} assert headers == expected
def test__prepare_request_with_headers(self): headers = {u'x-goog-cheetos': u'spicy'} upload = _upload.SimpleUpload(SIMPLE_URL, headers=headers) content_type = u'image/jpeg' data = b'some stuff' method, url, payload, new_headers = upload._prepare_request( data, content_type) assert method == u'POST' assert url == SIMPLE_URL assert payload == data assert new_headers is headers expected = {u'content-type': content_type, u'x-goog-cheetos': u'spicy'} assert headers == expected
def test_transmit(self): upload = _upload.SimpleUpload(SIMPLE_URL) with pytest.raises(NotImplementedError) as exc_info: upload.transmit(None, None, None) exc_info.match(u"virtual")