def test_utf1(self):
response = main.batcher(
[u'\u0041\u0041\u0041\u0041\u0041', u'\u0041\u0041'],
TEST_SMALL_MAX_RECORD_SIZE, TEST_SMALL_MAX_BATCH_SIZE,
TEST_SMALL_MAX_RECORDS_IN_BATCH)
#[['AAAAA', 'AA']]
self.assertEqual(len(response), 1, "Should be 1")
def test_utf3(self):
response = main.batcher([u'\u03c0\u03c0\u03c0\u03c0', u'\u03c0\u03c0'],
TEST_SMALL_MAX_RECORD_SIZE,
TEST_SMALL_MAX_BATCH_SIZE,
TEST_SMALL_MAX_RECORDS_IN_BATCH)
#[['ππππ', 'ππ']]
self.assertEqual(len(response[0]), 2, "Should be 2")
def test_max_size_3(self):
array = []
array.append(generate_fixed_size_record(False, 1024 * 1024))
array.append(generate_fixed_size_record(False, 1024 * 1024 + 1))
response = main.batcher(array)
self.assertEqual(len(response), 1, "Should be 1")
self.assertEqual(len(response[0]), 1, "Should be 1")
def test_utf2(self):
response = batcher(
[u'\u03c0\u03c0\u03c0\u03c0\u03c0', u'\u03c0\u03c0'],
TEST_SMALL_MAX_RECORD_SIZE, TEST_SMALL_MAX_BATCH_SIZE,
TEST_SMALL_MAX_RECORDS_IN_BATCH)
#[['ππ']]
self.assertEqual(len(response), 1, "Should be 1")
def test_record_order(self):
array = []
array.append(generate_fixed_size_record(False, 5))
array.append(generate_fixed_size_record(False, 10))
array.append(generate_fixed_size_record(False, 1024 * 1024))
array.append(generate_fixed_size_record(False, 1024 * 500))
array.append(generate_fixed_size_record(False, 1024 * 30))
array.append(generate_fixed_size_record(False, 1024))
array.append(generate_fixed_size_record(False, 321))
array.append(generate_fixed_size_record(False, 120546))
array.append(generate_fixed_size_record(False, 45))
array.append(generate_fixed_size_record(False, 555555))
response = main.batcher(array)
response_generator = (record for batch in response for record in batch)
response_string = "".join(response_generator)
array_string = ''.join([record for record in array])
self.assertEqual(response_string, array_string,
"Should be equal but are not")
def test_splitting3(self):
response = main.batcher(self.records1, TEST_SMALL_MAX_RECORD_SIZE,
TEST_SMALL_MAX_BATCH_SIZE,
TEST_SMALL_MAX_RECORDS_IN_BATCH)
#[["1234", "123456", "12", "123", "123456", "12"],["1234", "123456"]]
self.assertEqual(len(response[1]), 2, "Should be 2")
def test_max_size_2(self):
array = []
array.append(generate_fixed_size_record(False, 1024 * 1024 + 1))
response = batcher(array)
self.assertEqual(len(response), 0, "Should be 0")
