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))