Add tests around BaseFormatter#after_init

This commit is contained in:
Ian Cordasco 2016-06-04 12:37:56 -05:00
parent 250005b100
commit 9c2e747528
No known key found for this signature in database
GPG key ID: 656D3395E4A9791A

View file

@ -101,3 +101,17 @@ def test_write_uses_print(print_function):
mock.call(line),
mock.call(source),
]
class AfterInitFormatter(base.BaseFormatter):
"""Subclass for testing after_init."""
def after_init(self):
"""Define method to verify operation."""
self.post_initialized = True
def test_after_init_is_always_called():
"""Verify after_init is called."""
formatter = AfterInitFormatter()
assert getattr(formatter, 'post_initialized') is True