test_filemanager.py 文件源码

def test_get_recovery_data(self, mock_yaml_safe_load, mock_isfile):
        import os
        recovery_file = os.path.join("/path/to/a/base_folder", "print_recovery_data.yaml")

        mock_isfile.return_value = True

        data = dict(path="some_path.gco",
                    origin="local",
                    pos=1234,
                    date=123456789)
        mock_yaml_safe_load.return_value = data

        with mock.patch("__builtin__.open", mock.mock_open(read_data=data), create=True) as m:
            result = self.file_manager.get_recovery_data()

            self.assertDictEqual(data, result)

            m.assert_called_with(recovery_file)

            mock_handle = m()
            mock_yaml_safe_load.assert_called_with(mock_handle)