diff options
author | Ben Howard <ben.howard@ubuntu.com> | 2016-02-08 16:33:07 -0700 |
---|---|---|
committer | usd-importer <ubuntu-server@lists.ubuntu.com> | 2016-02-09 00:59:05 +0000 |
commit | a00729ff7421b3661e8b1a1e0fa46393379f2e96 (patch) | |
tree | 4563b927e3a57446a4a928a72a92d72c9ad4f6e6 /tests/utils/test_file_util.py | |
parent | 53f54030cae2de3d5fa474a61fe51f16c7a07c79 (diff) | |
download | vyos-walinuxagent-a00729ff7421b3661e8b1a1e0fa46393379f2e96.tar.gz vyos-walinuxagent-a00729ff7421b3661e8b1a1e0fa46393379f2e96.zip |
Import patches-unapplied version 2.1.3-0ubuntu1 to ubuntu/xenial-proposed
Imported using git-ubuntu import.
Changelog parent: 53f54030cae2de3d5fa474a61fe51f16c7a07c79
New changelog entries:
* New upstream release (LP: #1543359):
- Bug fixes for extension handling
- Feature enablement for AzureStack.
Diffstat (limited to 'tests/utils/test_file_util.py')
-rw-r--r-- | tests/utils/test_file_util.py | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/tests/utils/test_file_util.py b/tests/utils/test_file_util.py new file mode 100644 index 0000000..bf7c638 --- /dev/null +++ b/tests/utils/test_file_util.py @@ -0,0 +1,75 @@ +# Copyright 2014 Microsoft Corporation +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# Requires Python 2.4+ and Openssl 1.0+ +# +# Implements parts of RFC 2131, 1541, 1497 and +# http://msdn.microsoft.com/en-us/library/cc227282%28PROT.10%29.aspx +# http://msdn.microsoft.com/en-us/library/cc227259%28PROT.13%29.aspx + +from tests.tools import * +import uuid +import unittest +import os +import sys +from azurelinuxagent.future import ustr +import azurelinuxagent.utils.fileutil as fileutil + +class TestFileOperations(AgentTestCase): + def test_read_write_file(self): + test_file=os.path.join(self.tmp_dir, 'test_file') + content = ustr(uuid.uuid4()) + fileutil.write_file(test_file, content) + + content_read = fileutil.read_file(test_file) + self.assertEquals(content, content_read) + os.remove(test_file) + + def test_rw_utf8_file(self): + test_file=os.path.join(self.tmp_dir, 'test_file') + content = u"\u6211" + fileutil.write_file(test_file, content, encoding="utf-8") + + content_read = fileutil.read_file(test_file) + self.assertEquals(content, content_read) + os.remove(test_file) + + def test_remove_bom(self): + test_file=os.path.join(self.tmp_dir, 'test_file') + data = b'\xef\xbb\xbfhehe' + fileutil.write_file(test_file, data, asbin=True) + data = fileutil.read_file(test_file, remove_bom=True) + self.assertNotEquals(0xbb, ord(data[0])) + + def test_append_file(self): + test_file=os.path.join(self.tmp_dir, 'test_file') + content = ustr(uuid.uuid4()) + fileutil.append_file(test_file, content) + + content_read = fileutil.read_file(test_file) + self.assertEquals(content, content_read) + + os.remove(test_file) + + def test_get_last_path_element(self): + filepath = '/tmp/abc.def' + filename = fileutil.base_name(filepath) + self.assertEquals('abc.def', filename) + + filepath = '/tmp/abc' + filename = fileutil.base_name(filepath) + self.assertEquals('abc', filename) + +if __name__ == '__main__': + unittest.main() |