[wxWidgets.git] / tests / regex / wxregex.cpp

///////////////////////////////////////////////////////////////////////////////
// Name:        tests/regex/wxregex.cpp
// Purpose:     Test wxRegEx
// Author:      Vadim Zeitlin, Mike Wetherell
// RCS-ID:      $Id$
// Copyright:   Vadim Zeitlin, Mike Wetherell
// Licence:     wxWidgets licence
///////////////////////////////////////////////////////////////////////////////

#include "wx/wxprec.h"

#ifdef __BORLANDC__
#   pragma hdrstop
#endif

#ifndef WX_PRECOMP
#   include "wx/wx.h"
#endif

#if wxUSE_REGEX

#include "wx/regex.h"
#include "wx/cppunit.h"
#include "wx/tokenzr.h"
#include <string>

using namespace std;
using namespace CppUnit;


///////////////////////////////////////////////////////////////////////////////
// Compile Test

class RegExCompileTestCase : public TestCase
{
public:
    RegExCompileTestCase(const char *name, const wxString& pattern,
                         bool correct, int flags)
    :   TestCase(name),
        m_pattern(pattern),
        m_correct(correct),
        m_flags(flags)
    { }

protected:
    void runTest();

private:
    wxString m_pattern;
    bool m_correct;
    int m_flags;
};

void RegExCompileTestCase::runTest()
{
    wxRegEx re;
    bool ok = re.Compile(m_pattern, m_flags);

    if (m_correct)
        CPPUNIT_ASSERT_MESSAGE("compile failed", ok);
    else
        CPPUNIT_ASSERT_MESSAGE("compile succeeded (should fail)", !ok);
}


///////////////////////////////////////////////////////////////////////////////
// Match Test

class RegExMatchTestCase : public TestCase
{
public:
    RegExMatchTestCase(const char *name, const wxString& pattern,
                       const wxString& text, const char *expected,
                       int flags)
    :   TestCase(name),
        m_pattern(pattern),
        m_text(text),
        m_expected(expected),
        m_flags(flags)
    { }

protected:
    void runTest();

private:
    wxString m_pattern;
    wxString m_text;
    const char *m_expected;
    int m_flags;
};

void RegExMatchTestCase::runTest()
{
    int compileFlags = m_flags & ~(wxRE_NOTBOL | wxRE_NOTEOL);
    int matchFlags = m_flags & (wxRE_NOTBOL | wxRE_NOTEOL);

    wxRegEx re(m_pattern, compileFlags);
    CPPUNIT_ASSERT_MESSAGE("compile failed", re.IsValid());

    bool ok = re.Matches(m_text, matchFlags);

    if (m_expected) {
        CPPUNIT_ASSERT_MESSAGE("match failed", ok);

        wxStringTokenizer tkz(wxString(m_expected, *wxConvCurrent),
                              _T("\t"), wxTOKEN_RET_EMPTY);
        size_t i;

        for (i = 0; i < re.GetMatchCount() && tkz.HasMoreTokens(); i++) {
            wxString expected = tkz.GetNextToken();
            wxString result = re.GetMatch(m_text, i);

            wxString msgstr;
            msgstr.Printf(_T("\\%d == '%s' (expected '%s')"),
                          (int)i, result.c_str(), expected.c_str());
            const char *msg = msgstr.mb_str();

            CPPUNIT_ASSERT_MESSAGE(msg, result == expected);
        }

        if ((m_flags & wxRE_NOSUB) == 0)
            CPPUNIT_ASSERT(re.GetMatchCount() == i);
    }
    else {
        CPPUNIT_ASSERT_MESSAGE("match succeeded (should fail)", !ok);
    }
}


///////////////////////////////////////////////////////////////////////////////
// Replacement Test

class RegExReplaceTestCase : public TestCase
{
public:
    RegExReplaceTestCase(const char *name, const wxString& pattern,
                         const wxString& text, const wxString& repl,
                         const wxString& expected, size_t count, int flags)
    :   TestCase(name),
        m_pattern(pattern),
        m_text(text),
        m_repl(repl),
        m_expected(expected),
        m_count(count),
        m_flags(flags)
    { }    

protected:
    void runTest();

private:
    wxString m_pattern;
    wxString m_text;
    wxString m_repl;
    wxString m_expected;
    size_t m_count;
    int m_flags;
};

void RegExReplaceTestCase::runTest()
{
    wxRegEx re(m_pattern, m_flags);

    wxString text(m_text);
    size_t nRepl = re.Replace(&text, m_repl);

    wxString msgstr;
    msgstr.Printf(_T("returns '%s' (expected '%s')"), text.c_str(), m_expected.c_str());
    const char *msg = msgstr.mb_str();
    CPPUNIT_ASSERT_MESSAGE(msg, text == m_expected);

    msgstr.Printf(_T("matches %d times (expected %d)"), nRepl, m_count);
    msg = msgstr.mb_str();
    CPPUNIT_ASSERT_MESSAGE(msg, nRepl == m_count);
}


///////////////////////////////////////////////////////////////////////////////
// The suite

class wxRegExTestSuite : public TestSuite
{
public:
    wxRegExTestSuite() : TestSuite("wxRegExTestSuite") { }
    static Test *suite();

private:
    void add(const char *pattern, bool correct, int flags = wxRE_DEFAULT);
    void add(const char *pattern, const char *text,
             const char *expected = NULL, int flags = wxRE_DEFAULT);
    void add(const char *pattern, const char *text, const char *replacement,
             const char *expected, size_t count, int flags = wxRE_DEFAULT);

    static wxString FlagStr(int flags);
    static wxString Conv(const char *str) { return wxString(str, *wxConvCurrent); }
};

// Build the suite (static)
//
Test *wxRegExTestSuite::suite()
{
    wxRegExTestSuite *suite = new wxRegExTestSuite;

    // Compile tests
    // pattern, expected result
    suite->add("foo", true);
    suite->add("foo(", false);
    suite->add("foo(bar", false);
    suite->add("foo(bar)", true);
    suite->add("foo[", false);
    suite->add("foo[bar", false);
    suite->add("foo[bar]", true);
    suite->add("foo{1", false);
    suite->add("foo{1}", true);
    suite->add("foo{1,2}", true);
    suite->add("foo*", true);
    suite->add("foo+", true);
    suite->add("foo?", true);

    // Match tests
    // pattern, text, expected results (match, followed by submatches
    // tab separated, or NULL for no match expected)
    suite->add("foo", "bar");
    suite->add("foo", "foobar", "foo");
    suite->add("^foo", "foobar", "foo");
    suite->add("^foo", "barfoo");
    suite->add("bar$", "barbar", "bar");
    suite->add("bar$", "barbar ");
    suite->add("OoBa", "FoObAr", "oObA", wxRE_ICASE);
    suite->add("^[A-Z].*$", "AA\nbb\nCC", "AA\nbb\nCC");
    suite->add("^[A-Z].*$", "AA\nbb\nCC", "AA", wxRE_NEWLINE);
    suite->add("^[a-z].*$", "AA\nbb\nCC", "bb", wxRE_NEWLINE);
    suite->add("^[A-Z].*$", "AA\nbb\nCC", "CC", wxRE_NEWLINE | wxRE_NOTBOL);
    suite->add("^[A-Z].*$", "AA\nbb\nCC", NULL, wxRE_NEWLINE | wxRE_NOTBOL | wxRE_NOTEOL);
    suite->add("([[:alpha:]]+) ([[:alpha:]]+) ([[:digit:]]+).* ([[:digit:]]+)$",
        "Fri Jul 13 18:37:52 CEST 2001",
        "Fri Jul 13 18:37:52 CEST 2001\tFri\tJul\t13\t2001");

    // Replace tests
    // pattern, text, replacement, expected result and number of matches
    const char *patn = "([a-z]+)[^0-9]*([0-9]+)";
    suite->add(patn, "foo123", "bar", "bar", 1);
    suite->add(patn, "foo123", "\\2\\1", "123foo", 1);
    suite->add(patn, "foo_123", "\\2\\1", "123foo", 1);
    suite->add(patn, "123foo", "bar", "123foo", 0);
    suite->add(patn, "123foo456foo", "&&", "123foo456foo456foo", 1);
    suite->add(patn, "123foo456foo", "\\0\\0", "123foo456foo456foo", 1);
    suite->add(patn, "foo123foo123", "bar", "barbar", 2);
    suite->add(patn, "foo123_foo456_foo789", "bar", "bar_bar_bar", 3);

    return suite;
}

// Add a compile test
//
void wxRegExTestSuite::add(
    const char *pattern,
    bool correct,
    int flags /*=wxRE_DEFAULT*/)
{
    addTest(new RegExCompileTestCase(
                (_T("/") + Conv(pattern) + _T("/") + FlagStr(flags)).mb_str(),
                Conv(pattern), correct, flags));
}

// Add a match test
//
void wxRegExTestSuite::add(
    const char *pattern,
    const char *text,
    const char *expected /*=NULL*/,
    int flags /*=wxRE_DEFAULT*/)
{
    wxString name;

    name << _T("'") << Conv(text) << _T("' =~ /") << Conv(pattern) << _T("/")
         << FlagStr(flags);
    name.Replace(_T("\n"), _T("\\n"));

    addTest(new RegExMatchTestCase(name.mb_str(), Conv(pattern),
                                   Conv(text), expected, flags));
}

// Add a replace test
//
void wxRegExTestSuite::add(
    const char *pattern,
    const char *text,
    const char *replacement,
    const char *expected,
    size_t count,
    int flags /*=wxRE_DEFAULT*/)
{
    wxString name;

    name << _T("'") << Conv(text) << _T("' =~ s/") << Conv(pattern) << _T("/")
         << Conv(replacement) << _T("/g") << FlagStr(flags);
    name.Replace(_T("\n"), _T("\\n"));

    addTest(new RegExReplaceTestCase(
                    name.mb_str(), Conv(pattern), Conv(text),
                    Conv(replacement), Conv(expected), count, flags));
}

// Display string for the flags
//
wxString wxRegExTestSuite::FlagStr(int flags)
{
    wxString str;

    if (!flags)
        return str;

    for (int i = 0; (unsigned)flags >> i; i++) {
        switch (flags & (1 << i)) {
            case 0: break;
#ifdef wxHAS_REGEX_ADVANCED
            case wxRE_ADVANCED: str += _T(" | wxRE_ADVANCED"); break;
#endif
            case wxRE_BASIC:    str += _T(" | wxRE_BASIC"); break;
            case wxRE_ICASE:    str += _T(" | wxRE_ICASE"); break;
            case wxRE_NOSUB:    str += _T(" | wxRE_NOSUB"); break;
            case wxRE_NEWLINE:  str += _T(" | wxRE_NEWLINE"); break;
            case wxRE_NOTBOL:   str += _T(" | wxRE_NOTBOL"); break;
            case wxRE_NOTEOL:   str += _T(" | wxRE_NOTEOL"); break;
            default: wxFAIL; break;
        }
    }

    return _T(" (") + str.Mid(3) + _T(")");
};

// register in the unnamed registry so that these tests are run by default
CPPUNIT_TEST_SUITE_REGISTRATION(wxRegExTestSuite);

// also include in it's own registry so that these tests can be run alone
CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(wxRegExTestSuite, "wxRegExTestSuite");


#endif // wxUSE_REGEX
Commit	Line	Data
bc10103e VS	1	///////////////////////////////////////////////////////////////////////////////
	2	// Name: tests/regex/wxregex.cpp
	3	// Purpose: Test wxRegEx
	4	// Author: Vadim Zeitlin, Mike Wetherell
	5	// RCS-ID: $Id$
	6	// Copyright: Vadim Zeitlin, Mike Wetherell
	7	// Licence: wxWidgets licence
	8	///////////////////////////////////////////////////////////////////////////////
	9
	10	#include "wx/wxprec.h"
	11
	12	#ifdef __BORLANDC__
	13	# pragma hdrstop
	14	#endif
	15
	16	#ifndef WX_PRECOMP
	17	# include "wx/wx.h"
	18	#endif
	19
	20	#if wxUSE_REGEX
	21
	22	#include "wx/regex.h"
	23	#include "wx/cppunit.h"
	24	#include "wx/tokenzr.h"
	25	#include <string>
	26
	27	using namespace std;
	28	using namespace CppUnit;
	29
	30
	31	///////////////////////////////////////////////////////////////////////////////
	32	// Compile Test
	33
	34	class RegExCompileTestCase : public TestCase
	35	{
	36	public:
	37	RegExCompileTestCase(const char *name, const wxString& pattern,
	38	bool correct, int flags)
	39	: TestCase(name),
	40	m_pattern(pattern),
	41	m_correct(correct),
	42	m_flags(flags)
	43	{ }
	44
	45	protected:
	46	void runTest();
	47
	48	private:
	49	wxString m_pattern;
	50	bool m_correct;
	51	int m_flags;
	52	};
	53
	54	void RegExCompileTestCase::runTest()
	55	{
	56	wxRegEx re;
	57	bool ok = re.Compile(m_pattern, m_flags);
	58
	59	if (m_correct)
	60	CPPUNIT_ASSERT_MESSAGE("compile failed", ok);
	61	else
	62	CPPUNIT_ASSERT_MESSAGE("compile succeeded (should fail)", !ok);
	63	}
	64
65
66	///////////////////////////////////////////////////////////////////////////////
67	// Match Test
68
69	class RegExMatchTestCase : public TestCase
70	{
71	public:
72	RegExMatchTestCase(const char *name, const wxString& pattern,
73	const wxString& text, const char *expected,
74	int flags)
75	: TestCase(name),
76	m_pattern(pattern),
77	m_text(text),
78	m_expected(expected),
79	m_flags(flags)
80	{ }
81
82	protected:
83	void runTest();
84
85	private:
86	wxString m_pattern;
87	wxString m_text;
88	const char *m_expected;
89	int m_flags;
90	};
91
92	void RegExMatchTestCase::runTest()
93	{
94	int compileFlags = m_flags & ~(wxRE_NOTBOL \| wxRE_NOTEOL);
95	int matchFlags = m_flags & (wxRE_NOTBOL \| wxRE_NOTEOL);
96
97	wxRegEx re(m_pattern, compileFlags);
98	CPPUNIT_ASSERT_MESSAGE("compile failed", re.IsValid());
99
100	bool ok = re.Matches(m_text, matchFlags);
101
102	if (m_expected) {
103	CPPUNIT_ASSERT_MESSAGE("match failed", ok);
104
105	wxStringTokenizer tkz(wxString(m_expected, *wxConvCurrent),
106	_T("\t"), wxTOKEN_RET_EMPTY);
107	size_t i;
108
109	for (i = 0; i < re.GetMatchCount() && tkz.HasMoreTokens(); i++) {
110	wxString expected = tkz.GetNextToken();
111	wxString result = re.GetMatch(m_text, i);
112
113	wxString msgstr;
114	msgstr.Printf(_T("\\%d == '%s' (expected '%s')"),
115	(int)i, result.c_str(), expected.c_str());
116	const char *msg = msgstr.mb_str();
117
118	CPPUNIT_ASSERT_MESSAGE(msg, result == expected);
119	}
120
121	if ((m_flags & wxRE_NOSUB) == 0)
122	CPPUNIT_ASSERT(re.GetMatchCount() == i);
123	}
124	else {
125	CPPUNIT_ASSERT_MESSAGE("match succeeded (should fail)", !ok);
126	}
127	}
128
129
130	///////////////////////////////////////////////////////////////////////////////
131	// Replacement Test
132
133	class RegExReplaceTestCase : public TestCase
134	{
135	public:
136	RegExReplaceTestCase(const char *name, const wxString& pattern,
137	const wxString& text, const wxString& repl,
138	const wxString& expected, size_t count, int flags)
139	: TestCase(name),
140	m_pattern(pattern),
141	m_text(text),
142	m_repl(repl),
143	m_expected(expected),
144	m_count(count),
145	m_flags(flags)
146	{ }
147
148	protected:
149	void runTest();
150
151	private:
152	wxString m_pattern;
153	wxString m_text;
154	wxString m_repl;
155	wxString m_expected;
156	size_t m_count;
157	int m_flags;
158	};
159
160	void RegExReplaceTestCase::runTest()
161	{
162	wxRegEx re(m_pattern, m_flags);
163
164	wxString text(m_text);
165	size_t nRepl = re.Replace(&text, m_repl);
166
167	wxString msgstr;
168	msgstr.Printf(_T("returns '%s' (expected '%s')"), text.c_str(), m_expected.c_str());
169	const char *msg = msgstr.mb_str();
170	CPPUNIT_ASSERT_MESSAGE(msg, text == m_expected);
171
172	msgstr.Printf(_T("matches %d times (expected %d)"), nRepl, m_count);
173	msg = msgstr.mb_str();
174	CPPUNIT_ASSERT_MESSAGE(msg, nRepl == m_count);
175	}
176
177
178	///////////////////////////////////////////////////////////////////////////////
179	// The suite
180
181	class wxRegExTestSuite : public TestSuite
182	{
183	public:
184	wxRegExTestSuite() : TestSuite("wxRegExTestSuite") { }
185	static Test *suite();
186
187	private:
188	void add(const char *pattern, bool correct, int flags = wxRE_DEFAULT);
189	void add(const char pattern, const char text,
190	const char *expected = NULL, int flags = wxRE_DEFAULT);
191	void add(const char pattern, const char text, const char *replacement,
192	const char *expected, size_t count, int flags = wxRE_DEFAULT);
193
194	static wxString FlagStr(int flags);
195	static wxString Conv(const char str) { return wxString(str, wxConvCurrent); }
196	};
197
198	// Build the suite (static)
199	//
200	Test *wxRegExTestSuite::suite()
201	{
202	wxRegExTestSuite *suite = new wxRegExTestSuite;
203
204	// Compile tests
205	// pattern, expected result
206	suite->add("foo", true);
207	suite->add("foo(", false);
208	suite->add("foo(bar", false);
209	suite->add("foo(bar)", true);
210	suite->add("foo[", false);
211	suite->add("foo[bar", false);
212	suite->add("foo[bar]", true);
213	suite->add("foo{1", false);
214	suite->add("foo{1}", true);
215	suite->add("foo{1,2}", true);
216	suite->add("foo*", true);
217	suite->add("foo+", true);
218	suite->add("foo?", true);
219
220	// Match tests
221	// pattern, text, expected results (match, followed by submatches
222	// tab separated, or NULL for no match expected)
223	suite->add("foo", "bar");
224	suite->add("foo", "foobar", "foo");
225	suite->add("^foo", "foobar", "foo");
226	suite->add("^foo", "barfoo");
227	suite->add("bar$", "barbar", "bar");
228	suite->add("bar$", "barbar ");
229	suite->add("OoBa", "FoObAr", "oObA", wxRE_ICASE);
230	suite->add("^[A-Z].*$", "AA\nbb\nCC", "AA\nbb\nCC");
231	suite->add("^[A-Z].*$", "AA\nbb\nCC", "AA", wxRE_NEWLINE);
232	suite->add("^[a-z].*$", "AA\nbb\nCC", "bb", wxRE_NEWLINE);
233	suite->add("^[A-Z].*$", "AA\nbb\nCC", "CC", wxRE_NEWLINE \| wxRE_NOTBOL);
234	suite->add("^[A-Z].*$", "AA\nbb\nCC", NULL, wxRE_NEWLINE \| wxRE_NOTBOL \| wxRE_NOTEOL);
235	suite->add("([[:alpha:]]+) ([[:alpha:]]+) ([[:digit:]]+).* ([[:digit:]]+)$",
236	"Fri Jul 13 18:37:52 CEST 2001",
237	"Fri Jul 13 18:37:52 CEST 2001\tFri\tJul\t13\t2001");
238
239	// Replace tests
240	// pattern, text, replacement, expected result and number of matches
241	const char patn = "([a-z]+)[^0-9]([0-9]+)";
242	suite->add(patn, "foo123", "bar", "bar", 1);
243	suite->add(patn, "foo123", "\\2\\1", "123foo", 1);
244	suite->add(patn, "foo_123", "\\2\\1", "123foo", 1);
245	suite->add(patn, "123foo", "bar", "123foo", 0);
246	suite->add(patn, "123foo456foo", "&&", "123foo456foo456foo", 1);
247	suite->add(patn, "123foo456foo", "\\0\\0", "123foo456foo456foo", 1);
248	suite->add(patn, "foo123foo123", "bar", "barbar", 2);
249	suite->add(patn, "foo123_foo456_foo789", "bar", "bar_bar_bar", 3);
250
251	return suite;
252	}
253
254	// Add a compile test
255	//
256	void wxRegExTestSuite::add(
257	const char *pattern,
258	bool correct,
259	int flags /=wxRE_DEFAULT/)
260	{
261	addTest(new RegExCompileTestCase(
262	(_T("/") + Conv(pattern) + _T("/") + FlagStr(flags)).mb_str(),
263	Conv(pattern), correct, flags));
264	}
265
266	// Add a match test
267	//
268	void wxRegExTestSuite::add(
269	const char *pattern,
270	const char *text,
271	const char expected /=NULL*/,
272	int flags /=wxRE_DEFAULT/)
273	{
274	wxString name;
275
276	name << _T("'") << Conv(text) << _T("' =~ /") << Conv(pattern) << _T("/")
277	<< FlagStr(flags);
278	name.Replace(_T("\n"), _T("\\n"));
279
280	addTest(new RegExMatchTestCase(name.mb_str(), Conv(pattern),
281	Conv(text), expected, flags));
282	}
283
284	// Add a replace test
285	//
286	void wxRegExTestSuite::add(
287	const char *pattern,
288	const char *text,
289	const char *replacement,
290	const char *expected,
291	size_t count,
292	int flags /=wxRE_DEFAULT/)
293	{
294	wxString name;
295
296	name << _T("'") << Conv(text) << _T("' =~ s/") << Conv(pattern) << _T("/")
297	<< Conv(replacement) << _T("/g") << FlagStr(flags);
298	name.Replace(_T("\n"), _T("\\n"));
299
300	addTest(new RegExReplaceTestCase(
301	name.mb_str(), Conv(pattern), Conv(text),
302	Conv(replacement), Conv(expected), count, flags));
303	}
304
305	// Display string for the flags
306	//
307	wxString wxRegExTestSuite::FlagStr(int flags)
308	{
309	wxString str;
310
311	if (!flags)
312	return str;
313
314	for (int i = 0; (unsigned)flags >> i; i++) {
315	switch (flags & (1 << i)) {
316	case 0: break;
317	#ifdef wxHAS_REGEX_ADVANCED
318	case wxRE_ADVANCED: str += _T(" \| wxRE_ADVANCED"); break;
319	#endif
320	case wxRE_BASIC: str += _T(" \| wxRE_BASIC"); break;
321	case wxRE_ICASE: str += _T(" \| wxRE_ICASE"); break;
322	case wxRE_NOSUB: str += _T(" \| wxRE_NOSUB"); break;
323	case wxRE_NEWLINE: str += _T(" \| wxRE_NEWLINE"); break;
324	case wxRE_NOTBOL: str += _T(" \| wxRE_NOTBOL"); break;
325	case wxRE_NOTEOL: str += _T(" \| wxRE_NOTEOL"); break;
326	default: wxFAIL; break;
327	}
328	}
329
330	return _T(" (") + str.Mid(3) + _T(")");
331	};
332
333	// register in the unnamed registry so that these tests are run by default
334	CPPUNIT_TEST_SUITE_REGISTRATION(wxRegExTestSuite);
335
336	// also include in it's own registry so that these tests can be run alone
337	CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(wxRegExTestSuite, "wxRegExTestSuite");
338
339
340	#endif // wxUSE_REGEX