From 3020a8254cab11bdfc5bd5f5945d425d01a4bb3c Mon Sep 17 00:00:00 2001 From: Sami Z Date: Thu, 6 Apr 2017 16:34:38 -0700 Subject: [PATCH] test case for BYTES merger related to fix in streamed.py to allow BYTES merger --- spanner/tests/unit/test_streamed.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/spanner/tests/unit/test_streamed.py b/spanner/tests/unit/test_streamed.py index 674d34e2a03c..740a3e0f0ea0 100644 --- a/spanner/tests/unit/test_streamed.py +++ b/spanner/tests/unit/test_streamed.py @@ -185,6 +185,21 @@ def test__merge_chunk_string(self): self.assertEqual(merged.string_value, u'phredwylma') self.assertIsNone(streamed._pending_chunk) + def test__merge_chunk_string_w_bytes(self): + iterator = _MockCancellableIterator() + streamed = self._make_one(iterator) + FIELDS = [ + self._makeScalarField('image', 'BYTES'), + ] + streamed._metadata = _ResultSetMetadataPB(FIELDS) + streamed._pending_chunk = self._makeValue(u'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAAAAAA6fptVAAAACXBIWXMAAAsTAAALEwEAmpwYAAAA\n') + chunk = self._makeValue(u'B3RJTUUH4QQGFwsBTL3HMwAAABJpVFh0Q29tbWVudAAAAAAAU0FNUExFMG3E+AAAAApJREFUCNdj\nYAAAAAIAAeIhvDMAAAAASUVORK5CYII=\n') + + merged = streamed._merge_chunk(chunk) + + self.assertEqual(merged.string_value, u'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAAAAAA6fptVAAAACXBIWXMAAAsTAAALEwEAmpwYAAAA\nB3RJTUUH4QQGFwsBTL3HMwAAABJpVFh0Q29tbWVudAAAAAAAU0FNUExFMG3E+AAAAApJREFUCNdj\nYAAAAAIAAeIhvDMAAAAASUVORK5CYII=\n') + self.assertIsNone(streamed._pending_chunk) + def test__merge_chunk_array_of_bool(self): iterator = _MockCancellableIterator() streamed = self._make_one(iterator)