def setUp(self):
super(TestUploadCallbackConnection, self).setUp()
self.bytes_container = BytesTransferredContainer()
self.class_factory = UploadCallbackConnectionClassFactory(
self.bytes_container,
buffer_size=50,
total_size=100,
progress_callback='Sample')
self.instance = self.class_factory.GetConnectionClass()('host')
class TestUploadCallbackConnection(testcase.GsUtilUnitTestCase):
"""Tests for the upload callback connection."""
def setUp(self):
super(TestUploadCallbackConnection, self).setUp()
self.bytes_container = BytesTransferredContainer()
self.class_factory = UploadCallbackConnectionClassFactory(
self.bytes_container,
buffer_size=50,
total_size=100,
progress_callback='Sample')
self.instance = self.class_factory.GetConnectionClass()('host')
@mock.patch(https_connection)
def testHeaderDefaultBehavior(self, mock_conn):
"""Test the size modifier is correct under expected headers."""
mock_conn.putheader.return_value = None
self.instance.putheader('content-encoding', 'gzip')
self.instance.putheader('content-length', '10')
self.instance.putheader('content-range', 'bytes 0-104/*')
# Ensure the modifier is as expected.
self.assertAlmostEqual(self.instance.size_modifier, 10.5)
@mock.patch(https_connection)
def testHeaderIgnoreWithoutGzip(self, mock_conn):
"""Test that the gzip content-encoding is required to modify size."""
mock_conn.putheader.return_value = None
self.instance.putheader('content-length', '10')
self.instance.putheader('content-range', 'bytes 0-99/*')
# Ensure the modifier is unchanged.
self.assertAlmostEqual(self.instance.size_modifier, 1.0)
@mock.patch(https_connection)
def testHeaderRangeFormatX_YSlashStar(self, mock_conn):
"""Test content-range header format X-Y/* """
self.instance.putheader('content-encoding', 'gzip')
self.instance.putheader('content-length', '10')
self.instance.putheader('content-range', 'bytes 0-99/*')
# Ensure the modifier is as expected.
self.assertAlmostEqual(self.instance.size_modifier, 10.0)
@mock.patch(https_connection)
def testHeaderRangeFormatStarSlash100(self, mock_conn):
"""Test content-range header format */100 """
self.instance.putheader('content-encoding', 'gzip')
self.instance.putheader('content-length', '10')
self.instance.putheader('content-range', 'bytes */100')
# Ensure the modifier is as expected.
self.assertAlmostEqual(self.instance.size_modifier, 1.0)
@mock.patch(https_connection)
def testHeaderRangeFormat0_99Slash100(self, mock_conn):
"""Test content-range header format 0-99/100 """
self.instance.putheader('content-encoding', 'gzip')
self.instance.putheader('content-length', '10')
self.instance.putheader('content-range', 'bytes 0-99/100')
# Ensure the modifier is as expected.
self.assertAlmostEqual(self.instance.size_modifier, 10.0)
@mock.patch(https_connection)
def testHeaderParseFailure(self, mock_conn):
"""Test incorrect header values do not raise exceptions."""
mock_conn.putheader.return_value = None
self.instance.putheader('content-encoding', 'gzip')
self.instance.putheader('content-length', 'bytes 10')
self.instance.putheader('content-range', 'not a number')
# Ensure the modifier is unchanged.
self.assertAlmostEqual(self.instance.size_modifier, 1.0)
@mock.patch('gslib.progress_callback.ProgressCallbackWithTimeout')
@mock.patch('httplib2.HTTPSConnectionWithTimeout')
def testSendDefaultBehavior(self, mock_conn, mock_callback):
mock_conn.send.return_value = None
self.instance.size_modifier = 2
self.instance.processed_initial_bytes = True
self.instance.callback_processor = mock_callback
# Send 10 bytes of data.
sample_data = b'0123456789'
self.instance.send(sample_data)
# Ensure the data is fully sent since the buffer size is 50 bytes.
self.assertTrue(mock_conn.send.called)
(_, sent_data), _ = mock_conn.send.call_args_list[0]
self.assertEqual(sent_data, sample_data)
# Ensure the progress callback is correctly scaled.
self.assertTrue(mock_callback.Progress.called)
[sent_bytes], _ = mock_callback.Progress.call_args_list[0]
self.assertEqual(sent_bytes, 20)