tests: add support for inline doctests in test files This adds doctest like syntax to .t files, that can be interleaved with regular shell code: $ echo -n a > file >>> print open('file').read() a >>> open('file', 'a').write('b') $ cat file ab The syntax is exactly the same as regular doctests, so multiline statements look like this: >>> for i in range(3): ... print i 0 1 2 Each block has its own context, i.e.: >>> x = 0 >>> print x 0 $ echo 'foo' foo >>> print x will result in a NameError. Errors are displayed in standard doctest format: >>> print 'foo' bar --- /home/idan/dev/hg/default/tests/test-test.t +++ /home/idan/dev/hg/default/tests/test-test.t.err @@ -2,3 +2,16 @@ > >>> print 'foo' > bar > EOF + ********************************************************************** + File "/tmp/tmps8X_0ohg-tst", line 1, in tmps8X_0ohg-tst + Failed example: + print 'foo' + Expected: + bar + Got: + foo + ********************************************************************** + 1 items had failures: + 1 of 1 in tmps8X_0ohg-tst + ***Test Failed*** 1 failures. + [1] As for the implementation, it's quite simple: when the test runner sees a line starting with '>>>' it converts it, and all subsequent lines until the next line that begins with '$' to a 'python -m heredoctest <<EOF' call with the proper heredoc to follow. So if we have this test file: >>> for c in 'abcd': ... print c a b c d $ echo foo foo It gets converted to: $ python -m heredoctest <<EOF > >>> for c in 'abcd': > ... print c > a > b > c > d > EOF $ echo foo foo And then processed like every other test file by converting it to a sh script. [ original upstream message ]

     1

#!/usr/bin/env python

     2

     3

"""This does HTTP GET requests given a host:port and path and returns

     4

a subset of the headers plus the body of the result."""

     5

     6

import httplib, sys, re

     7

     8

try:

     9

    import msvcrt, os

    10

    msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)

    11

    msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)

    12

except ImportError:

    13

    pass

    14

    15

headers = [h.lower() for h in sys.argv[3:]]

    16

conn = httplib.HTTPConnection(sys.argv[1])

    17

conn.request("GET", sys.argv[2])

    18

response = conn.getresponse()

    19

print response.status, response.reason

    20

for h in headers:

    21

    if response.getheader(h, None) is not None:

    22

        print "%s: %s" % (h, response.getheader(h))

    23

print

    24

data = response.read()

    25

sys.stdout.write(data)

    26

    27

if 200 <= response.status <= 299:

    28

    sys.exit(0)

    29

sys.exit(1)