import os
import StringIO
import unittest

from trac.Wiki import Formatter

class WikiTestCase(unittest.TestCase):

    def __init__(self, input, correct):
        unittest.TestCase.__init__(self, 'test')
        self.input = input
        self.correct = correct
    
    def test(self):
        """Testing WikiFormatter"""
        import Href
        import Logging
        class Environment:
            def __init__(self):
                self.log = Logging.logger_factory('null')
                self.href = Href.Href('/')
                self._wiki_pages = {}
        class Cursor:
            def execute(self, *kwargs): pass
            def fetchone(self): return []
        class Connection:
            def cursor(self):
                return Cursor()

        out = StringIO.StringIO()
        Formatter(None, Environment(), Connection()).format(self.input, out)
        self.assertEquals(self.correct, out.getvalue())

def suite():
    suite = unittest.TestSuite()
    data = open(os.path.join(os.path.split(__file__)[0],
                             'wiki-tests.txt'), 'r').read()
    tests = data.split('=' * 30 + '\n')
    for test in tests:
        input, correct = test.split('-' * 30 + '\n')
        suite.addTest(WikiTestCase(input, correct))
    return suite