--- a/tests/test-draft.t	Fri Jul 01 17:03:49 2011 +0200
+++ b/tests/test-draft.t	Fri Jul 01 17:20:45 2011 +0200
@@ -42,7 +42,7 @@
   0:5caa672bac26: published
 
 turn draft on (repo side)
-  $ echo draft > .hg/states/Enabled
+  $ hg states draft
   $ hg log --template='{rev}:{node|short}: {state}\n'
   3:73585b17392a: draft
   2:3c8695235a32: draft
@@ -74,7 +74,7 @@
   added 2 changesets with 2 changes to 1 files
 
 turn draft off again (repo side)
-  $ sed -i 's/draft//' .hg/states/Enabled
+  $ hg states --off draft
   $ hg log --template='{rev}:{node|short}: {state}\n'
   3:73585b17392a: published
   2:3c8695235a32: published
@@ -87,7 +87,7 @@
   3:73585b17392a
 
 turn draft on again (repo side)
-  $ echo draft > .hg/states/Enabled
+  $ hg states draft
 
 test incoming and pull
 
@@ -163,7 +163,7 @@
 turn draft off again (repo side)
   $ cd ..
   $ "$TESTDIR/killdaemons.py"
-  $ sed -i 's/draft//' ./local/.hg/states/Enabled
+  $ hg -R local states --off draft
   $ hg -R local serve -p $HGPORT -d --pid-file=local.pid
   $ cat local.pid >> "$DAEMON_PIDS"
   $ cd fromhttp2
@@ -185,7 +185,7 @@
   $ "$TESTDIR/killdaemons.py"
 
 turn draft on again (repo side)
-  $ echo draft > ./local/.hg/states/Enabled
+  $ hg -R local states draft
   $ hg init httpto
   $ hg -R httpto serve -p $HGPORT -d --pid-file=remote.pid
   $ cat remote.pid >> "$DAEMON_PIDS"