diff options
| author | Bryan Newbold <bnewbold@archive.org> | 2018-04-05 19:24:07 -0700 |
|---|---|---|
| committer | Bryan Newbold <bnewbold@archive.org> | 2018-04-05 19:24:07 -0700 |
| commit | e68d43e2369eed7ddf288be8c8f2edd0a85974e1 (patch) | |
| tree | 204c2dd9e2933c848affe76f6c56aaf2b032d4d4 /mapreduce/tests/test_backfill_hbase_from_cdx.py | |
| parent | 5db075beaa55b2d619798154c06c2df625346972 (diff) | |
| download | sandcrawler-e68d43e2369eed7ddf288be8c8f2edd0a85974e1.tar.gz sandcrawler-e68d43e2369eed7ddf288be8c8f2edd0a85974e1.zip | |
make happybase mock injection slightly less horrible
Diffstat (limited to 'mapreduce/tests/test_backfill_hbase_from_cdx.py')
| -rw-r--r-- | mapreduce/tests/test_backfill_hbase_from_cdx.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/mapreduce/tests/test_backfill_hbase_from_cdx.py b/mapreduce/tests/test_backfill_hbase_from_cdx.py index 2dbbc25..070662b 100644 --- a/mapreduce/tests/test_backfill_hbase_from_cdx.py +++ b/mapreduce/tests/test_backfill_hbase_from_cdx.py @@ -15,12 +15,13 @@ def job(): Note: this mock only seems to work with job.run_mapper(), not job.run(); the later results in a separate instantiation without the mock? """ + job = MRCDXBackfillHBase(['--no-conf', '-']) + conn = happybase_mock.Connection() conn.create_table('wbgrp-journal-extract-test', {'file': {}, 'grobid0': {}, 'f': {}}) - table = conn.table('wbgrp-journal-extract-test') + job.hb_table = conn.table('wbgrp-journal-extract-test') - job = MRCDXBackfillHBase(['--no-conf', '-'], hb_table=table) return job |