Ejemplos de main en Python

Ejemplo n.º 1

Archivo: test_cli.py Proyecto: rbwinslow/hq

def test_syntax_error_prints_proper_error_message(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression='child:://')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body('')

    main()

    _, actual = capture_console_output(capsys)
    assert re.match(r'^syntax error.+expected.+name.+got.+slash', actual.lower())

Ejemplo n.º 2

Archivo: test_cli.py Proyecto: rbwinslow/hq

def test_query_error_prints_proper_error_message(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression='no-such-function()')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body('')

    main()

    _, actual = capture_console_output(capsys)
    assert re.match(r'^query error.+unknown function.+no-such-function', actual.lower())

Ejemplo n.º 3

Archivo: test_cli.py Proyecto: rbwinslow/hq

def test_ugly_flag_preserves_markup_formatting(capsys, mocker):
    expected = '<p>I, too, enjoy PyCharm.</p>'
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression='//p', ugly=True)
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(expected)

    main()

    actual, _ = capture_console_output(capsys, strip=False)
    assert actual == expected

Ejemplo n.º 4

Archivo: test_unicode_support.py Proyecto: rbwinslow/hq

def test_tolerates_latin_characters_in_comments(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression='//comment()')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(u"""
    <!-- sacr\xe9 bleu! -->""")

    main()

    actual, _ = capture_console_output(capsys)
    assert actual == expected_result(u'<!-- sacr\xe9 bleu! -->')

Ejemplo n.º 5

def test_query_error_prints_proper_error_message(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='no-such-function()')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body('')

    main()

    _, actual = capture_console_output(capsys)
    assert re.match(r'^query error.+unknown function.+no-such-function',
                    actual.lower())

Ejemplo n.º 6

def test_syntax_error_prints_proper_error_message(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='child:://')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body('')

    main()

    _, actual = capture_console_output(capsys)
    assert re.match(r'^syntax error.+expected.+name.+got.+slash',
                    actual.lower())

Ejemplo n.º 7

def test_ugly_flag_preserves_markup_formatting(capsys, mocker):
    expected = '<p>I, too, enjoy PyCharm.</p>'
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='//p', ugly=True)
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(expected)

    main()

    actual, _ = capture_console_output(capsys, strip=False)
    assert actual == expected

Ejemplo n.º 8

Archivo: test_unicode_support.py Proyecto: rbwinslow/hq

def test_tolerates_latin_characters_in_comments(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='//comment()')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(u"""
    <!-- sacr\xe9 bleu! -->""")

    main()

    actual, _ = capture_console_output(capsys)
    assert actual == expected_result(u'<!-- sacr\xe9 bleu! -->')

Ejemplo n.º 9

Archivo: test_unicode_support.py Proyecto: rbwinslow/hq

def test_tolerates_latin_characters_in_attribute_contents(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression='//div/@role')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(u"""
    <div role="prim\xe4r">
    </div>""")

    main()

    actual, _ = capture_console_output(capsys)
    assert actual == expected_result(u'role="prim\xe4r"')

Ejemplo n.º 10

Archivo: test_unicode_support.py Proyecto: rbwinslow/hq

def test_tolerates_latin_characters_in_attribute_contents(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='//div/@role')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(u"""
    <div role="prim\xe4r">
    </div>""")

    main()

    actual, _ = capture_console_output(capsys)
    assert actual == expected_result(u'role="prim\xe4r"')

Ejemplo n.º 11

Archivo: test_cli.py Proyecto: rbwinslow/hq

def test_preserve_space_flag_causes_non_breaking_spaces_to_be_how_shall_we_say_preserved(capsys, mocker):
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(u'<p>non\u00a0breaking&nbsp;spaces?</p>')

    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression='//p/text()', preserve='s')
    main()
    actual, _ = capture_console_output(capsys)
    assert actual == u'non\u00a0breaking\u00a0spaces?'

    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression='//p/text()', preserve='')
    main()
    actual, _ = capture_console_output(capsys)
    assert actual == u'non breaking spaces?'

Ejemplo n.º 12

Archivo: test_cli.py Proyecto: rbwinslow/hq

def test_reading_input_from_a_file_instead_of_stdin(capsys, mocker):
    expected_filename = 'filename.html'
    mocked_open = mock_open(read_data=wrap_html_body('<p>foo</p>'))
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='//p/text()', file=expected_filename)
    mocker.patch('hq.hq.open', mocked_open, create=True)

    main()

    actual, _ = capture_console_output(capsys)
    mocked_open.assert_called_with(expected_filename)
    assert actual == 'foo'

Ejemplo n.º 13

def test_reading_input_from_a_file_instead_of_stdin(capsys, mocker):
    expected_filename = 'filename.html'
    mocked_open = mock_open(read_data=wrap_html_body('<p>foo</p>'))
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='//p/text()', file=expected_filename)
    mocker.patch('hq.hq.open', mocked_open, create=True)

    main()

    actual, _ = capture_console_output(capsys)
    mocked_open.assert_called_with(expected_filename)
    assert actual == 'foo'

Ejemplo n.º 14

Archivo: test_unicode_support.py Proyecto: rbwinslow/hq

def test_tolerates_latin_characters_in_element_contents(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression='//div')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(u"""
    <div>
        T\xeate\xa0\xe0\xa0t\xeate
    </div>""")

    main()

    actual, _ = capture_console_output(capsys)
    assert actual == expected_result(u"""
    <div>
     T\xeate\xa0\xe0\xa0t\xeate
    </div>""")

Ejemplo n.º 15

Archivo: test_cli.py Proyecto: rbwinslow/hq

def test_preserve_space_flag_turns_off_space_normalization(capsys, mocker):
    hquery = '`${//p}`'
    content_with_spaces = '   PyCharm     rocks!    '
    mocker.patch('sys.stdin.read').return_value = wrap_html_body('<p>{0}</p>'.format(content_with_spaces))

    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression=hquery, preserve='s')
    main()
    actual, _ = capture_console_output(capsys, strip=False)
    assert actual == content_with_spaces

    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(expression=hquery, preserve='')
    main()
    actual, _ = capture_console_output(capsys, strip=False)
    assert actual == 'PyCharm rocks!'

Ejemplo n.º 16

Archivo: test_unicode_support.py Proyecto: rbwinslow/hq

def test_tolerates_latin_characters_in_element_contents(capsys, mocker):
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='//div')
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(u"""
    <div>
        T\xeate\xa0\xe0\xa0t\xeate
    </div>""")

    main()

    actual, _ = capture_console_output(capsys)
    assert actual == expected_result(u"""
    <div>
     T\xeate\xa0\xe0\xa0t\xeate
    </div>""")

Ejemplo n.º 17

def test_preserve_space_flag_causes_non_breaking_spaces_to_be_how_shall_we_say_preserved(
        capsys, mocker):
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(
        u'<p>non\u00a0breaking&nbsp;spaces?</p>')

    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='//p/text()', preserve='s')
    main()
    actual, _ = capture_console_output(capsys)
    assert actual == u'non\u00a0breaking\u00a0spaces?'

    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression='//p/text()', preserve='')
    main()
    actual, _ = capture_console_output(capsys)
    assert actual == u'non breaking spaces?'

Ejemplo n.º 18

Archivo: test_cli.py Proyecto: rbwinslow/hq

def test_program_flag_reads_hquery_program_from_file(capsys, mocker):
    expected_filename = 'filename.hq'
    mocked_open = mock_open(read_data='''
                                        //p
                                        ->
                                        $_/text()''')
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        program=expected_filename)
    mocker.patch('sys.stdin.read').return_value = wrap_html_body('<p>foo</p>')
    mocker.patch('hq.hq.open', mocked_open, create=True)

    main()

    actual, _ = capture_console_output(capsys)
    mocked_open.assert_called_with(expected_filename)
    assert actual == 'foo'

Ejemplo n.º 19

def test_program_flag_reads_hquery_program_from_file(capsys, mocker):
    expected_filename = 'filename.hq'
    mocked_open = mock_open(read_data='''
                                        //p
                                        ->
                                        $_/text()''')
    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        program=expected_filename)
    mocker.patch('sys.stdin.read').return_value = wrap_html_body('<p>foo</p>')
    mocker.patch('hq.hq.open', mocked_open, create=True)

    main()

    actual, _ = capture_console_output(capsys)
    mocked_open.assert_called_with(expected_filename)
    assert actual == 'foo'

Ejemplo n.º 20

def test_preserve_space_flag_turns_off_space_normalization(capsys, mocker):
    hquery = '`${//p}`'
    content_with_spaces = '   PyCharm     rocks!    '
    mocker.patch('sys.stdin.read').return_value = wrap_html_body(
        '<p>{0}</p>'.format(content_with_spaces))

    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression=hquery, preserve='s')
    main()
    actual, _ = capture_console_output(capsys, strip=False)
    assert actual == content_with_spaces

    mocker.patch('hq.hq.docopt').return_value = simulate_args_dict(
        expression=hquery, preserve='')
    main()
    actual, _ = capture_console_output(capsys, strip=False)
    assert actual == 'PyCharm rocks!'

Ejemplo n.º 21

Archivo: hq_runner.py Proyecto: rbwinslow/hq

#!/usr/bin/env python

"""
Convenience wrapper for running hq directly from source tree. (Based on Jan-Philip Gehrcke's python-cmdline-bootstrap.)
"""

from hq.hq import main


if __name__ == '__main__':
    main()

Ejemplo n.º 22

#!/usr/bin/env python
"""
Convenience wrapper for running hq directly from source tree. (Based on Jan-Philip Gehrcke's python-cmdline-bootstrap.)
"""

from hq.hq import main

if __name__ == '__main__':
    main()