Python connect Examples

Example #1

File: test_qemu_nbd.py Project: crimv42/saltstack

    def test_connect(self):
        '''
        Test if it activate nbd for an image file.
        '''
        mock = MagicMock(return_value=True)
        with patch.dict(qemu_nbd.__salt__, {'cmd.run': mock}):
            with patch.object(os.path, 'isfile',
                              MagicMock(return_value=False)):
                self.assertEqual(qemu_nbd.connect('/tmp/image.raw'), '')
                self.assertEqual(qemu_nbd.connect('/tmp/image.raw'), '')

        with patch.object(os.path, 'isfile', mock):
            with patch.object(glob, 'glob',
                              MagicMock(return_value=['/dev/nbd0'])):
                with patch.dict(qemu_nbd.__salt__, {
                        'cmd.run': mock,
                        'cmd.retcode': MagicMock(side_effect=[1, 0])
                }):
                    self.assertEqual(qemu_nbd.connect('/tmp/image.raw'),
                                     '/dev/nbd0')

                with patch.dict(qemu_nbd.__salt__, {
                        'cmd.run': mock,
                        'cmd.retcode': MagicMock(return_value=False)
                }):
                    self.assertEqual(qemu_nbd.connect('/tmp/image.raw'), '')

Example #2

File: test_qemu_nbd.py Project: exe01/salt

    def test_connect(self):
        """
        Test if it activate nbd for an image file.
        """
        mock = MagicMock(return_value=True)
        with patch.dict(qemu_nbd.__salt__, {"cmd.run": mock}):
            with patch.object(os.path, "isfile",
                              MagicMock(return_value=False)):
                self.assertEqual(qemu_nbd.connect("/tmp/image.raw"), "")
                self.assertEqual(qemu_nbd.connect("/tmp/image.raw"), "")

        with patch.object(os.path, "isfile", mock):
            with patch.object(glob, "glob",
                              MagicMock(return_value=["/dev/nbd0"])):
                with patch.dict(
                        qemu_nbd.__salt__,
                    {
                        "cmd.run": mock,
                        "cmd.retcode": MagicMock(side_effect=[1, 0])
                    },
                ):
                    self.assertEqual(qemu_nbd.connect("/tmp/image.raw"),
                                     "/dev/nbd0")

                with patch.dict(
                        qemu_nbd.__salt__,
                    {
                        "cmd.run": mock,
                        "cmd.retcode": MagicMock(return_value=False)
                    },
                ):
                    self.assertEqual(qemu_nbd.connect("/tmp/image.raw"), "")

Example #3

File: qemu_nbd_test.py Project: DaveQB/salt

    def test_connect(self):
        '''
        Test if it activate nbd for an image file.
        '''
        mock = MagicMock(return_value=True)
        with patch.dict(qemu_nbd.__salt__, {'cmd.run': mock}):
            with patch.object(os.path, 'isfile',
                              MagicMock(return_value=False)):
                self.assertEqual(qemu_nbd.connect('/tmp/image.raw'), '')
                self.assertEqual(qemu_nbd.connect('/tmp/image.raw'), '')

        with patch.object(os.path, 'isfile', mock):
            with patch.object(glob, 'glob',
                              MagicMock(return_value=['/dev/nbd0'])):
                with patch.dict(qemu_nbd.__salt__,
                                {'cmd.run': mock,
                                 'cmd.retcode': MagicMock(side_effect=[1, 0])}):
                    self.assertEqual(qemu_nbd.connect('/tmp/image.raw'),
                                     '/dev/nbd0')

                with patch.dict(qemu_nbd.__salt__,
                                {'cmd.run': mock,
                                 'cmd.retcode': MagicMock(return_value=False)}):
                    self.assertEqual(qemu_nbd.connect('/tmp/image.raw'), '')