test_pgantomizer.py 文件源码

def test_command_line_invokation(original_db, anonymized, monkeypatch):
    monkeypatch.setattr('argparse.ArgumentParser.parse_args', lambda self: SimpleNamespace(
        verbose=False,
        schema=SCHEMA_PATH,
        dump_file=DUMP_PATH,
        **{arg: ORIGINAL_DB_ARGS[arg] for arg in ('dbname', 'user', 'host', 'port', 'password')}
    ))
    dump_main()

    assert os.path.getsize(DUMP_PATH) > 2000
    assert_db_empty(anonymized)
    monkeypatch.setattr('argparse.ArgumentParser.parse_args', lambda self: SimpleNamespace(
        verbose=False,
        leave_dump=False,
        schema=SCHEMA_PATH,
        dump_file=DUMP_PATH,
        **{arg: ANONYMIZED_DB_ARGS[arg] for arg in ('dbname', 'user', 'host', 'port', 'password')}
    ))
    anonymize_main()
    assert_db_anonymized(anonymized)