summaryrefslogtreecommitdiff
path: root/tests/utils
diff options
context:
space:
mode:
authorƁukasz 'sil2100' Zemczak <lukasz.zemczak@ubuntu.com>2017-05-18 19:58:02 +0200
committerusd-importer <ubuntu-server@lists.ubuntu.com>2017-05-31 09:53:12 +0000
commit4fb0b5a09b26135ade285844da5d7dfe582a8d4c (patch)
tree09b1e5867d6e7501118cdd0af0012b51fc216530 /tests/utils
parent473ad6fbfe0b9c3b362b530492928303f2b4c7f3 (diff)
downloadvyos-walinuxagent-4fb0b5a09b26135ade285844da5d7dfe582a8d4c.tar.gz
vyos-walinuxagent-4fb0b5a09b26135ade285844da5d7dfe582a8d4c.zip
Import patches-unapplied version 2.2.12-0ubuntu1 to ubuntu/artful-proposed
Imported using git-ubuntu import. Changelog parent: 473ad6fbfe0b9c3b362b530492928303f2b4c7f3 New changelog entries: * New upstream release (LP: #1690854). - Refreshed debian/patches/disable_import_test.patch.
Diffstat (limited to 'tests/utils')
-rw-r--r--tests/utils/test_file_util.py30
1 files changed, 25 insertions, 5 deletions
diff --git a/tests/utils/test_file_util.py b/tests/utils/test_file_util.py
index 76fb15b..0b92513 100644
--- a/tests/utils/test_file_util.py
+++ b/tests/utils/test_file_util.py
@@ -15,9 +15,14 @@
# Requires Python 2.4+ and Openssl 1.0+
#
+import glob
+import random
+import string
+import tempfile
import uuid
import azurelinuxagent.common.utils.fileutil as fileutil
+
from azurelinuxagent.common.future import ustr
from tests.tools import *
@@ -69,9 +74,6 @@ class TestFileOperations(AgentTestCase):
self.assertEquals('abc', filename)
def test_remove_files(self):
- import random
- import string
- import glob
random_word = lambda : ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(5))
#Create 10 test files
@@ -90,9 +92,27 @@ class TestFileOperations(AgentTestCase):
self.assertEqual(0, len(glob.glob(os.path.join(self.tmp_dir, test_file_pattern))))
self.assertEqual(0, len(glob.glob(os.path.join(self.tmp_dir, test_file_pattern2))))
+ def test_remove_dirs(self):
+ dirs = []
+ for n in range(0,5):
+ dirs.append(tempfile.mkdtemp())
+ for d in dirs:
+ for n in range(0, random.choice(range(0,10))):
+ fileutil.write_file(os.path.join(d, "test"+str(n)), "content")
+ for n in range(0, random.choice(range(0,10))):
+ dd = os.path.join(d, "testd"+str(n))
+ os.mkdir(dd)
+ for nn in range(0, random.choice(range(0,10))):
+ os.symlink(dd, os.path.join(dd, "sym"+str(nn)))
+ for n in range(0, random.choice(range(0,10))):
+ os.symlink(d, os.path.join(d, "sym"+str(n)))
+
+ fileutil.rm_dirs(*dirs)
+
+ for d in dirs:
+ self.assertEqual(len(os.listdir(d)), 0)
+
def test_get_all_files(self):
- import random
- import string
random_word = lambda: ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(5))
# Create 10 test files at the root dir and 10 other in the sub dir