Skip to content

Commit d489376

Browse files
committed
test if neither value provided
1 parent 6a5340b commit d489376

File tree

1 file changed

+20
-6
lines changed

1 file changed

+20
-6
lines changed

tests/test_blob.py

Lines changed: 20 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def test_blob_input_with_metadata_no_trigger_metadata(self):
9797
# Verify result metadata
9898
self.assertIsInstance(result, InputStream)
9999
self.assertEqual(result.name, 'blob_trigger_name')
100-
self.assertEqual(result.length, len(b'blob_content'))
100+
self.assertEqual(result.length, 12)
101101
self.assertEqual(result.uri, 'https://test.io/blob_trigger')
102102
self.assertEqual(result.blob_properties,
103103
json.loads(sample_blob_properties))
@@ -130,7 +130,7 @@ def test_blob_input_with_metadata_with_trigger_metadata(self):
130130
# Verify result metadata
131131
self.assertIsInstance(result, InputStream)
132132
self.assertEqual(result.name, 'blob_trigger_name')
133-
self.assertEqual(result.length, len(b'blob_content'))
133+
self.assertEqual(result.length, 12)
134134
self.assertEqual(result.uri, 'https://test.io/blob_trigger')
135135
self.assertEqual(result.blob_properties,
136136
json.loads(sample_blob_properties))
@@ -153,7 +153,7 @@ def test_blob_input_with_metadata_with_incorrect_trigger_metadata(self):
153153
# Verify result metadata
154154
self.assertIsInstance(result, InputStream)
155155
self.assertEqual(result.name, 'blob_trigger_name')
156-
self.assertEqual(result.length, len(b'blob_content'))
156+
self.assertEqual(result.length, 12)
157157
self.assertEqual(result.uri, 'https://test.io/blob_trigger')
158158
self.assertEqual(result.blob_properties,
159159
json.loads(sample_blob_properties))
@@ -240,7 +240,7 @@ def test_blob_input_with_metadata_with_length(self):
240240

241241
# Verify result metadata
242242
self.assertIsInstance(result, InputStream)
243-
self.assertEqual(result.length, len(b'blob_content'))
243+
self.assertEqual(result.length, 12)
244244

245245
def test_blob_input_with_metadata_with_both_length(self):
246246
sample_blob_properties = '''{
@@ -254,6 +254,20 @@ def test_blob_input_with_metadata_with_both_length(self):
254254
result: InputStream = afb. \
255255
BlobConverter.decode(data=datum, trigger_metadata=trigger_metadata)
256256

257-
# Verify result metadata
257+
# Verify result metadata.
258+
# This should be 12, since we check for ContentLength first
258259
self.assertIsInstance(result, InputStream)
259-
self.assertEqual(result.length, len(b'blob_content'))
260+
self.assertEqual(result.length, 12)
261+
262+
def test_blob_input_with_metadata_with_no_length(self):
263+
sample_blob_properties = '''{}'''
264+
datum: Datum = Datum(value=b'blob_content', type='bytes')
265+
trigger_metadata: Dict[str, Any] = {
266+
'Properties': Datum(sample_blob_properties, 'json')
267+
}
268+
result: InputStream = afb. \
269+
BlobConverter.decode(data=datum, trigger_metadata=trigger_metadata)
270+
271+
# Verify result metadata.
272+
self.assertIsInstance(result, InputStream)
273+
self.assertEqual(result.length, None)

0 commit comments

Comments
 (0)