KaneAI - World’s First E2E Software Testing Agent.Book a Demo 
How to use mock_isdir method in Nose
Best Python code snippet using nose
test_set_config.py
Source:test_set_config.py 
1# Licensed under the Apache License, Version 2.0 (the "License");2# you may not use this file except in compliance with the License.3# You may obtain a copy of the License at4#5# http://www.apache.org/licenses/LICENSE-2.06#7# Unless required by applicable law or agreed to in writing, software8# distributed under the License is distributed on an "AS IS" BASIS,9# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.10# See the License for the specific language governing permissions and11# limitations under the License.12import collections13import copy14import imp15import json16import mock17import os.path18import sys19from oslotest import base20# nasty: to import set_config (not a part of the kolla package)21this_dir = os.path.dirname(sys.modules[__name__].__file__)22set_configs_file = os.path.abspath(23 os.path.join(this_dir, '..',24 'docker', 'base', 'set_configs.py'))25set_configs = imp.load_source('set_configs', set_configs_file)26class LoadFromFile(base.BaseTestCase):27 def test_load_ok(self):28 in_config = json.dumps({'command': '/bin/true',29 'config_files': {}})30 mo = mock.mock_open(read_data=in_config)31 with mock.patch.object(set_configs, 'open', mo):32 config = set_configs.load_config()33 set_configs.copy_config(config)34 self.assertEqual([35 mock.call('/var/lib/kolla/config_files/config.json'),36 mock.call().__enter__(),37 mock.call().read(),38 mock.call().__exit__(None, None, None),39 mock.call('/run_command', 'w+'),40 mock.call().__enter__(),41 mock.call().write(u'/bin/true'),42 mock.call().__exit__(None, None, None)], mo.mock_calls)43class LoadFromEnv(base.BaseTestCase):44 def test_load_ok(self):45 in_config = json.dumps({'command': '/bin/true',46 'config_files': {}})47 mo = mock.mock_open()48 with mock.patch.object(set_configs, 'open', mo):49 with mock.patch.dict('os.environ', {'KOLLA_CONFIG': in_config}):50 config = set_configs.load_config()51 set_configs.copy_config(config)52 self.assertEqual([mock.call('/run_command', 'w+'),53 mock.call().__enter__(),54 mock.call().write(u'/bin/true'),55 mock.call().__exit__(None, None, None)],56 mo.mock_calls)57FAKE_CONFIG_FILES = [58 set_configs.ConfigFile(59 '/var/lib/kolla/config_files/bar.conf',60 '/foo/bar.conf', 'user1', '0644')61]62FAKE_CONFIG_FILE = FAKE_CONFIG_FILES[0]63class ConfigFileTest(base.BaseTestCase):64 @mock.patch('os.path.exists', return_value=False)65 def test_delete_path_not_exists(self, mock_exists):66 config_file = copy.deepcopy(FAKE_CONFIG_FILE)67 config_file._delete_path(config_file.dest)68 mock_exists.assert_called_with(config_file.dest)69 @mock.patch('os.path.exists', return_value=True)70 @mock.patch('os.path.isdir', return_value=True)71 @mock.patch('shutil.rmtree')72 def test_delete_path_exist_dir(self,73 mock_rmtree,74 mock_isdir,75 mock_exists):76 config_file = copy.deepcopy(FAKE_CONFIG_FILE)77 config_file._delete_path(config_file.dest)78 mock_exists.assert_called_with(config_file.dest)79 mock_isdir.assert_called_with(config_file.dest)80 mock_rmtree.assert_called_with(config_file.dest)81 @mock.patch('os.path.exists', return_value=True)82 @mock.patch('os.path.isdir', return_value=False)83 @mock.patch('os.remove')84 def test_delete_path_exist_file(self,85 mock_remove,86 mock_isdir,87 mock_exists):88 config_file = copy.deepcopy(FAKE_CONFIG_FILE)89 config_file._delete_path(config_file.dest)90 mock_exists.assert_called_with(config_file.dest)91 mock_isdir.assert_called_with(config_file.dest)92 mock_remove.assert_called_with(config_file.dest)93 @mock.patch('os.chmod')94 @mock.patch('os.chown')95 @mock.patch('pwd.getpwnam')96 def test_set_permission(self,97 mock_getpwnam,98 mock_chown,99 mock_chmod):100 User = collections.namedtuple('User', ['pw_uid', 'pw_gid'])101 user = User(3333, 4444)102 mock_getpwnam.return_value = user103 config_file = copy.deepcopy(FAKE_CONFIG_FILE)104 config_file._set_permission(config_file.dest)105 mock_getpwnam.assert_called_with(config_file.owner)106 mock_chown.assert_called_with(config_file.dest, 3333, 4444)107 mock_chmod.assert_called_with(config_file.dest, 420)108 @mock.patch('glob.glob', return_value=[])109 def test_copy_no_source_not_optional(self,110 mock_glob):111 config_file = copy.deepcopy(FAKE_CONFIG_FILE)112 self.assertRaises(set_configs.MissingRequiredSource,113 config_file.copy)114 @mock.patch('glob.glob', return_value=[])115 def test_copy_no_source_optional(self, mock_glob):116 config_file = copy.deepcopy(FAKE_CONFIG_FILE)117 config_file.optional = True118 config_file.copy()119 mock_glob.assert_called_with(config_file.source)120 @mock.patch.object(set_configs.ConfigFile, '_copy_file')121 @mock.patch('os.path.isdir', return_value=False)122 @mock.patch.object(set_configs.ConfigFile, '_create_parent_dirs')123 @mock.patch.object(set_configs.ConfigFile, '_delete_path')124 @mock.patch('glob.glob')125 def test_copy_one_source_file(self, mock_glob, mock_delete_path,126 mock_create_parent_dirs, mock_isdir,127 mock_copy_file):128 config_file = copy.deepcopy(FAKE_CONFIG_FILE)129 mock_glob.return_value = [config_file.source]130 config_file.copy()131 mock_glob.assert_called_with(config_file.source)132 mock_delete_path.assert_called_with(config_file.dest)133 mock_create_parent_dirs.assert_called_with(config_file.dest)134 mock_isdir.assert_called_with(config_file.source)135 mock_copy_file.assert_called_with(config_file.source,136 config_file.dest)137 @mock.patch.object(set_configs.ConfigFile, '_copy_dir')138 @mock.patch('os.path.isdir', return_value=True)139 @mock.patch.object(set_configs.ConfigFile, '_create_parent_dirs')140 @mock.patch.object(set_configs.ConfigFile, '_delete_path')141 @mock.patch('glob.glob')142 def test_copy_one_source_dir(self, mock_glob, mock_delete_path,143 mock_create_parent_dirs, mock_isdir,144 mock_copy_dir):145 config_file = copy.deepcopy(FAKE_CONFIG_FILE)146 mock_glob.return_value = [config_file.source]147 config_file.copy()148 mock_glob.assert_called_with(config_file.source)149 mock_delete_path.assert_called_with(config_file.dest)150 mock_create_parent_dirs.assert_called_with(config_file.dest)151 mock_isdir.assert_called_with(config_file.source)152 mock_copy_dir.assert_called_with(config_file.source,153 config_file.dest)154 @mock.patch.object(set_configs.ConfigFile, '_copy_file')155 @mock.patch('os.path.isdir', return_value=False)156 @mock.patch.object(set_configs.ConfigFile, '_create_parent_dirs')157 @mock.patch.object(set_configs.ConfigFile, '_delete_path')158 @mock.patch('glob.glob')159 def test_copy_glob_source_file(self, mock_glob, mock_delete_path,160 mock_create_parent_dirs, mock_isdir,161 mock_copy_file):162 config_file = set_configs.ConfigFile(163 '/var/lib/kolla/config_files/bar.*', '/foo/', 'user1', '0644')164 mock_glob.return_value = ['/var/lib/kolla/config_files/bar.conf',165 '/var/lib/kolla/config_files/bar.yml']166 config_file.copy()167 mock_glob.assert_called_with(config_file.source)168 self.assertEqual(mock_delete_path.mock_calls,169 [mock.call('/foo/bar.conf'),170 mock.call('/foo/bar.yml')])171 self.assertEqual(mock_create_parent_dirs.mock_calls,172 [mock.call('/foo/bar.conf'),173 mock.call('/foo/bar.yml')])174 self.assertEqual(mock_isdir.mock_calls,175 [mock.call('/var/lib/kolla/config_files/bar.conf'),176 mock.call('/var/lib/kolla/config_files/bar.yml')])177 self.assertEqual(mock_copy_file.mock_calls,178 [mock.call('/var/lib/kolla/config_files/bar.conf',179 '/foo/bar.conf'),180 mock.call('/var/lib/kolla/config_files/bar.yml',181 '/foo/bar.yml')])182 @mock.patch.object(set_configs.ConfigFile, '_cmp_file')183 @mock.patch('os.path.isdir', return_value=False)184 @mock.patch('glob.glob')185 def test_check_glob_source_file(self, mock_glob, mock_isdir,186 mock_cmp_file):187 config_file = set_configs.ConfigFile(188 '/var/lib/kolla/config_files/bar.*', '/foo/', 'user1', '0644')189 mock_glob.return_value = ['/var/lib/kolla/config_files/bar.conf',190 '/var/lib/kolla/config_files/bar.yml']191 mock_cmp_file.return_value = True192 config_file.check()193 self.assertEqual(mock_isdir.mock_calls,194 [mock.call('/var/lib/kolla/config_files/bar.conf'),195 mock.call('/var/lib/kolla/config_files/bar.yml')])196 self.assertEqual(mock_cmp_file.mock_calls,197 [mock.call('/var/lib/kolla/config_files/bar.conf',198 '/foo/bar.conf'),199 mock.call('/var/lib/kolla/config_files/bar.yml',200 '/foo/bar.yml')])201 @mock.patch.object(set_configs.ConfigFile, '_cmp_file')202 @mock.patch('os.path.isdir', return_value=False)203 @mock.patch('glob.glob')204 def test_check_glob_source_file_no_equal(self, mock_glob, mock_isdir,205 mock_cmp_file):206 config_file = set_configs.ConfigFile(207 '/var/lib/kolla/config_files/bar.*', '/foo/', 'user1', '0644')208 mock_glob.return_value = ['/var/lib/kolla/config_files/bar.conf',209 '/var/lib/kolla/config_files/bar.yml']210 mock_cmp_file.side_effect = [True, False]211 self.assertRaises(set_configs.ConfigFileBadState,212 config_file.check)213 self.assertEqual(mock_isdir.mock_calls,214 [mock.call('/var/lib/kolla/config_files/bar.conf'),215 mock.call('/var/lib/kolla/config_files/bar.yml')])216 self.assertEqual(mock_cmp_file.mock_calls,217 [mock.call('/var/lib/kolla/config_files/bar.conf',218 '/foo/bar.conf'),219 mock.call('/var/lib/kolla/config_files/bar.yml',...
test_globmatch.py
Source:test_globmatch.py
1# -*- Mode:Python; indent-tabs-mode:nil; tab-width:4 -*-2#3# Copyright 2002 Ben Escoto <ben@emerose.org>4# Copyright 2007 Kenneth Loafman <kenneth@loafman.com>5# Copyright 2014 Aaron Whitehouse <aaron@whitehouse.kiwi.nz>6#7# This file is part of duplicity.8#9# Duplicity is free software; you can redistribute it and/or modify it10# under the terms of the GNU General Public License as published by the11# Free Software Foundation; either version 2 of the License, or (at your12# option) any later version.13#14# Duplicity is distributed in the hope that it will be useful, but15# WITHOUT ANY WARRANTY; without even the implied warranty of16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU17# General Public License for more details.18#19# You should have received a copy of the GNU General Public License20# along with duplicity; if not, write to the Free Software Foundation,21# Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA22import sys23from duplicity.globmatch import *24from duplicity.path import *25from . import UnitTestCase26from mock import patch27import unittest28class MatchingTest(UnitTestCase):29 """Test matching of file names against various selection functions"""30 def test_glob_re(self):31 """test_glob_re - test translation of shell pattern to regular exp"""32 assert glob_to_regex("hello") == "hello"33 assert glob_to_regex(".e?ll**o") == "\\.e[^/]ll.*o"34 r = glob_to_regex("[abc]el[^de][!fg]h")35 assert r == "[abc]el[^de][^fg]h", r36 r = glob_to_regex("/usr/*/bin/")37 assert r == "\\/usr\\/[^/]*\\/bin\\/", r38 assert glob_to_regex("[a.b/c]") == "[a.b/c]"39 r = glob_to_regex("[a*b-c]e[!]]")40 assert r == "[a*b-c]e[^]]", r41class TestDoubleAsteriskOnIncludesExcludes(UnitTestCase):42 """Test ** on includes and exclude patterns"""43 def test_double_asterisk_include(self):44 """Test a few globbing patterns, including **"""45 self.assertEqual(46 path_matches_glob_fn("**", 1)(Path("foo.txt")), 1)47 with patch('duplicity.path.Path.isdir') as mock_isdir:48 mock_isdir.return_value = True49 self.assertEqual(50 path_matches_glob_fn("**", 1)(Path("folder")), 1)51 def test_double_asterisk_extension_include(self):52 """Test **.py"""53 self.assertEqual(54 path_matches_glob_fn("**.py", 1)(Path("what/ever.py")), 1)55 self.assertEqual(56 path_matches_glob_fn("**.py", 1)(Path("what/ever.py/foo")), 1)57 with patch('duplicity.path.Path.isdir') as mock_isdir:58 mock_isdir.return_value = True59 self.assertEqual(60 path_matches_glob_fn("**.py", 1)(Path("foo")), 2)61 self.assertEqual(62 path_matches_glob_fn("**.py", 1)(Path("usr/local/bin")), 2)63 self.assertEqual(64 path_matches_glob_fn("**.py", 1)(Path("/usr/local/bin")), 2)65class TestTrailingSlash(UnitTestCase):66 """Test glob matching where the glob has a trailing slash"""67 def test_trailing_slash_matches_only_dirs(self):68 """Test matching where glob includes a trailing slash"""69 with patch('duplicity.path.Path.isdir') as mock_isdir:70 mock_isdir.return_value = True71 self.assertEqual(72 path_matches_glob_fn("fold*/", 1)(Path("folder")), 1)73 # Test the file named "folder" is not included if it is not a dir74 mock_isdir.return_value = False75 self.assertEqual(76 path_matches_glob_fn("fold*/", 1)(Path("folder")), None)77 # Test the file named "folder" is not included if it is not a dir78 mock_isdir.return_value = False79 self.assertEqual(80 path_matches_glob_fn("folder/", 1)(Path("folder")), None)81 mock_isdir.return_value = False82 self.assertEqual(83 path_matches_glob_fn("fo*/", 1)(Path("foo.txt")), None)84 def test_included_files_are_matched_no_slash(self):85 """Test that files within an included folder are matched"""86 with patch('duplicity.path.Path.isdir') as mock_isdir:87 mock_isdir.return_value = False88 self.assertEqual(89 path_matches_glob_fn("fold*", 1)(Path("folder/file.txt")), 1)90 with patch('duplicity.path.Path.isdir') as mock_isdir:91 mock_isdir.return_value = False92 self.assertEqual(93 path_matches_glob_fn("fold*", 1)(Path("folder/2/file.txt")), 1)94 def test_included_files_are_matched_no_slash_2(self):95 """Test that files within an included folder are matched"""96 with patch('duplicity.path.Path.isdir') as mock_isdir:97 mock_isdir.return_value = False98 self.assertEqual(99 path_matches_glob_fn("folder", 1)(Path("folder/file.txt")), 1)100 with patch('duplicity.path.Path.isdir') as mock_isdir:101 mock_isdir.return_value = False102 self.assertEqual(103 path_matches_glob_fn("folder/2", 1)(Path("folder/2/file.txt")), 1)104 def test_included_files_are_matched_slash(self):105 """Test that files within an included folder are matched with /"""106 # Bug #1624725107 # https://bugs.launchpad.net/duplicity/+bug/1624725108 with patch('duplicity.path.Path.isdir') as mock_isdir:109 mock_isdir.return_value = False110 self.assertEqual(111 path_matches_glob_fn("folder/", 1)(Path("folder/file.txt")), 1)112 def test_included_files_are_matched_slash_2(self):113 """Test that files within an included folder are matched with /"""114 # Bug #1624725115 # https://bugs.launchpad.net/duplicity/+bug/1624725116 with patch('duplicity.path.Path.isdir') as mock_isdir:117 mock_isdir.return_value = False118 self.assertEqual(119 path_matches_glob_fn("testfiles/select2/1/1sub1/1sub1sub1/", 1)120 (Path("testfiles/select2/1/1sub1/1sub1sub1/1sub1sub1_file.txt")121 ), 1)122 def test_included_files_are_matched_slash_2_parents(self):123 """Test that duplicity will scan parent of glob/"""124 # Bug #1624725125 # https://bugs.launchpad.net/duplicity/+bug/1624725126 with patch('duplicity.path.Path.isdir') as mock_isdir:127 mock_isdir.return_value = True128 self.assertEqual(129 path_matches_glob_fn("testfiles/select2/1/1sub1/1sub1sub1/", 1)130 (Path("testfiles/select2/1/1sub1/1sub1sub1")131 ), 1)132 self.assertEqual(133 path_matches_glob_fn("testfiles/select2/1/1sub1/1sub1sub1/", 1)134 (Path("testfiles/select2/1/1sub1")135 ), 2)136 def test_included_files_are_matched_slash_wildcard(self):137 """Test that files within an included folder are matched with /"""138 # Bug #1624725139 # https://bugs.launchpad.net/duplicity/+bug/1624725140 with patch('duplicity.path.Path.isdir') as mock_isdir:141 mock_isdir.return_value = False142 self.assertEqual(143 path_matches_glob_fn("fold*/", 1)(Path("folder/file.txt")), 1)144 #145 # def test_slash_matches_everything(self):146 # """Test / matches everything"""147 # # ToDo: Not relevant at this stage, as "/" would not go through148 # # globmatch because it has no special characters, but it should be149 # # made to work150 # with patch('duplicity.path.Path.isdir') as mock_isdir:151 # mock_isdir.return_value = True152 # self.assertEqual(153 # path_matches_glob_fn("/",154 # 1)(Path("/tmp/testfiles/select/1/2")), 1)155 # self.assertEqual(156 # path_matches_glob_fn("/",157 # 1)(Path("/test/random/path")), 1)158 # self.assertEqual(159 # path_matches_glob_fn("/",160 # 1)(Path("/")), 1)161 # self.assertEqual(162 # path_matches_glob_fn("/",163 # 1)(Path("/var/log")), 1)164 # self.assertEqual(165 # path_matches_glob_fn("/",166 # 1)(Path("/var/log/log.txt")), 1)167 def test_slash_star_scans_folder(self):168 """Test that folder/* scans folder/"""169 # This behaviour is a bit ambiguous - either include or scan could be170 # argued as most appropriate here, but only an empty folder is at stake171 # so long as test_slash_star_includes_folder_contents passes.172 with patch('duplicity.path.Path.isdir') as mock_isdir:173 mock_isdir.return_value = True174 self.assertEqual(175 path_matches_glob_fn("folder/*", 1)(Path("folder")), 2)176 def test_slash_star_includes_folder_contents(self):177 """Test that folder/* includes folder contents"""178 self.assertEqual(path_matches_glob_fn("folder/*", 1)179 (Path("folder/file.txt")), 1)180 self.assertEqual(path_matches_glob_fn("folder/*", 1)181 (Path("folder/other_file.log")), 1)182 def test_slash_star_star_includes_folder(self):183 """Test that folder/** includes folder/"""184 with patch('duplicity.path.Path.isdir') as mock_isdir:185 mock_isdir.return_value = True186 def test_simple_trailing_slash_match(self):187 """Test that a normal folder string ending in / matches that path"""188 with patch('duplicity.path.Path.isdir') as mock_isdir:189 mock_isdir.return_value = True190 test_path = "testfiles/select/1/2/1"191 self.assertEqual(192 path_matches_glob_fn(test_path, 1)(Path(test_path)), 1)193 def test_double_asterisk_string_slash(self):194 """Test string starting with ** and ending in /"""195 with patch('duplicity.path.Path.isdir') as mock_isdir:196 mock_isdir.return_value = True197 self.assertEqual(198 path_matches_glob_fn("**/1/2/",199 1)(Path("testfiles/select/1/2")), 1)200 def test_string_double_asterisk_string_slash(self):201 """Test string ** string /"""202 with patch('duplicity.path.Path.isdir') as mock_isdir:203 mock_isdir.return_value = True204 self.assertEqual(205 path_matches_glob_fn("testfiles**/2/",206 1)(Path("testfiles/select/1/2")), 1)207class TestDoubleAsterisk(UnitTestCase):208 """Test glob matching where the glob finishes with a **"""209 def test_simple_trailing_slash_match(self):210 """Test that a folder string ending in /** matches that path"""211 with patch('duplicity.path.Path.isdir') as mock_isdir:212 mock_isdir.return_value = True213 self.assertEqual(214 path_matches_glob_fn("/test/folder/**", 1)(215 Path("/test/foo")), None)216 def test_simple_trailing_slash_match_2(self):217 """Test folder string ending in */**"""218 with patch('duplicity.path.Path.isdir') as mock_isdir:219 mock_isdir.return_value = True220 self.assertEqual(221 path_matches_glob_fn("fold*/**", 1)(...
arguments_test.py
Source:arguments_test.py
1"""Test suite for arguments.py"""2from mock import patch3import pytest4from ocrcode import arguments5class TestArgumentParser:6 """ Test class for arguments.argument_parser """7 @pytest.mark.parametrize(8 "params_in, expected",9 [10 (["file.jpg"], ["file.jpg"]),11 (["file.jpg", "-v"], ["file.jpg"]),12 (["file.jpg", "--verbose"], ["file.jpg"]),13 (["file.jpg", "-s", "dir"], ["file.jpg"]),14 (["file.jpg", "--save", "dir"], ["file.jpg"]),15 (["file.jpg", "-s", "dir", "-v"], ["file.jpg"]),16 (["file.jpg", "file2.jpg"], ["file.jpg", "file2.jpg"]),17 ([], []),18 (["-v"], []),19 (["-s", "dir", "-v", "-t", "dir2"], []),20 ],21 )22 def test_files_to_process_added_to_list(self, params_in, expected):23 files, _, _, _ = arguments.argument_parser(params_in)24 assert files == expected25 @pytest.mark.parametrize(26 "params_in, expected",27 [28 (["file.jpg", "-v"], None),29 (["file.jpg", "--verbose"], None),30 (["file.jpg", "-s", "dir", "-v"], "dir"),31 (["file.jpg", "-s", "dir", "--verbose", "-t", "dir2"], "dir"),32 (["file.jpg", "--save", "dir"], "dir"),33 (["file.jpg"], None),34 ],35 )36 def test_save_flag_returns_save_path_if_given(self, params_in, expected):37 _, save_path, _, _ = arguments.argument_parser(params_in)38 if save_path is None:39 assert save_path is expected40 else:41 assert save_path == expected42 @pytest.mark.parametrize(43 "params_in, expected",44 [45 (["file.jpg", "-v"], True),46 (["file.jpg", "--verbose"], True),47 (["file.jpg", "-s", "dir", "-v"], True),48 (["file.jpg", "-s", "dir", "--verbose"], True),49 (["file.jpg", "-s", "dir", "-t", "dir"], False),50 (["file.jpg"], False),51 ],52 )53 def test_verbose_flag_returns_verbose_true(self, params_in, expected):54 _, _, verbose, _ = arguments.argument_parser(params_in)55 assert verbose == expected56 @pytest.mark.parametrize(57 "params_in, expected",58 [59 (["file.jpg", "-v"], None),60 (["file.jpg", "-s", "dir", "--verbose"], None),61 (["file.jpg", "--tesseract", "dir"], "dir"),62 (["file.jpg", "-t", "dir"], "dir"),63 (["file.jpg", "-s", "dir", "--verbose", "-t", "dir"], "dir"),64 ],65 )66 def test_tesseract_flag_returns_tesseract_path(self, params_in, expected):67 _, _, _, tesseract = arguments.argument_parser(params_in)68 assert tesseract == expected69class TestPathsToFiles:70 """ Test class for arguments.paths_to_files """71 @patch("os.path.isdir")72 @patch("os.path.isfile")73 def test_returns_empty_list_for_invalid_file(self, mock_isfile, mock_isdir):74 mock_isdir.return_value = False75 mock_isfile.return_value = False76 with pytest.raises(FileNotFoundError):77 arguments.paths_to_files(["invalid"])78 @patch("os.path.isdir")79 @patch("os.path.isfile")80 def test_raises_error_if_not_image(self, mock_isfile, mock_isdir):81 mock_isdir.return_value = False82 mock_isfile.return_value = True83 with pytest.raises(FileNotFoundError):84 arguments.paths_to_files(["valid.txt"])85 @patch("os.path.isdir")86 @patch("os.path.isfile")87 def test_returns_list_if_valid_image(self, mock_isfile, mock_isdir):88 mock_isdir.return_value = False89 mock_isfile.return_value = True90 assert arguments.paths_to_files(["valid.jpg"]) == ["valid.jpg"]91 @patch("os.listdir")92 @patch("os.path.isdir")93 @patch("os.path.isfile")94 def test_returns_list_if_valid_dir(95 self, mock_isfile, mock_isdir, mock_listdir,96 ):97 mock_isdir.return_value = True98 mock_isfile.return_value = True99 mock_listdir.return_value = ["valid.jpg", "other.txt"]100 assert arguments.paths_to_files(["dir/"]) == ["dir/valid.jpg"]101 @patch("os.listdir")102 @patch("os.path.isdir")103 @patch("os.path.isfile")104 def test_returns_empty_list_if_valid_dir_but_not_image(105 self, mock_isfile, mock_isdir, mock_listdir,106 ):107 mock_isdir.return_value = True108 mock_isfile.return_value = True109 mock_listdir.return_value = ["valid.txt"]110 with pytest.raises(FileNotFoundError):111 arguments.paths_to_files(["dir/"])112class TestValidateSave:113 """ Test class for arguments.validate_save """114 @patch("os.path.isdir")115 @patch("os.getcwd")116 def test_returns_none_when_no_save(self, mock_getcwd, mock_isdir):117 mock_getcwd.return_value = "cwd"118 mock_isdir.return_value = False119 assert arguments.validate_save(save_path=None) is None120 @patch("os.path.isdir")121 @patch("os.getcwd")122 def test_returns_cwd_when_no_parameter(self, mock_getcwd, mock_isdir):123 mock_getcwd.return_value = "cwd"124 mock_isdir.return_value = False125 assert arguments.validate_save(save_path="||cwd||") == "cwd"126 @patch("os.path.isdir")127 @patch("os.getcwd")128 def test_returns_valid_path(self, mock_getcwd, mock_isdir):129 mock_getcwd.return_value = "cwd"130 mock_isdir.return_value = True131 assert arguments.validate_save(save_path="valid") == "valid"132 @patch("os.path.isdir")133 @patch("os.getcwd")134 def test_raises_exception_for_invalid_path(self, mock_getcwd, mock_isdir):135 mock_getcwd.return_value = "cwd"136 mock_isdir.return_value = False137 with pytest.raises(Exception):138 arguments.validate_save(save_path="invalid")139class TestValidateTesseract:140 """ Test class for arguments.validate_tesseract """141 @patch("os.path.isfile")142 def test_returns_none_when_passed_none(self, mock_isfile):143 mock_isfile.return_value = False144 assert arguments.validate_tesseract(None) is None145 @patch("os.path.isfile")146 def test_returns_valid_tesseract_path(self, mock_isfile):147 mock_isfile.return_value = True148 input = "C:\\Program Files\\Tesseract-OCR\\tesseract.exe"149 expected = "C:\\Program Files\\Tesseract-OCR\\tesseract.exe"150 assert arguments.validate_tesseract(input) == expected151 @patch("os.path.isfile")152 def test_raises_error_for_invalid_path(self, mock_isfile):153 mock_isfile.return_value = False154 with pytest.raises(FileNotFoundError):155 arguments.validate_tesseract("invalid")156 @patch("os.path.isfile")157 def test_raises_error_for_non_tesseract_path(self, mock_isfile):158 mock_isfile.return_value = True159 with pytest.raises(FileNotFoundError):...
config_validator_scanner_test.py
Source:config_validator_scanner_test.py
1# Copyright 2020 The Forseti Security Authors. All rights reserved.2#3# Licensed under the Apache License, Version 2.0 (the "License");4# you may not use this file except in compliance with the License.5# You may obtain a copy of the License at6#7# http://www.apache.org/licenses/LICENSE-2.08#9# Unless required by applicable law or agreed to in writing, software10# distributed under the License is distributed on an "AS IS" BASIS,11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.12# See the License for the specific language governing permissions and13# limitations under the License.14import unittest.mock as mock15from unittest.mock import patch16from tests.unittest_utils import ForsetiTestCase17from google.cloud.forseti.scanner.scanners.config_validator_scanner import (18 ConfigValidatorScanner)19from google.cloud.forseti.scanner.scanners.config_validator_util import (20 errors)21class ConfigValidatorScannerTest(ForsetiTestCase):22 """Tests for the Config Validator Scanner."""23 def setUp(self):24 self.scanner = ConfigValidatorScanner({},25 {},26 mock.MagicMock(),27 'ScannerBuilderTestModel',28 '20201237T121212Z',29 '')30 @patch('os.listdir')31 @patch('os.path.isdir')32 def test_verify_policy_library_empty_constraints_raises_exception(self,33 mock_isdir,34 mock_listdir):35 mock_isdir.side_effect = [True, True, True]36 mock_listdir.side_effect = [[], ['file1']]37 self.assertRaises(errors.ConfigValidatorPolicyLibraryError, self.scanner.verify_policy_library)38 self.assertEqual(mock_listdir.call_count, 1)39 @patch('os.listdir')40 @patch('os.path.isdir')41 def test_verify_policy_library_empty_lib_raises_exception(self,42 mock_isdir,43 mock_listdir):44 mock_isdir.side_effect = [True, True, True]45 mock_listdir.side_effect = [['file1'], []]46 self.assertRaises(errors.ConfigValidatorPolicyLibraryError, self.scanner.verify_policy_library)47 self.assertEqual(mock_listdir.call_count, 2)48 @patch('os.listdir')49 @patch('os.path.isdir')50 def test_verify_policy_library_missing_constraints_raises_exception(self,51 mock_isdir,52 mock_listdir):53 mock_isdir.side_effect = [True, False, True]54 mock_listdir.side_effect = [['file1'], ['file1']]55 self.assertRaises(errors.ConfigValidatorPolicyLibraryError, self.scanner.verify_policy_library)56 self.assertEqual(mock_isdir.call_count, 2)57 @patch('os.listdir')58 @patch('os.path.isdir')59 def test_verify_policy_library_missing_lib_raises_exception(self,60 mock_isdir,61 mock_listdir):62 mock_isdir.side_effect = [True, True, False]63 mock_listdir.side_effect = [['file1'], ['file1']]64 self.assertRaises(errors.ConfigValidatorPolicyLibraryError, self.scanner.verify_policy_library)65 self.assertEqual(mock_isdir.call_count, 3)66 @patch('os.listdir')67 @patch('os.path.isdir')68 def test_verify_policy_library_missing_library_raises_exception(self,69 mock_isdir,70 mock_listdir):71 mock_isdir.side_effect = [False, True, True]72 mock_listdir.side_effect = [['file1'], ['file1']]73 self.assertRaises(errors.ConfigValidatorPolicyLibraryError, self.scanner.verify_policy_library)74 self.assertEqual(mock_isdir.call_count, 1)75 @patch('os.listdir')76 @patch('os.path.isdir')77 def test_verify_policy_library_success(self, mock_isdir, mock_listdir):78 mock_listdir.side_effect = [['file1'], ['file1']]79 mock_isdir.side_effect = [True, True, True]80 self.scanner.verify_policy_library()81 self.assertEqual(mock_listdir.call_count, 2)...
Automation Testing Tutorials
Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.
LambdaTest Learning Hubs:
- JUnit Tutorial
- TestNG Tutorial
- Webdriver Tutorial
- WebDriverIO Tutorial
- Protractor Tutorial
- Selenium 4 Tutorial
- Jenkins Tutorial
- NUnit Tutorial
- Jest Tutorial
- Playwright Tutorial
- Cypress Tutorial
- PyTest Tutorial
YouTube
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Run Nose automation tests on LambdaTest cloud grid
Perform automation testing on 3000+ real desktop and mobile devices online.
Try LambdaTest Now !!
Get 100 minutes of automation test minutes FREE!!
Was this article helpful?
