def test_drop_filter(self):
logger = FakeLogger()
filtered = filters.KeyFilter(logger, drop=('foo', ))
filtered.write({'foo': 'bar', 'baz': 12})
row, *_ = logger.data
self.assertIn('baz', row)
self.assertNotIn('foo', row)
def test_bad_arguments(self):
with self.assertRaises(ValueError):
filters.KeyFilter(FakeLogger())
with self.assertRaises(ValueError):
filters.KeyFilter(FakeLogger(), keep=('a', ), drop=('b', ))