sphinx-to-github/sphinxtogithub/tests/renamer.py

import unittest
import os

import sphinxtogithub


class MockExists(object):

    def __call__(self, name):
        self.name = name
        return True

class MockRemove(MockExists):
    pass

class MockRename(object):

    def __call__(self, from_, to):
        self.from_ = from_
        self.to = to

class MockStream(object):

    def write(self, msg):
        self.msg = msg


class TestRemover(unittest.TestCase):

    def testCall(self):

        exists = MockExists()
        remove = MockRemove()
        remover = sphinxtogithub.Remover(exists, remove)

        filepath = "filepath"
        remover(filepath)

        self.assertEqual(filepath, exists.name)
        self.assertEqual(filepath, remove.name)

class TestForceRename(unittest.TestCase):

    def testCall(self):

        rename = MockRename()
        remove = MockRemove()
        renamer = sphinxtogithub.ForceRename(rename, remove)

        from_ = "from"
        to = "to"
        renamer(from_, to)

        self.assertEqual(rename.from_, from_)
        self.assertEqual(rename.to, to)
        self.assertEqual(remove.name, to)


class TestVerboseRename(unittest.TestCase):

    def testCall(self):

        rename = MockRename()
        stream = MockStream()
        renamer = sphinxtogithub.VerboseRename(rename, stream)

        from_ = os.path.join("path", "to", "from")
        to = os.path.join("path", "to", "to")
        renamer(from_, to)

        self.assertEqual(rename.from_, from_)
        self.assertEqual(rename.to, to)
        self.assertEqual(
                stream.msg,
                "Renaming directory '%s' -> '%s'\n" % (os.path.basename(from_), os.path.basename(to))
                )



def testSuite():
    suite = unittest.TestSuite()

    suite.addTest(TestRemover("testCall"))
    suite.addTest(TestForceRename("testCall"))
    suite.addTest(TestVerboseRename("testCall"))

    return suite