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

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

# -*- Mode:Python; indent-tabs-mode:nil; tab-width:4 -*- 

# 

# Copyright (C) 2015 Canonical Ltd 

# 

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

# it under the terms of the GNU General Public License version 3 as 

# published by the Free Software Foundation. 

# 

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

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

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 

# GNU General Public License for more details. 

# 

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

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

 

import os.path 

from unittest.mock import Mock 

 

from snapcraft.plugins.copy import CopyPlugin 

from snapcraft.tests import TestCase 

 

 

class TestCopyPlugin(TestCase): 

 

    def setUp(self): 

        super().setUp() 

        self.mock_options = Mock() 

        self.mock_options.files = {} 

        # setup the expected target dir in our tempdir 

        self.dst_prefix = 'parts/copy/install/' 

        os.makedirs(self.dst_prefix) 

 

    def test_copy_plugin_any_missing_src_raises_exception(self): 

        # ensure that a bad file causes a warning and fails the build even 

        # if there is a good file last 

        self.mock_options.files = { 

            'src': 'dst', 

            'zzz': 'zzz', 

        } 

        open('zzz', 'w').close() 

        c = CopyPlugin('copy', self.mock_options) 

 

        with self.assertRaises(EnvironmentError) as raised: 

            c.build() 

 

        self.assertEqual(raised.exception.__str__(), 

                         "[Errno 2] No such file or directory: 'src'") 

 

    def test_copy_plugin_copies(self): 

        self.mock_options.files = { 

            'src': 'dst', 

        } 

        open('src', 'w').close() 

 

        c = CopyPlugin('copy', self.mock_options) 

        c.build() 

        self.assertTrue(os.path.exists(os.path.join(self.dst_prefix, 'dst'))) 

 

    def test_copy_plugin_creates_prefixes(self): 

        self.mock_options.files = { 

            'src': 'dir/dst', 

        } 

        open('src', 'w').close() 

 

        c = CopyPlugin('copy', self.mock_options) 

        c.build() 

        self.assertTrue(os.path.exists(os.path.join(self.dst_prefix, 

                                                    'dir/dst'))) 

 

    def test_copy_directories(self): 

        self.mock_options.files = { 

            'dirs1': 'dir/dst', 

        } 

        os.mkdir('dirs1') 

        file = os.path.join('dirs1', 'f') 

        open(file, 'w').close() 

 

        c = CopyPlugin('copy', self.mock_options) 

        c.build() 

        self.assertTrue( 

            os.path.exists(os.path.join(self.dst_prefix, 'dir', 'dst', 'f'))) 

 

    def test_copy_plugin_glob(self): 

        self.mock_options.files = { 

            '*.txt': '.', 

        } 

 

        for filename in ('file-a.txt', 'file-b.txt', 'file-c.notxt'): 

            with open(filename, 'w') as datafile: 

                datafile.write(filename) 

 

        c = CopyPlugin('copy', self.mock_options) 

        c.build() 

 

        self.assertTrue(os.path.exists( 

            os.path.join(self.dst_prefix, 'file-a.txt'))) 

        self.assertTrue(os.path.exists( 

            os.path.join(self.dst_prefix, 'file-b.txt'))) 

        self.assertFalse(os.path.exists( 

            os.path.join(self.dst_prefix, 'file-c.notxt')))