comparison tests/test-commandserver.t @ 40355:77ab5fbdbbf0

py3: use bprint() helper in test-commandserver.t

40354:f9f2faf25fc4

  $ mv $HGRCPATH.new $HGRCPATH

  $ hg init repo
  $ cd repo

  >>> from __future__ import absolute_import, print_function
  >>> import os
  >>> import sys
  >>> from hgclient import check, readchannel, runcommand
  >>> @check
  ... def hellomessage(server):
  ...     ch, data = readchannel(server)
  ...     print(b'%c, %r' % (ch, data))
  ...     # run an arbitrary command to make sure the next thing the server
  ...     # sends isn't part of the hello message
  ...     runcommand(server, [b'id'])
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  *** runcommand id

  000000000000 tip
  *** runcommand id -runknown
  abort: unknown revision 'unknown'!
   [255]

  >>> from hgclient import check, readchannel
  >>> @check
  ... def inputeof(server):
  ...     readchannel(server)
  ...     server.stdin.write(b'runcommand\n')
  ...     # close stdin while server is waiting for input
  ...     server.stdin.close()
  ... 
  ...     # server exits with 1 if the pipe closed while reading the command
  ...     print(b'server exit code =', server.wait())
  server exit code = 1

  >>> from hgclient import check, readchannel, runcommand, stringio
  >>> @check
  ... def serverinput(server):

#else
  $ PYTHONPATH="$TESTTMP/repo:$PYTHONPATH"
#endif

  $ cat <<EOF > hook.py
  > from __future__ import print_function
  > import sys
  > def hook(**args):
  >     print(b'hook talking')
  >     print(b'now try to read something: %r' % sys.stdin.read())
  > EOF

  >>> from hgclient import check, readchannel, runcommand, stringio
  >>> @check
  ... def hookoutput(server):

  summary:     2
  
  *** runcommand status

  >>> import os
  >>> from hgclient import check, readchannel, runcommand
  >>> @check
  ... def bookmarks(server):
  ...     readchannel(server)
  ...     runcommand(server, [b'bookmarks'])
  ... 

  ...     f = open('a', 'ab')
  ...     f.write(b'a\n') and None
  ...     f.close()
  ...     runcommand(server, [b'commit', b'-Amm'])
  ...     runcommand(server, [b'bookmarks'])
  ...     print(b'')
  *** runcommand bookmarks
  no bookmarks set
  *** runcommand bookmarks
     bm1                       1:d3a0a68be6de
     bm2                       1:d3a0a68be6de

  3: draft
  *** runcommand phase -r .
  3: public

  $ echo a >> a
  >>> from hgclient import check, readchannel, runcommand
  >>> @check
  ... def rollback(server):
  ...     readchannel(server)
  ...     runcommand(server, [b'phase', b'-r', b'.', b'-p'])
  ...     runcommand(server, [b'commit', b'-Am.'])
  ...     runcommand(server, [b'rollback'])
  ...     runcommand(server, [b'phase', b'-r', b'.'])
  ...     print(b'')
  *** runcommand phase -r . -p
  no phases changed
  *** runcommand commit -Am.
  *** runcommand rollback
  repository tip rolled back to revision 3 (undo commit)

  marked working directory as branch default
  (branches are permanent and global, did you want a bookmark?)

  $ touch .hgignore
  >>> import os
  >>> from hgclient import check, readchannel, runcommand
  >>> @check
  ... def hgignore(server):
  ...     readchannel(server)
  ...     runcommand(server, [b'commit', b'-Am.'])
  ...     f = open('ignored-file', 'ab')

  ...     f.close()
  ...     f = open('.hgignore', 'ab')
  ...     f.write(b'ignored-file')
  ...     f.close()
  ...     runcommand(server, [b'status', b'-i', b'-u'])
  ...     print(b'')
  *** runcommand commit -Am.
  adding .hgignore
  *** runcommand status -i -u
  I ignored-file
  

cache of non-public revisions should be invalidated on repository change
(issue4855):

  >>> import os
  >>> from hgclient import check, readchannel, runcommand
  >>> @check
  ... def phasesetscacheaftercommit(server):
  ...     readchannel(server)
  ...     # load _phasecache._phaserevs and _phasesets
  ...     runcommand(server, [b'log', b'-qr', b'draft()'])

  ...         f.write(b'a\n') and None
  ...         f.close()
  ...         os.system('hg commit -Aqm%d' % i)
  ...     # new commits should be listed as draft revisions
  ...     runcommand(server, [b'log', b'-qr', b'draft()'])
  ...     print(b'')
  *** runcommand log -qr draft()
  4:7966c8e3734d
  *** runcommand log -qr draft()
  4:7966c8e3734d
  5:41f6602d1c4f
  6:10501e202c35
  

  >>> import os
  >>> from hgclient import check, readchannel, runcommand
  >>> @check
  ... def phasesetscacheafterstrip(server):
  ...     readchannel(server)
  ...     # load _phasecache._phaserevs and _phasesets
  ...     runcommand(server, [b'log', b'-qr', b'draft()'])
  ...     # strip cached revisions by another process
  ...     os.system('hg --config extensions.strip= strip -q 5')
  ...     # shouldn't abort by "unknown revision '6'"
  ...     runcommand(server, [b'log', b'-qr', b'draft()'])
  ...     print(b'')
  *** runcommand log -qr draft()
  4:7966c8e3734d
  5:41f6602d1c4f
  6:10501e202c35
  *** runcommand log -qr draft()

  stdout should be redirected to stderr


run commandserver in commandserver, which is silly but should work:

  >>> from __future__ import print_function
  >>> from hgclient import check, readchannel, runcommand, stringio
  >>> @check
  ... def nested(server):
  ...     print(b'%c, %r' % readchannel(server))
  ...     class nestedserver(object):
  ...         stdin = stringio(b'getencoding\n')
  ...         stdout = stringio()
  ...     runcommand(server, [b'serve', b'--cmdserver', b'pipe'],
  ...                output=nestedserver.stdout, input=nestedserver.stdin)
  ...     nestedserver.stdout.seek(0)
  ...     print(b'%c, %r' % readchannel(nestedserver))  # hello
  ...     print(b'%c, %r' % readchannel(nestedserver))  # getencoding
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  *** runcommand serve --cmdserver pipe
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  r, '*' (glob)


start without repository:

  $ cd ..

  >>> from __future__ import print_function
  >>> from hgclient import check, readchannel, runcommand
  >>> @check
  ... def hellomessage(server):
  ...     ch, data = readchannel(server)
  ...     print(b'%c, %r' % (ch, data))
  ...     # run an arbitrary command to make sure the next thing the server
  ...     # sends isn't part of the hello message
  ...     runcommand(server, [b'id'])
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  *** runcommand id

  $ cd repo
  $ hg update -q

#if unix-socket unix-permissions

  >>> from __future__ import print_function
  >>> from hgclient import check, readchannel, runcommand, stringio, unixserver
  >>> server = unixserver(b'.hg/server.sock', b'.hg/server.log')
  >>> def hellomessage(conn):
  ...     ch, data = readchannel(conn)
  ...     print(b'%c, %r' % (ch, data))
  ...     runcommand(conn, [b'id'])
  >>> check(hellomessage, server.connect)
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  *** runcommand id
  eff892de26ec tip bm1/bm2/bm3

  $ cat <<EOF >> .hg/hgrc
  > [cmdserver]
  > log = inexistent/path.log
  > EOF
  >>> from __future__ import print_function
  >>> from hgclient import check, readchannel, unixserver
  >>> server = unixserver(b'.hg/server.sock', b'.hg/server.log')
  >>> def earlycrash(conn):
  ...     while True:
  ...         try:
  ...             ch, data = readchannel(conn)
  ...             if not data.startswith(b'  '):
  ...                 print(b'%c, %r' % (ch, data))
  ...         except EOFError:
  ...             break
  >>> check(earlycrash, server.connect)
  e, 'Traceback (most recent call last):\n'
  e, "IOError: *" (glob)

40355:77ab5fbdbbf0

  $ mv $HGRCPATH.new $HGRCPATH

  $ hg init repo
  $ cd repo

  >>> from __future__ import absolute_import
  >>> import os
  >>> import sys
  >>> from hgclient import bprint, check, readchannel, runcommand
  >>> @check
  ... def hellomessage(server):
  ...     ch, data = readchannel(server)
  ...     bprint(b'%c, %r' % (ch, data))
  ...     # run an arbitrary command to make sure the next thing the server
  ...     # sends isn't part of the hello message
  ...     runcommand(server, [b'id'])
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  *** runcommand id

  000000000000 tip
  *** runcommand id -runknown
  abort: unknown revision 'unknown'!
   [255]

  >>> from hgclient import bprint, check, readchannel
  >>> @check
  ... def inputeof(server):
  ...     readchannel(server)
  ...     server.stdin.write(b'runcommand\n')
  ...     # close stdin while server is waiting for input
  ...     server.stdin.close()
  ... 
  ...     # server exits with 1 if the pipe closed while reading the command
  ...     bprint(b'server exit code =', b'%d' % server.wait())
  server exit code = 1

  >>> from hgclient import check, readchannel, runcommand, stringio
  >>> @check
  ... def serverinput(server):

#else
  $ PYTHONPATH="$TESTTMP/repo:$PYTHONPATH"
#endif

  $ cat <<EOF > hook.py
  > import sys
  > from hgclient import bprint
  > def hook(**args):
  >     bprint(b'hook talking')
  >     bprint(b'now try to read something: %r' % sys.stdin.read())
  > EOF

  >>> from hgclient import check, readchannel, runcommand, stringio
  >>> @check
  ... def hookoutput(server):

  summary:     2
  
  *** runcommand status

  >>> import os
  >>> from hgclient import bprint, check, readchannel, runcommand
  >>> @check
  ... def bookmarks(server):
  ...     readchannel(server)
  ...     runcommand(server, [b'bookmarks'])
  ... 

  ...     f = open('a', 'ab')
  ...     f.write(b'a\n') and None
  ...     f.close()
  ...     runcommand(server, [b'commit', b'-Amm'])
  ...     runcommand(server, [b'bookmarks'])
  ...     bprint(b'')
  *** runcommand bookmarks
  no bookmarks set
  *** runcommand bookmarks
     bm1                       1:d3a0a68be6de
     bm2                       1:d3a0a68be6de

  3: draft
  *** runcommand phase -r .
  3: public

  $ echo a >> a
  >>> from hgclient import bprint, check, readchannel, runcommand
  >>> @check
  ... def rollback(server):
  ...     readchannel(server)
  ...     runcommand(server, [b'phase', b'-r', b'.', b'-p'])
  ...     runcommand(server, [b'commit', b'-Am.'])
  ...     runcommand(server, [b'rollback'])
  ...     runcommand(server, [b'phase', b'-r', b'.'])
  ...     bprint(b'')
  *** runcommand phase -r . -p
  no phases changed
  *** runcommand commit -Am.
  *** runcommand rollback
  repository tip rolled back to revision 3 (undo commit)

  marked working directory as branch default
  (branches are permanent and global, did you want a bookmark?)

  $ touch .hgignore
  >>> import os
  >>> from hgclient import bprint, check, readchannel, runcommand
  >>> @check
  ... def hgignore(server):
  ...     readchannel(server)
  ...     runcommand(server, [b'commit', b'-Am.'])
  ...     f = open('ignored-file', 'ab')

  ...     f.close()
  ...     f = open('.hgignore', 'ab')
  ...     f.write(b'ignored-file')
  ...     f.close()
  ...     runcommand(server, [b'status', b'-i', b'-u'])
  ...     bprint(b'')
  *** runcommand commit -Am.
  adding .hgignore
  *** runcommand status -i -u
  I ignored-file
  

cache of non-public revisions should be invalidated on repository change
(issue4855):

  >>> import os
  >>> from hgclient import bprint, check, readchannel, runcommand
  >>> @check
  ... def phasesetscacheaftercommit(server):
  ...     readchannel(server)
  ...     # load _phasecache._phaserevs and _phasesets
  ...     runcommand(server, [b'log', b'-qr', b'draft()'])

  ...         f.write(b'a\n') and None
  ...         f.close()
  ...         os.system('hg commit -Aqm%d' % i)
  ...     # new commits should be listed as draft revisions
  ...     runcommand(server, [b'log', b'-qr', b'draft()'])
  ...     bprint(b'')
  *** runcommand log -qr draft()
  4:7966c8e3734d
  *** runcommand log -qr draft()
  4:7966c8e3734d
  5:41f6602d1c4f
  6:10501e202c35
  

  >>> import os
  >>> from hgclient import bprint, check, readchannel, runcommand
  >>> @check
  ... def phasesetscacheafterstrip(server):
  ...     readchannel(server)
  ...     # load _phasecache._phaserevs and _phasesets
  ...     runcommand(server, [b'log', b'-qr', b'draft()'])
  ...     # strip cached revisions by another process
  ...     os.system('hg --config extensions.strip= strip -q 5')
  ...     # shouldn't abort by "unknown revision '6'"
  ...     runcommand(server, [b'log', b'-qr', b'draft()'])
  ...     bprint(b'')
  *** runcommand log -qr draft()
  4:7966c8e3734d
  5:41f6602d1c4f
  6:10501e202c35
  *** runcommand log -qr draft()

  stdout should be redirected to stderr


run commandserver in commandserver, which is silly but should work:

  >>> from hgclient import bprint, check, readchannel, runcommand, stringio
  >>> @check
  ... def nested(server):
  ...     bprint(b'%c, %r' % readchannel(server))
  ...     class nestedserver(object):
  ...         stdin = stringio(b'getencoding\n')
  ...         stdout = stringio()
  ...     runcommand(server, [b'serve', b'--cmdserver', b'pipe'],
  ...                output=nestedserver.stdout, input=nestedserver.stdin)
  ...     nestedserver.stdout.seek(0)
  ...     bprint(b'%c, %r' % readchannel(nestedserver))  # hello
  ...     bprint(b'%c, %r' % readchannel(nestedserver))  # getencoding
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  *** runcommand serve --cmdserver pipe
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  r, '*' (glob)


start without repository:

  $ cd ..

  >>> from hgclient import bprint, check, readchannel, runcommand
  >>> @check
  ... def hellomessage(server):
  ...     ch, data = readchannel(server)
  ...     bprint(b'%c, %r' % (ch, data))
  ...     # run an arbitrary command to make sure the next thing the server
  ...     # sends isn't part of the hello message
  ...     runcommand(server, [b'id'])
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  *** runcommand id

  $ cd repo
  $ hg update -q

#if unix-socket unix-permissions

  >>> from hgclient import bprint, check, readchannel, runcommand, stringio, unixserver
  >>> server = unixserver(b'.hg/server.sock', b'.hg/server.log')
  >>> def hellomessage(conn):
  ...     ch, data = readchannel(conn)
  ...     bprint(b'%c, %r' % (ch, data))
  ...     runcommand(conn, [b'id'])
  >>> check(hellomessage, server.connect)
  o, 'capabilities: getencoding runcommand\nencoding: *\npid: *' (glob)
  *** runcommand id
  eff892de26ec tip bm1/bm2/bm3

  $ cat <<EOF >> .hg/hgrc
  > [cmdserver]
  > log = inexistent/path.log
  > EOF
  >>> from hgclient import bprint, check, readchannel, unixserver
  >>> server = unixserver(b'.hg/server.sock', b'.hg/server.log')
  >>> def earlycrash(conn):
  ...     while True:
  ...         try:
  ...             ch, data = readchannel(conn)
  ...             if not data.startswith(b'  '):
  ...                 bprint(b'%c, %r' % (ch, data))
  ...         except EOFError:
  ...             break
  >>> check(earlycrash, server.connect)
  e, 'Traceback (most recent call last):\n'
  e, "IOError: *" (glob)