[PATCH 3 of 3 STABLE] rust-cpython: raising error.WdirUnsupported

[Georges Racinet]

# HG changeset patch
# User Georges Racinet <georges.racinet at octobus.net>
# Date 1548247776 18000
#      Wed Jan 23 07:49:36 2019 -0500
# Branch stable
# Node ID 47d5458a4c32e20de4af1237cfeb157153586cbc
# Parent  a35cfd592a90ae325b452c56fe8bff86cac097dd
# EXP-Topic rust-wdirunsupported
rust-cpython: raising error.WdirUnsupported

The Graph implementation of hg-cpython returns the appropriate error
upon encounter with the working directory special revision number, and
this gives us in particular a code path to test from test-rust-ancestors.py

In the current implementation, the exception is actually raised from
the iterator instantiation; we are nonetheless consuming the iterator
in the test with `list()` in order not to depend on implementation details.

diff -r a35cfd592a90 -r 47d5458a4c32 mercurial/revlog.py
--- a/mercurial/revlog.py	Wed Jan 23 07:47:04 2019 -0500
+++ b/mercurial/revlog.py	Wed Jan 23 07:49:36 2019 -0500
@@ -896,8 +896,6 @@
             common = [nullrev]
 
         if rustext is not None:
-            # TODO: WdirUnsupported should be raised instead of GraphError
-            # if common includes wdirrev
             return rustext.ancestor.MissingAncestors(self.index, common)
         return ancestor.incrementalmissingancestors(self.parentrevs, common)
 
diff -r a35cfd592a90 -r 47d5458a4c32 rust/hg-cpython/src/cindex.rs
--- a/rust/hg-cpython/src/cindex.rs	Wed Jan 23 07:47:04 2019 -0500
+++ b/rust/hg-cpython/src/cindex.rs	Wed Jan 23 07:49:36 2019 -0500
@@ -16,7 +16,7 @@
 
 use self::python_sys::PyCapsule_Import;
 use cpython::{PyClone, PyErr, PyObject, PyResult, Python};
-use hg::{Graph, GraphError, Revision};
+use hg::{Graph, GraphError, Revision, WORKING_DIRECTORY_REVISION};
 use libc::c_int;
 use std::ffi::CStr;
 use std::mem::transmute;
@@ -86,6 +86,9 @@
 impl Graph for Index {
     /// wrap a call to the C extern parents function
     fn parents(&self, rev: Revision) -> Result<[Revision; 2], GraphError> {
+        if rev == WORKING_DIRECTORY_REVISION {
+            return Err(GraphError::WorkingDirectoryUnsupported);
+        }
         let mut res: [c_int; 2] = [0; 2];
         let code = unsafe {
             (self.parents)(
diff -r a35cfd592a90 -r 47d5458a4c32 tests/test-rust-ancestor.py
--- a/tests/test-rust-ancestor.py	Wed Jan 23 07:47:04 2019 -0500
+++ b/tests/test-rust-ancestor.py	Wed Jan 23 07:49:36 2019 -0500
@@ -1,6 +1,10 @@
 from __future__ import absolute_import
 import sys
 import unittest
+from mercurial import (
+    error,
+    node,
+    )
 
 try:
     from mercurial import rustext
@@ -153,6 +157,12 @@
         # rust-cpython issues appropriate str instances for Python 2 and 3
         self.assertEqual(exc.args, ('ParentOutOfRange', 1))
 
+    def testwdirunsupported(self):
+        # trying to access ancestors of the working directory raises
+        # WdirUnsupported directly
+        idx = self.parseindex()
+        with self.assertRaises(error.WdirUnsupported):
+            list(AncestorsIterator(idx, [node.wdirrev], -1, False))
 
 if __name__ == '__main__':
     import silenttestrunner