From 5303ec39f6f08ccf06f56ff6d5166fc572b2c735 Mon Sep 17 00:00:00 2001 From: Christian Poessinger Date: Sat, 26 Jun 2021 09:17:52 +0200 Subject: vyos.util: add new helper copy_file() Copy a file from A -> B but also support adjusting Bs file permissions and creation of Bs base directory if required. --- src/tests/test_util.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'src/tests') diff --git a/src/tests/test_util.py b/src/tests/test_util.py index 22bc085c5..1efd4868f 100644 --- a/src/tests/test_util.py +++ b/src/tests/test_util.py @@ -1,6 +1,6 @@ #!/usr/bin/env python3 # -# Copyright (C) 2020 VyOS maintainers and contributors +# Copyright (C) 2020-2021 VyOS maintainers and contributors # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License version 2 or later as @@ -15,7 +15,7 @@ # along with this program. If not, see . from unittest import TestCase -from vyos.util import mangle_dict_keys +from vyos.util import * class TestVyOSUtil(TestCase): def test_key_mangline(self): @@ -24,3 +24,9 @@ class TestVyOSUtil(TestCase): new_data = mangle_dict_keys(data, '-', '_') self.assertEqual(new_data, expected_data) + def test_copy_file(self): + source = '/proc/cmdline' + destination = '/tmp/foo/cmdline' + copy_file(source, destination, True) + self.assertEqual(read_file(source), read_file(destination)) + -- cgit v1.2.3