def test_min_max_size(self):
message = reaction_pb2.Data(value='test_value')
with self.assertRaisesRegex(ValueError, 'must be less than or equal'):
data_storage.write_data(message,
self.test_subdirectory,
min_size=2.0,
max_size=1.0)
def test_bytes_value(self):
message = reaction_pb2.Data(bytes_value=b'test value')
filename = data_storage.write_data(message, self.test_subdirectory)
expected = os.path.join(
self.test_subdirectory, 'ord_data-'
'47d1d8273710fd6f6a5995fac1a0983fe0e8828c288e35e80450ddc5c4412def'
'.txt')
self.assertEqual(filename, expected)
with open(filename, 'rb') as f:
self.assertEqual(message.bytes_value, f.read())
def test_string_value(self):
message = reaction_pb2.Data(value='test value')
filename = data_storage.write_data(message, self.test_subdirectory)
expected = os.path.join(
self.test_subdirectory, 'ord_data-'
'47d1d8273710fd6f6a5995fac1a0983fe0e8828c288e35e80450ddc5c4412def'
'.txt')
self.assertEqual(filename, expected)
# NOTE(kearnes): Open with 'r' to get the decoded string.
with open(filename, 'r') as f:
self.assertEqual(message.value, f.read())
def test_max_size(self):
message = reaction_pb2.Data(value='test value')
with self.assertRaisesRegex(ValueError, 'larger than max_size'):
data_storage.write_data(message,
self.test_subdirectory,
max_size=1e-6)
def test_min_size(self):
message = reaction_pb2.Data(value='test_value')
self.assertIsNone(
data_storage.write_data(message,
self.test_subdirectory,
min_size=1.0))
def test_missing_value(self):
message = reaction_pb2.Data()
with self.assertRaisesRegex(ValueError, 'no value to write'):
data_storage.write_data(message, self.test_subdirectory)
def test_url_value(self):
message = reaction_pb2.Data(url='test value')
self.assertIsNone(
data_storage.write_data(message, self.test_subdirectory))
def test_min_size(self):
message = reaction_pb2.Data(string_value='test_value')
filename, _ = data_storage.write_data(message,
self.test_subdirectory,
min_size=1.0)
self.assertIsNone(filename)
