# HG changeset patch
# User Augie Fackler <augie@google.com>
# Date 1524802101 14400
# Node ID 0dcd03637d365c9cc43af7a273fd50ba31c328e8
# Parent  bb1b15acfd06c55118bffe3cd7973dbb4d62148a
tests: fix error case in test-url.py's doctest

This required some careful attention, so I wanted to split it out from
the uninteresting bits that'll be in the next change.

Differential Revision: https://phab.mercurial-scm.org/D3469

diff -r bb1b15acfd06 -r 0dcd03637d36 tests/test-url.py
--- a/tests/test-url.py	Fri Apr 27 00:06:08 2018 -0400
+++ b/tests/test-url.py	Fri Apr 27 00:08:21 2018 -0400
@@ -236,7 +236,8 @@
 
 def test_url():
     """
-    >>> from mercurial.util import url
+    >>> from mercurial import error, pycompat
+    >>> from mercurial.util import forcebytestr, url
 
     This tests for edge cases in url.URL's parsing algorithm. Most of
     these aren't useful for documentation purposes, so they aren't
@@ -349,10 +350,11 @@
 
     Non-localhost file URL:
 
-    >>> u = url('file://mercurial-scm.org/foo')
-    Traceback (most recent call last):
-      File "<stdin>", line 1, in ?
-    Abort: file:// URLs can only refer to localhost
+    >>> try:
+    ...   u = url(b'file://mercurial-scm.org/foo')
+    ... except error.Abort as e:
+    ...   forcebytestr(e)
+    'file:// URLs can only refer to localhost'
 
     Empty URL: