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 daf_butler command line interface commands. 

23""" 

24 

25import click 

26import click.testing 

27import os 

28import unittest 

29import yaml 

30 

31from lsst.daf.butler.cli import butler 

32 

33 

34TESTDIR = os.path.abspath(os.path.dirname(__file__)) 

35 

36os.environ["DAF_BUTLER_PLUGINS"] = "lsst.daf.butler.cli.cmd" 

37 

38 

39class Suite(unittest.TestCase): 

40 

41 def testCreate(self): 

42 """Test creating a repostiry.""" 

43 runner = click.testing.CliRunner() 

44 with runner.isolated_filesystem(): 

45 result = runner.invoke(butler.cli, ["create", "--repo", "here"]) 

46 self.assertEqual(result.exit_code, 0) 

47 

48 def testCreate_outfile(self): 

49 """Test creating a repository and specify an outfile location.""" 

50 runner = click.testing.CliRunner() 

51 with runner.isolated_filesystem(): 

52 result = runner.invoke(butler.cli, ["create", "--repo", "here", "--outfile=there"]) 

53 self.assertEqual(result.exit_code, 0) 

54 self.assertTrue(os.path.exists("there")) # verify the separate config file was made 

55 with open("there", "r") as f: 

56 cfg = yaml.safe_load(f) 

57 self.assertIn("datastore", cfg) 

58 self.assertIn("PosixDatastore", cfg["datastore"]["cls"]) 

59 self.assertIn("root", cfg) 

60 

61 

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

63 unittest.main()