tests: Make clean_file() handle directories

Signed-off-by: Volker Lendecke <vl@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/python/samba/tests/smb2symlink.py b/python/samba/tests/smb2symlink.py
index 81e75718f8220781dc879b1b9614c9b9e8d8ba55..3ac35e52c09a6dd12c79d6d5b9c1f7e6ace9319f 100644 (file)
--- a/python/samba/tests/smb2symlink.py
+++ b/python/samba/tests/smb2symlink.py
@@ -59,7 +59,10 @@ class Smb2SymlinkTests(samba.tests.libsmb.LibsmbTests):
         try:
             conn.unlink(filename)
         except NTSTATUSError as e:
-            if e.args[0] != ntstatus.NT_STATUS_OBJECT_NAME_NOT_FOUND:
+            if e.args[0] == ntstatus.NT_STATUS_FILE_IS_A_DIRECTORY:
+                conn.rmdir(filename)
+            elif not (e.args[0] == ntstatus.NT_STATUS_OBJECT_NAME_NOT_FOUND or
+                      e.args[0] == ntstatus.NT_STATUS_OBJECT_PATH_NOT_FOUND):
                 raise
 
     def create_symlink(self, conn, target, symlink):