diff options
Diffstat (limited to 'cliff/tests/test_lister.py')
-rw-r--r-- | cliff/tests/test_lister.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/cliff/tests/test_lister.py b/cliff/tests/test_lister.py index c0452d5..7d1876f 100644 --- a/cliff/tests/test_lister.py +++ b/cliff/tests/test_lister.py @@ -1,5 +1,7 @@ #!/usr/bin/env python +import weakref + from cliff.lister import Lister import mock @@ -9,6 +11,7 @@ class FauxFormatter(object): def __init__(self): self.args = [] + self.obj = weakref.proxy(self) def emit_list(self, columns, data, stdout, args): self.args.append((columns, data)) @@ -16,8 +19,8 @@ class FauxFormatter(object): class ExerciseLister(Lister): - def load_formatter_plugins(self): - self.formatters = { + def _load_formatter_plugins(self): + return { 'test': FauxFormatter(), } return |