# HG changeset patch
# User Eric Hopper <hopper@omnifarious.org>
# Date 1179457784 25200
# Node ID 871ff96a86ccb9513d783fd904a04c4a7365666c
# Parent  3900f684a150ba9a847c499dd7bf6bf139070866
Add a test for the Branch header being in hg export.

diff --git a/tests/test-impexp-branch b/tests/test-impexp-branch
new file mode 100755
--- /dev/null
+++ b/tests/test-impexp-branch
@@ -0,0 +1,39 @@
+#!/bin/sh
+
+cat >findbranch.py <<EOF
+import re, sys
+
+head_re = re.compile('^#(?:(?:\\s+([A-Za-z][A-Za-z0-9_]*)(?:\\s.*)?)|(?:\\s*))$')
+
+for line in sys.stdin:
+    hmatch = head_re.match(line)
+    if not hmatch:
+        sys.exit(1)
+    if hmatch.group(1) == 'Branch':
+        sys.exit(0)
+sys.exit(1)
+EOF
+hg init a
+cd a
+echo "Rev 1" >rev
+hg add rev
+hg commit -m "No branch."
+hg branch abranch
+echo "Rev  2" >rev
+hg commit -m "With branch."
+if hg export 0 | python ../findbranch.py; then
+    echo "Export of default branch revision has Branch header" 1>&2
+    exit 1
+fi
+if hg export 1 | python ../findbranch.py; then
+    :  # Do nothing
+else
+    echo "Export of branch revision is missing Branch header" 1>&2
+    exit 1
+fi
+# Make sure import still works with branch information in patches.
+cd ..
+hg init b
+cd b
+hg -R ../a export 0 | hg import -
+hg -R ../a export 1 | hg import -
diff --git a/tests/test-impexp-branch.out b/tests/test-impexp-branch.out
new file mode 100644
--- /dev/null
+++ b/tests/test-impexp-branch.out
@@ -0,0 +1,2 @@
+applying patch from stdin
+applying patch from stdin