def test_remove_directory(self): """Test file_host.remove_directory.""" result = untrusted_runner_pb2.RemoveDirectoryResponse(result=True) self.mock.stub().RemoveDirectory.return_value = result self.assertTrue(file_host.remove_directory('/path', True)) result = untrusted_runner_pb2.RemoveDirectoryResponse(result=False) self.mock.stub().RemoveDirectory.return_value = result self.assertFalse(file_host.remove_directory('/path', True))
def remove_directory(request, _): """Remove a directory.""" result = shell.remove_directory(request.path, request.recreate) return untrusted_runner_pb2.RemoveDirectoryResponse(result=result)