Question

Is this the correct approach when creating a mock which should track the order in which methods are called for a given class?

class MockFoo():

    def __init__(self):
        self.calledMethods = []

    def medthod1(self):
        self.calledMethods.append(self.medthod1.__name__)

    def medthod2(self):
        self.calledMethods.append(self.medthod2.__name__)

self.assertEqual(len(foo.calledMethods),2)
self.assertEqual(foo.calledMethods[0],MockFoo.title.__name__)

Can I save something better than __name__?

Answer 1

You can certainly do this, but consider the following two approaches and decide which is the more readable:

% cat mockfoo.py
#!/usr/bin/env python

import unittest

class MockFoo(object):

    def __init__(self):
        self.called_methods = []

    def method1(self):
        self.called_methods.append(self.method1.__name__)

    def method2(self):
        self.called_methods.append('method2')


class TestMockFoo(unittest.TestCase):

    def test_list_with_name(self):
        foo = MockFoo()
        foo.method1()
        self.assertEquals(1, len(foo.called_methods))
        self.assertEquals(foo.method1.__name__, foo.called_methods[0])

    def test_list_readable(self):
        foo = MockFoo()
        foo.method2()
        self.assertEquals(1, len(foo.called_methods))
        self.assertEquals('method2', foo.called_methods[0])

if __name__ == '__main__':
    unittest.main()

% ./mockfoo.py
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

“Readability counts.”