def test_delete_files_fails_with_io_error(self): path = os.path.join(self.tmpdir, 'f2') with self.assertRaises(BeamIOError) as error: utils.delete_files([path]) self.assertTrue( error.exception.args[0].startswith('Delete operation failed')) self.assertEqual(error.exception.exception_details.keys(), [path])
def test_delete_files_fails_with_io_error(self, mocked_delete): f = tempfile.NamedTemporaryFile(dir=self.tmpdir, delete=False) assert FileSystems.exists(f.name) with self.assertRaises(BeamIOError): utils.delete_files([f.name]) self.assertTrue(mocked_delete.called) self.assertEqual(mocked_delete.call_count, 4)
def test_delete_files_fails_with_io_error(self, mocked_delete): f = tempfile.NamedTemporaryFile(dir=self.tmpdir, delete=False) assert FileSystems.exists(f.name) with self.assertRaises(BeamIOError): utils.delete_files([f.name]) self.assertTrue(mocked_delete.called) self.assertEqual(mocked_delete.call_count, 4)
def test_delete_files_succeeds(self): path = os.path.join(self.tmpdir, 'f1') with open(path, 'a') as f: f.write('test') assert FileSystems.exists(path) utils.delete_files([path]) assert not FileSystems.exists(path)
def test_delete_files_fails_with_invalid_arg(self): with self.assertRaises(RuntimeError): utils.delete_files([])
def test_delete_files_succeeds(self): f = tempfile.NamedTemporaryFile(dir=self.tmpdir, delete=False) assert FileSystems.exists(f.name) utils.delete_files([f.name]) assert not FileSystems.exists(f.name)
def test_delete_files_succeeds(self): f = tempfile.NamedTemporaryFile(dir=self.tmpdir, delete=False) assert FileSystems.exists(f.name) utils.delete_files([f.name]) assert not FileSystems.exists(f.name)