+from mock import patch
+
from .. import lock
class Mock: pass
self.fake_ctx.num_to_lock = 1
self.fake_ctx.lock = False
+ def fake_downburst_executable():
+ return ''
+
+ self.fake_downburst_executable = fake_downburst_executable
+
def test_ubuntu_precise(self):
self.fake_ctx.os_type = 'ubuntu'
self.fake_ctx.os_version = 'precise'
- check_value = lock.vps_version_or_type_valid(
- self.fake_ctx.machine_type,
- self.fake_ctx.os_type,
- self.fake_ctx.os_version)
+ with patch.multiple(
+ lock.provision,
+ downburst_executable=self.fake_downburst_executable,
+ ):
+ check_value = lock.vps_version_or_type_valid(
+ self.fake_ctx.machine_type,
+ self.fake_ctx.os_type,
+ self.fake_ctx.os_version)
assert check_value
def test_ubuntu_number(self):
self.fake_ctx.os_type = 'ubuntu'
self.fake_ctx.os_version = '12.04'
- check_value = lock.vps_version_or_type_valid(
- self.fake_ctx.machine_type,
- self.fake_ctx.os_type,
- self.fake_ctx.os_version)
+ with patch.multiple(
+ lock.provision,
+ downburst_executable=self.fake_downburst_executable,
+ ):
+ check_value = lock.vps_version_or_type_valid(
+ self.fake_ctx.machine_type,
+ self.fake_ctx.os_type,
+ self.fake_ctx.os_version)
assert check_value
def test_mixup(self):
self.fake_ctx.os_type = '6.5'
self.fake_ctx.os_version = 'rhel'
- check_value = lock.vps_version_or_type_valid(
- self.fake_ctx.machine_type,
- self.fake_ctx.os_type,
- self.fake_ctx.os_version)
+ with patch.multiple(
+ lock.provision,
+ downburst_executable=self.fake_downburst_executable,
+ ):
+ check_value = lock.vps_version_or_type_valid(
+ self.fake_ctx.machine_type,
+ self.fake_ctx.os_type,
+ self.fake_ctx.os_version)
assert not check_value
def test_bad_type(self):
self.fake_ctx.os_type = 'aardvark'
self.fake_ctx.os_version = '6.5'
- check_value = lock.vps_version_or_type_valid(
- self.fake_ctx.machine_type,
- self.fake_ctx.os_type,
- self.fake_ctx.os_version)
+ with patch.multiple(
+ lock.provision,
+ downburst_executable=self.fake_downburst_executable,
+ ):
+ check_value = lock.vps_version_or_type_valid(
+ self.fake_ctx.machine_type,
+ self.fake_ctx.os_type,
+ self.fake_ctx.os_version)
assert not check_value
def test_bad_version(self):
self.fake_ctx.os_type = 'rhel'
self.fake_ctx.os_version = 'vampire_bat'
- check_value = lock.vps_version_or_type_valid(
- self.fake_ctx.machine_type,
- self.fake_ctx.os_type,
- self.fake_ctx.os_version)
+ with patch.multiple(
+ lock.provision,
+ downburst_executable=self.fake_downburst_executable,
+ ):
+ check_value = lock.vps_version_or_type_valid(
+ self.fake_ctx.machine_type,
+ self.fake_ctx.os_type,
+ self.fake_ctx.os_version)
assert not check_value