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