Hide keyboard shortcuts

Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1# This file is part of daf_butler. 

2# 

3# Developed for the LSST Data Management System. 

4# This product includes software developed by the LSST Project 

5# (http://www.lsst.org). 

6# See the COPYRIGHT file at the top-level directory of this distribution 

7# for details of code ownership. 

8# 

9# This program is free software: you can redistribute it and/or modify 

10# it under the terms of the GNU General Public License as published by 

11# the Free Software Foundation, either version 3 of the License, or 

12# (at your option) any later version. 

13# 

14# This program is distributed in the hope that it will be useful, 

15# but WITHOUT ANY WARRANTY; without even the implied warranty of 

16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

17# GNU General Public License for more details. 

18# 

19# You should have received a copy of the GNU General Public License 

20# along with this program. If not, see <http://www.gnu.org/licenses/>. 

21 

22"""Unit tests for the 'repo' shared CLI option. 

23""" 

24 

25import click 

26import click.testing 

27import unittest 

28 

29from lsst.daf.butler.cli.opt import repo_argument 

30 

31 

32@click.command() 

33@repo_argument(required=True) 

34def repoRequired(repo): 

35 pass 

36 

37 

38@click.command() 

39@repo_argument(required=False) 

40def repoNotRequired(repo): 

41 pass 

42 

43 

44@click.command() 

45@repo_argument(help="REPO custom help text") 

46def repoWithHelpText(repo): 

47 pass 

48 

49 

50@click.command() 

51@repo_argument(help=repo_argument.will_create_repo) 

52def repoWithWillCreateHelpText(repo): 

53 pass 

54 

55 

56class Suite(unittest.TestCase): 

57 

58 def testRequired_provided(self): 

59 runner = click.testing.CliRunner() 

60 result = runner.invoke(repoRequired, ["location"]) 

61 self.assertEqual(result.exit_code, 0) 

62 

63 def testRequired_notProvided(self): 

64 runner = click.testing.CliRunner() 

65 result = runner.invoke(repoRequired) 

66 self.assertNotEqual(result.exit_code, 0) 

67 self.assertIn('Error: Missing argument "REPO"', result.output) 

68 

69 def testNotRequired_provided(self): 

70 runner = click.testing.CliRunner() 

71 result = runner.invoke(repoNotRequired, ["location"]) 

72 self.assertEqual(result.exit_code, 0) 

73 

74 def testNotRequired_notProvided(self): 

75 runner = click.testing.CliRunner() 

76 result = runner.invoke(repoNotRequired) 

77 self.assertEqual(result.exit_code, 0) 

78 

79 def testHelp(self): 

80 runner = click.testing.CliRunner() 

81 result = runner.invoke(repoWithHelpText, "--help") 

82 self.assertIn("custom help text", result.output) 

83 

84 def testWillCreateHelpText(self): 

85 runner = click.testing.CliRunner() 

86 result = runner.invoke(repoWithWillCreateHelpText, "--help") 

87 self.assertIn(repo_argument.will_create_repo, result.output.replace("\n ", "")) 

88 

89 

90if __name__ == "__main__": 90 ↛ 91line 90 didn't jump to line 91, because the condition on line 90 was never true

91 unittest.main()