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)