def test_write(self, save_user_config): """ test writing the file to disk """ dc = DnstestConfig() conf_str = dc.to_string() mock_open = mock.mock_open() if sys.version_info[0] == 3: mock_target = 'builtins.open' else: mock_target = '__builtin__.open' with mock.patch(mock_target, mock_open, create=True): dc.write() assert mock_open.call_count == 1 fh = mock_open.return_value.__enter__.return_value assert fh.write.call_count == 1 assert fh.write.call_args == mock.call(conf_str)