def test006_write_sample_on_disk_raises_IrmaFileSystemError(self):
sha, data = "sha_test", "data_test"
with patch("%s.build_sha256_path" % module.__name__,
create=True) as mock_build_path:
with patch("%s.open" % module.__name__, create=True) as mock_open:
mock_open().__enter__().write.side_effect = IOError()
with self.assertRaises(IrmaFileSystemError) as context:
result = module.write_sample_on_disk(sha, data)
expected = "Cannot add the sample {0} to the collection".format(sha)
self.assertEqual(str(context.exception),
expected)
def test005_write_sample_on_disk_ok(self):
sha, data = "sha_test", "data_test"
with patch("%s.build_sha256_path" % module.__name__,
create=True) as mock_build_path:
with patch("%s.open" % module.__name__, create=True) as mock_open:
result = module.write_sample_on_disk(sha, data)
self.assertEqual(mock_build_path.call_count, 1)
self.assertEqual(mock_build_path.call_args, ((sha,),))
self.assertEqual(mock_open.call_count, 1)
self.assertEqual(mock_open.call_args, ((mock_build_path(sha), "wb"),))
self.assertEqual(mock_open().__enter__().write.call_count, 1)
self.assertEqual(mock_open().__enter__().write.call_args, ((data,),))
self.assertEqual(result, mock_build_path(sha))
