add salt.utils.url unit tests

tests/unit/utils/url_test.py

@@ -0,0 +1,169 @@
+# -*- coding: utf-8 -*-
+
+# Import python libs
+from __future__ import absolute_import
+
+# Import Salt Libs
+import salt.utils.url
+
+# Import Salt Testing Libs
+from salttesting import TestCase, skipIf
+from salttesting.mock import NO_MOCK, NO_MOCK_REASON
+from salttesting.helpers import ensure_in_syspath
+
+ensure_in_syspath('../../')
+
+
+@skipIf(NO_MOCK, NO_MOCK_REASON)
+class UrlTestCase(TestCase):
+    '''
+    TestCase for salt.utils.url module
+    '''
+
+    # parse tests
+
+    def test_parse_path(self):
+        '''
+        Test parsing an ordinary path
+        '''
+        path = 'interesting?/path&.conf:and other things'
+
+        self.assertEqual(salt.utils.url.parse(path), (path, None))
+
+    def test_parse_salt_url(self):
+        '''
+        Test parsing a 'salt://' URL
+        '''
+        path = '?funny/path with {interesting|chars}'
+        url = 'salt://' + path
+
+        self.assertEqual(salt.utils.url.parse(url), (path, None))
+
+    def test_parse_salt_env(self):
+        '''
+        Test parsing a 'salt://' URL with an '?env=' query
+        '''
+        env = 'milieu'
+        path = '?funny/path&with {interesting|chars}'
+        url = 'salt://' + path + '?env=' + env
+
+        self.assertEqual(salt.utils.url.parse(url), (path, env))
+
+    def test_parse_salt_saltenv(self):
+        '''
+        Test parsing a 'salt://' URL with a '?saltenv=' query
+        '''
+        saltenv = 'ambience'
+        path = '?funny/path&with {interesting|chars}'
+        url = 'salt://' + path + '?saltenv=' + saltenv
+
+        self.assertEqual(salt.utils.url.parse(url), (path, saltenv))
+
+    # create tests
+
+    def test_create_url(self):
+        '''
+        Test creating a 'salt://' URL
+        '''
+        path = '? interesting/&path.filetype'
+        url = 'salt://' + path
+
+        self.assertEqual(salt.utils.url.create(path), url)
+
+    def test_create_url_saltenv(self):
+        '''
+        Test creating a 'salt://' URL with a saltenv
+        '''
+        saltenv = 'raumklang'
+        path = '? interesting/&path.filetype'
+
+        url = 'salt://' + path + '?saltenv=' + saltenv
+
+        self.assertEqual(salt.utils.url.create(path, saltenv), url)
+
+    # add_env tests
+
+    def test_add_env_not_salt(self):
+        '''
+        Test not adding a saltenv to a non 'salt://' URL
+        '''
+        saltenv = 'higgs'
+        url = 'https://pdg.lbl.gov/'
+
+        self.assertEqual(salt.utils.url.add_env(url, saltenv), url)
+
+    def test_add_env(self):
+        '''
+        Test adding a saltenv to a 'salt://' URL
+        '''
+        saltenv = 'erstwhile'
+        url = 'salt://salted/file.conf'
+        url_env = url + '?saltenv=' + saltenv
+
+        self.assertEqual(salt.utils.url.add_env(url, saltenv), url_env)
+
+    # split_env tests
+
+    def test_split_env_non_salt(self):
+        '''
+        Test not splitting a saltenv from a non 'salt://' URL
+        '''
+        saltenv = 'gravitodynamics'
+        url = 'https://arxiv.org/find/all/?' + saltenv
+
+        self.assertEqual(salt.utils.url.split_env(url), (url, None))
+
+    def test_split_env(self):
+        '''
+        Test splitting a 'salt://' URL
+        '''
+        saltenv = 'elsewhere'
+        url = 'salt://salted/file.conf'
+        url_env = url + '?saltenv=' + saltenv
+
+        self.assertEqual(salt.utils.url.split_env(url_env), (url, saltenv))
+
+    # validate tests
+
+    def test_validate_valid(self):
+        '''
+        Test URL valid validation
+        '''
+        url = 'salt://config/file.name?saltenv=vapid'
+        protos = ['salt', 'pepper', 'cinnamon', 'melange']
+
+        self.assertTrue(salt.utils.url.validate(url, protos))
+
+    def test_validate_invalid(self):
+        '''
+        Test URL invalid validation
+        '''
+        url = 'cumin://config/file.name?saltenv=vapid'
+        protos = ['salt', 'pepper', 'cinnamon', 'melange']
+
+        self.assertFalse(salt.utils.url.validate(url, protos))
+
+    # strip tests
+
+    def test_strip_url_with_scheme(self):
+        '''
+        Test stripping of URL scheme
+        '''
+        scheme = 'git+salt+rsync+AYB://'
+        resource = 'all/the/things.stuff;parameter?query=I guess'
+        url = scheme + resource
+
+        self.assertEqual(salt.utils.url.strip_proto(url), resource)
+
+    def test_strip_url_without_scheme(self):
+        '''
+        Test stripping of a URL without a scheme
+        '''
+        resource = 'all/the/things.stuff;parameter?query=I guess'
+
+        self.assertEqual(salt.utils.url.strip_proto(resource), resource)
+
+
+if __name__ == '__main__':
+    from integration import run_tests
+    run_tests(UrlTestCase, needs_daemon=False)