class SnapperTestCase(TestCase): def setUp(self): self.dbus_mock = MagicMock() self.DBusExceptionMock = MagicMock() # pylint: disable=invalid-name self.dbus_mock.configure_mock(DBusException=self.DBusExceptionMock) snapper.dbus = self.dbus_mock snapper.snapper = MagicMock() def test__snapshot_to_data(self): data = snapper._snapshot_to_data(DBUS_RET['ListSnapshots'][0]) # pylint: disable=protected-access self.assertEqual(data['id'], 42) self.assertNotIn('pre', data) self.assertEqual(data['type'], 'pre') self.assertEqual(data['user'], 'root') self.assertEqual(data['timestamp'], 1457006571) self.assertEqual(data['description'], 'Some description') self.assertEqual(data['cleanup'], '') self.assertEqual(data['userdata']['userdata1'], 'userval1') @patch('salt.modules.snapper.snapper.ListSnapshots', MagicMock(return_value=DBUS_RET['ListSnapshots'])) def test_list_snapshots(self): self.assertEqual(snapper.list_snapshots(), MODULE_RET["SNAPSHOTS"]) @patch('salt.modules.snapper.snapper.GetSnapshot', MagicMock(return_value=DBUS_RET['ListSnapshots'][0])) def test_get_snapshot(self): self.assertEqual(snapper.get_snapshot(), MODULE_RET["SNAPSHOTS"][0]) self.assertEqual(snapper.get_snapshot(number=42), MODULE_RET["SNAPSHOTS"][0]) self.assertNotEqual(snapper.get_snapshot(number=42), MODULE_RET["SNAPSHOTS"][1]) @patch('salt.modules.snapper.snapper.ListConfigs', MagicMock(return_value=DBUS_RET['ListConfigs'])) def test_list_configs(self): self.assertEqual(snapper.list_configs(), MODULE_RET["LISTCONFIGS"]) @patch('salt.modules.snapper.snapper.GetConfig', MagicMock(return_value=DBUS_RET['ListConfigs'][0])) def test_get_config(self): self.assertEqual(snapper.get_config(), DBUS_RET["ListConfigs"][0]) @patch('salt.modules.snapper.snapper.SetConfig', MagicMock()) def test_set_config(self): opts = {'sync_acl': True, 'dummy': False, 'foobar': 1234} self.assertEqual(snapper.set_config(opts), True) def test_status_to_string(self): self.assertEqual(snapper.status_to_string(1), ["created"]) self.assertEqual(snapper.status_to_string(2), ["deleted"]) self.assertEqual(snapper.status_to_string(4), ["type changed"]) self.assertEqual(snapper.status_to_string(8), ["modified"]) self.assertEqual(snapper.status_to_string(16), ["permission changed"]) self.assertListEqual(snapper.status_to_string(24), ["modified", "permission changed"]) self.assertEqual(snapper.status_to_string(32), ["owner changed"]) self.assertEqual(snapper.status_to_string(64), ["group changed"]) self.assertListEqual(snapper.status_to_string(97), ["created", "owner changed", "group changed"]) self.assertEqual(snapper.status_to_string(128), ["extended attributes changed"]) self.assertEqual(snapper.status_to_string(256), ["ACL info changed"]) @patch('salt.modules.snapper.snapper.CreateSingleSnapshot', MagicMock(return_value=1234)) @patch('salt.modules.snapper.snapper.CreatePreSnapshot', MagicMock(return_value=1234)) @patch('salt.modules.snapper.snapper.CreatePostSnapshot', MagicMock(return_value=1234)) def test_create_snapshot(self): for snapshot_type in ['pre', 'post', 'single']: opts = { '__pub_jid': 20160607130930720112, 'type': snapshot_type, 'description': 'Test description', 'cleanup_algorithm': 'number', 'pre_number': 23, } self.assertEqual(snapper.create_snapshot(**opts), 1234) @patch('salt.modules.snapper._get_last_snapshot', MagicMock(return_value={'id': 42})) def test__get_num_interval(self): self.assertEqual(snapper._get_num_interval(config=None, num_pre=None, num_post=None), (42, 0)) # pylint: disable=protected-access self.assertEqual(snapper._get_num_interval(config=None, num_pre=None, num_post=50), (42, 50)) # pylint: disable=protected-access self.assertEqual(snapper._get_num_interval(config=None, num_pre=42, num_post=50), (42, 50)) # pylint: disable=protected-access def test_run(self): patch_dict = { 'snapper.create_snapshot': MagicMock(return_value=43), 'test.ping': MagicMock(return_value=True), } with patch.dict(snapper.__salt__, patch_dict): self.assertEqual(snapper.run("test.ping"), True) self.assertRaises(CommandExecutionError, snapper.run, "unknown.func") @patch('salt.modules.snapper._get_num_interval', MagicMock(return_value=(42, 43))) @patch('salt.modules.snapper.snapper.GetComparison', MagicMock()) @patch('salt.modules.snapper.snapper.GetFiles', MagicMock(return_value=DBUS_RET['GetFiles'])) def test_status(self): self.assertItemsEqual(snapper.status(), MODULE_RET['GETFILES']) self.assertItemsEqual(snapper.status(num_pre="42", num_post=43), MODULE_RET['GETFILES']) self.assertItemsEqual(snapper.status(num_pre=42), MODULE_RET['GETFILES']) self.assertItemsEqual(snapper.status(num_post=43), MODULE_RET['GETFILES']) @patch('salt.modules.snapper.status', MagicMock(return_value=MODULE_RET['GETFILES'])) def test_changed_files(self): self.assertEqual(snapper.changed_files(), MODULE_RET['GETFILES'].keys()) @patch('salt.modules.snapper._get_num_interval', MagicMock(return_value=(42, 43))) @patch('salt.modules.snapper.status', MagicMock(return_value=MODULE_RET['GETFILES'])) def test_undo(self): cmd_ret = 'create:0 modify:1 delete:0' with patch.dict(snapper.__salt__, {'cmd.run': MagicMock(return_value=cmd_ret)}): module_ret = {'create': '0', 'delete': '0', 'modify': '1'} self.assertEqual(snapper.undo(files=['/tmp/foo']), module_ret) cmd_ret = 'create:1 modify:1 delete:0' with patch.dict(snapper.__salt__, {'cmd.run': MagicMock(return_value=cmd_ret)}): module_ret = {'create': '1', 'delete': '0', 'modify': '1'} self.assertEqual(snapper.undo(files=['/tmp/foo', '/tmp/foo2']), module_ret) cmd_ret = 'create:1 modify:1 delete:1' with patch.dict(snapper.__salt__, {'cmd.run': MagicMock(return_value=cmd_ret)}): module_ret = {'create': '1', 'delete': '1', 'modify': '1'} self.assertEqual(snapper.undo(files=['/tmp/foo', '/tmp/foo2', '/tmp/foo3']), module_ret) @patch('salt.modules.snapper.list_snapshots', MagicMock(return_value=MODULE_RET['SNAPSHOTS'])) def test__get_jid_snapshots(self): self.assertEqual( snapper._get_jid_snapshots("20160607130930720112"), # pylint: disable=protected-access (MODULE_RET['SNAPSHOTS'][0]['id'], MODULE_RET['SNAPSHOTS'][1]['id']) ) @patch('salt.modules.snapper._get_jid_snapshots', MagicMock(return_value=(42, 43))) @patch('salt.modules.snapper.undo', MagicMock(return_value='create:1 modify:1 delete:1')) def test_undo_jid(self): self.assertEqual(snapper.undo_jid(20160607130930720112), 'create:1 modify:1 delete:1') @patch('salt.modules.snapper._get_num_interval', MagicMock(return_value=(42, 43))) @patch('salt.modules.snapper.snapper.MountSnapshot', MagicMock(side_effect=["/.snapshots/55/snapshot", ""])) @patch('salt.modules.snapper.snapper.UmountSnapshot', MagicMock(return_value="")) @patch('os.path.isdir', MagicMock(return_value=False)) @patch('salt.modules.snapper.changed_files', MagicMock(return_value=["/tmp/foo2"])) @patch('salt.modules.snapper._is_text_file', MagicMock(return_value=True)) @patch('os.path.isfile', MagicMock(side_effect=[False, True])) @patch('salt.utils.fopen', mock_open(read_data=FILE_CONTENT["/tmp/foo2"]['post'])) def test_diff_text_file(self): self.assertEqual(snapper.diff(), {"/tmp/foo2": MODULE_RET['DIFF']['/tmp/foo2']}) @patch('salt.modules.snapper._get_num_interval', MagicMock(return_value=(55, 0))) @patch('salt.modules.snapper.snapper.MountSnapshot', MagicMock( side_effect=["/.snapshots/55/snapshot", "", "/.snapshots/55/snapshot", ""])) @patch('salt.modules.snapper.snapper.UmountSnapshot', MagicMock(return_value="")) @patch('salt.modules.snapper.changed_files', MagicMock(return_value=["/tmp/foo", "/tmp/foo2"])) @patch('salt.modules.snapper._is_text_file', MagicMock(return_value=True)) @patch('os.path.isfile', MagicMock(side_effect=[True, True, False, True])) @patch('os.path.isdir', MagicMock(return_value=False)) def test_diff_text_files(self): fopen_effect = [ mock_open(read_data=FILE_CONTENT["/tmp/foo"]['pre']).return_value, mock_open(read_data=FILE_CONTENT["/tmp/foo"]['post']).return_value, mock_open(read_data=FILE_CONTENT["/tmp/foo2"]['post']).return_value, ] with patch('salt.utils.fopen') as fopen_mock: fopen_mock.side_effect = fopen_effect module_ret = { "/tmp/foo": MODULE_RET['DIFF']["/tmp/foo"], "/tmp/foo2": MODULE_RET['DIFF']["/tmp/foo2"], } self.assertEqual(snapper.diff(), module_ret) @patch('salt.modules.snapper._get_num_interval', MagicMock(return_value=(55, 0))) @patch('salt.modules.snapper.snapper.MountSnapshot', MagicMock( side_effect=["/.snapshots/55/snapshot", "", "/.snapshots/55/snapshot", ""])) @patch('salt.modules.snapper.snapper.UmountSnapshot', MagicMock(return_value="")) @patch('salt.modules.snapper.changed_files', MagicMock(return_value=["/tmp/foo3"])) @patch('salt.modules.snapper._is_text_file', MagicMock(return_value=False)) @patch('os.path.isfile', MagicMock(side_effect=[True, True])) @patch('os.path.isdir', MagicMock(return_value=False)) @patch.dict(snapper.__salt__, { 'hashutil.sha256_digest': MagicMock(side_effect=[ "e61f8b762d83f3b4aeb3689564b0ffbe54fa731a69a1e208dc9440ce0f69d19b", "f18f971f1517449208a66589085ddd3723f7f6cefb56c141e3d97ae49e1d87fa", ]) }) def test_diff_binary_files(self): fopen_effect = [ mock_open(read_data="dummy binary").return_value, mock_open(read_data="dummy binary").return_value, ] with patch('salt.utils.fopen') as fopen_mock: fopen_mock.side_effect = fopen_effect module_ret = { "/tmp/foo3": MODULE_RET['DIFF']["/tmp/foo3"], } self.assertEqual(snapper.diff(), module_ret)