Python _get_vmの例

コード例 #1

ファイル: test_xen.py プロジェクト: steverweber/salt

def test_get_vm():
    XenAPI = MagicMock(name="mock_session")
    XenAPI.xenapi.VM.get_by_name_label = MagicMock(return_value=["0000"],)
    XenAPI.xenapi.VM.get_is_a_template = MagicMock(return_value=False)
    with patch("salt.cloud.clouds.xen._get_session", MagicMock(return_value=XenAPI)):
        result = xen._get_vm(name="test")
        assert result == "0000"

コード例 #2

ファイル: test_xen.py プロジェクト: steverweber/salt

def test_get_vm_multiple():
    """Verify correct behavior if VM and template is returned"""
    vms = {"0000": False, "0001": True}
    XenAPI = MagicMock(name="mock_session")
    XenAPI.xenapi.VM.get_by_name_label = MagicMock(return_value=vms.keys(),)
    XenAPI.xenapi.VM.get_is_a_template = MagicMock(side_effect=lambda x: vms[x])
    with patch("salt.cloud.clouds.xen._get_session", MagicMock(return_value=XenAPI)):
        result = xen._get_vm(name="test")
        assert result == "0000"