Ejemplo n.º 1
0
class TestProvisioner(unittest.TestCase):
    def setUp(self):
        self.ec2 = Mock()
        self.provisioner = Provisioner(self.ec2, quiet=True)

    def test_copy(self):
        self.copy(times=1)

    def test_no_copy(self):
        self.copy(times=0)

    def test_multiple_copies(self):
        self.copy(times=4)

    def copy(self, times=1):
        copy = []
        source = []
        target = []
        mode = []

        for i in xrange(0, times):
            source.append("/path/to/source%d" % i)
            target.append("/path/to/target%d" % i)
            mode.append((i + 1) * 222)  # 222, 444, 666

            copy.append({'from': source[i], 'to': target[i], 'mode': mode[i]})

        with contextlib.nested(
                patch('amibaker.provisioner.put', return_value=True),
                patch('amibaker.provisioner.sudo',
                      return_value=True)) as (put, sudo):
            self.provisioner._Provisioner__copy(copy)

            self.assertEqual(
                put.call_count, times,
                "put() should have been called as many times "
                "there are files to be copied")

            calls = [
                put(source[i], target[i], mode=mode[i])
                for i in reversed(xrange(0, times))
            ]

            put.has_calls(calls)
Ejemplo n.º 2
0
class TestProvisioner(unittest.TestCase):
    def setUp(self):
        self.ec2 = Mock()
        self.provisioner = Provisioner(self.ec2, quiet=True)

    def test_copy(self):
        self.copy(times=1)

    def test_no_copy(self):
        self.copy(times=0)

    def test_multiple_copies(self):
        self.copy(times=4)

    def copy(self, times=1):
        copy = []
        source = []
        target = []
        mode = []

        for i in xrange(0, times):
            source.append("/path/to/source%d" % i)
            target.append("/path/to/target%d" % i)
            mode.append((i + 1) * 222)  # 222, 444, 666

            copy.append({"from": source[i], "to": target[i], "mode": mode[i]})

        with contextlib.nested(
            patch("amibaker.provisioner.put", return_value=True), patch("amibaker.provisioner.sudo", return_value=True)
        ) as (put, sudo):
            self.provisioner._Provisioner__copy(copy)

            self.assertEqual(
                put.call_count, times, "put() should have been called as many times " "there are files to be copied"
            )

            calls = [put(source[i], target[i], mode=mode[i]) for i in reversed(xrange(0, times))]

            put.has_calls(calls)
Ejemplo n.º 3
0
 def setUp(self):
     self.ec2 = Mock()
     self.provisioner = Provisioner(self.ec2, quiet=True)
Ejemplo n.º 4
0
 def setUp(self):
     self.ec2 = Mock()
     self.provisioner = Provisioner(self.ec2, quiet=True)