def test_bookmark_present_new(utils_patch):
"""
Test if bookmark is present (new)
"""
ret = {
"name":
"myzpool/filesystem#mybookmark",
"result":
True,
"comment":
("myzpool/filesystem@snap bookmarked as myzpool/filesystem#mybookmark"
),
"changes": {
"myzpool/filesystem#mybookmark": "myzpool/filesystem@snap"
},
}
mock_exists = MagicMock(return_value=False)
mock_bookmark = MagicMock(return_value=OrderedDict([("bookmarked", True)]))
with patch.dict(zfs.__salt__, {"zfs.exists": mock_exists}), patch.dict(
zfs.__salt__,
{"zfs.bookmark": mock_bookmark}), patch.dict(zfs.__utils__,
utils_patch):
assert ret == zfs.bookmark_present("mybookmark",
"myzpool/filesystem@snap")
def test_bookmark_present_fail(self):
"""
Test if bookmark is present (using non existing snapshot)
"""
ret = {
"name": "myzpool/filesystem#mybookmark",
"result": False,
"comment":
"cannot bookmark snapshot 'zsalt/filesystem@snap': dataset does not exist",
"changes": {},
}
mock_exists = MagicMock(return_value=False)
mock_bookmark = MagicMock(return_value=OrderedDict([
("bookmarked", False),
(
"error",
"cannot bookmark snapshot 'zsalt/filesystem@snap': dataset does not exist",
),
]))
with patch.dict(zfs.__salt__, {"zfs.exists": mock_exists}), patch.dict(
zfs.__salt__, {"zfs.bookmark": mock_bookmark}), patch.dict(
zfs.__utils__, self.utils_patch):
self.assertEqual(
ret,
zfs.bookmark_present("mybookmark", "myzpool/filesystem@snap"))
def test_bookmark_present(utils_patch):
"""
Test if bookmark is present (bookmark already present)
"""
ret = {
"name": "myzpool/filesystem#mybookmark",
"result": True,
"comment": "bookmark is present",
"changes": {},
}
mock_exists = MagicMock(return_value=True)
with patch.dict(zfs.__salt__, {"zfs.exists": mock_exists}), patch.dict(
zfs.__utils__, utils_patch):
assert ret == zfs.bookmark_present("mybookmark",
"myzpool/filesystem@snap")
