minor test edits

This commit is contained in:
Barbara Miller 2018-02-28 12:15:12 -08:00
parent 6ce5119a48
commit e65dee57d4

View File

@ -1427,9 +1427,9 @@ def test_controller_with_defaults():
assert wwp.writer_pool.default_warc_writer.record_builder.digest_algorithm == 'sha1' assert wwp.writer_pool.default_warc_writer.record_builder.digest_algorithm == 'sha1'
class MyEarlyPlugin(warcprox.BaseStandardPostfetchProcessor): class EarlyPlugin(warcprox.BaseStandardPostfetchProcessor):
CHAIN_POSITION = 'early' CHAIN_POSITION = 'early'
def _process_put(self): def _process_url(self):
pass pass
@ -1438,7 +1438,7 @@ def test_load_plugin():
'warcprox.stats.RunningStats', 'warcprox.stats.RunningStats',
'warcprox.BaseStandardPostfetchProcessor', 'warcprox.BaseStandardPostfetchProcessor',
'warcprox.BaseBatchPostfetchProcessor', 'warcprox.BaseBatchPostfetchProcessor',
'%s.%s' % (__name__, MyEarlyPlugin.__name__),]) '%s.%s' % (__name__, EarlyPlugin.__name__),])
controller = warcprox.controller.WarcproxController(options) controller = warcprox.controller.WarcproxController(options)
assert isinstance( assert isinstance(
controller._postfetch_chain[-1], controller._postfetch_chain[-1],
@ -1459,7 +1459,7 @@ def test_load_plugin():
assert isinstance( assert isinstance(
controller._postfetch_chain[-4].listener, controller._postfetch_chain[-4].listener,
warcprox.stats.RunningStats) warcprox.stats.RunningStats)
# MyEarlyPlugin # EarlyPlugin
assert isinstance( assert isinstance(
controller._postfetch_chain[0], controller._postfetch_chain[0],
warcprox.BaseStandardPostfetchProcessor) warcprox.BaseStandardPostfetchProcessor)