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

import iscsilib 

import mock 

import unittest 

 

 

class Test_iscsilib(unittest.TestCase): 

 

    @mock.patch('iscsilib.get_rootdisk_IQNs') 

    @mock.patch('util.doexec') 

    def test_save_rootdisk_nodes(self, doexec, get_rootdisk_iqns): 

        get_rootdisk_iqns.return_value = ['iqn.2003-01.com.bla:00.ecd28.mo121'] 

 

        iscsilib.save_rootdisk_nodes() 

 

        doexec.assert_called_with(['/bin/cp', '-a', 

                                   '/var/lib/iscsi/nodes/iqn.2003-01.com.bla:' 

                                   '00.ecd28.mo121', 

                                   '/tmp']) 

 

    @mock.patch('iscsilib.get_rootdisk_IQNs') 

    @mock.patch('util.doexec') 

    def test_restore_rootdisk_nodes(self, doexec, get_rootdisk_iqns): 

        get_rootdisk_iqns.return_value = ['iqn.2003-01.com.bla:00.ecd28.mo121'] 

 

        iscsilib.restore_rootdisk_nodes() 

 

        doexec.assert_called_with(['/bin/cp', '-a', 

                                   '/tmp/iqn.2003-01.com.bla:00.ecd28.mo121', 

                                   '/var/lib/iscsi/nodes']) 

 

    @mock.patch('iscsilib.stop_daemon', mock.Mock()) 

    @mock.patch('iscsilib.exn_on_failure', mock.Mock()) 

    @mock.patch('util.doexec', mock.Mock()) 

    @mock.patch('os.path.exists') 

    @mock.patch('shutil.rmtree') 

    def test_restart_daemon(self, rmtree, exists): 

        exists.return_value = True 

 

        iscsilib.restart_daemon() 

 

        rmtree.assert_has_calls([mock.call('/var/lib/iscsi/nodes'), 

                                 mock.call('/var/lib/iscsi/send_targets')])