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