Add fetch & push refspec.

Author: xtao

src/remote.c

@@ -729,11 +729,60 @@ Remote_push(Remote *self, PyObject *args)
 }
 
 
+PyDoc_STRVAR(Remote_add_push__doc__,
+    "add_push(refspec)\n"
+    "\n"
+    "Add a push refspec to the remote.");
+
+PyObject *
+Remote_add_push(Remote *self, PyObject *args)
+{
+    git_remote *remote;
+    char *refspec = NULL;
+    int err = 0;
+
+    if (!PyArg_ParseTuple(args, "s", &refspec))
+        return NULL;
+
+    remote = self->remote;
+    err = git_remote_add_push(remote, refspec);
+    if (err < 0)
+        return Error_set(err);
+
+    Py_RETURN_NONE;
+}
+
+
+PyDoc_STRVAR(Remote_add_fetch__doc__,
+    "add_fetch(refspec)\n"
+    "\n"
+    "Add a fetch refspec to the remote.");
+
+PyObject *
+Remote_add_fetch(Remote *self, PyObject *args)
+{
+    git_remote *remote;
+    char *refspec = NULL;
+    int err = 0;
+
+    if (!PyArg_ParseTuple(args, "s", &refspec))
+        return NULL;
+
+    remote = self->remote;
+    err = git_remote_add_fetch(remote, refspec);
+    if (err < 0)
+        return Error_set(err);
+
+    Py_RETURN_NONE;
+}
+
 PyMethodDef Remote_methods[] = {
     METHOD(Remote, fetch, METH_NOARGS),
     METHOD(Remote, save, METH_NOARGS),
     METHOD(Remote, get_refspec, METH_O),
     METHOD(Remote, push, METH_VARARGS),
+    METHOD(Remote, add_push, METH_VARARGS),
+    METHOD(Remote, add_fetch, METH_VARARGS),
     METHOD(Remote, get_fetch_refspecs, METH_NOARGS),
     METHOD(Remote, set_fetch_refspecs, METH_O),
     METHOD(Remote, get_push_refspecs, METH_NOARGS),

test/test_remote.py

@@ -151,6 +151,15 @@ def test_remote_save(self):
         self.assertEqual('http://example.com/test.git',
                          self.repo.remotes[0].url)
 
+    def test_add_refspec(self):
+        remote = self.repo.create_remote('test_add_refspec', REMOTE_URL)
+        remote.add_push('refs/heads/*:refs/heads/test_refspec/*')
+        self.assertEqual('refs/heads/*:refs/heads/test_refspec/*',
+                         remote.get_push_refspecs()[0])
+        remote.add_fetch('+refs/heads/*:refs/remotes/test_refspec/*')
+        self.assertEqual('+refs/heads/*:refs/remotes/test_refspec/*',
+                         remote.get_fetch_refspecs()[1])
+
 
 class EmptyRepositoryTest(utils.EmptyRepoTestCase):
     def test_fetch(self):