def test_modify_snapshot(self):
with patch("salt.modules.snapper.snapper.SetSnapshot", MagicMock()):
_ret = {
"userdata": {
"userdata2": "uservalue2"
},
"description": "UPDATED DESCRIPTION",
"timestamp": 1457006571,
"cleanup": "number",
"user": "******",
"type": "pre",
"id": 42,
}
_opts = {
"config": "root",
"snapshot_id": 42,
"cleanup": "number",
"description": "UPDATED DESCRIPTION",
"userdata": {
"userdata2": "uservalue2"
},
}
with patch(
"salt.modules.snapper.get_snapshot",
MagicMock(
side_effect=[DBUS_RET["ListSnapshots"][0], _ret]),
):
self.assertDictEqual(snapper.modify_snapshot(**_opts), _ret)
def test_modify_snapshot(self):
with patch('salt.modules.snapper.snapper.SetSnapshot', MagicMock()):
_ret = {
'userdata': {
'userdata2': 'uservalue2'
},
'description': 'UPDATED DESCRIPTION',
'timestamp': 1457006571,
'cleanup': 'number',
'user': '******',
'type': 'pre',
'id': 42
}
_opts = {
'config': 'root',
'snapshot_id': 42,
'cleanup': 'number',
'description': 'UPDATED DESCRIPTION',
'userdata': {
'userdata2': 'uservalue2'
},
}
with patch(
'salt.modules.snapper.get_snapshot',
MagicMock(
side_effect=[DBUS_RET['ListSnapshots'][0], _ret])):
self.assertDictEqual(snapper.modify_snapshot(**_opts), _ret)
