From bc409569f58a2c9c1c0393c923650bf84f159ddc Mon Sep 17 00:00:00 2001
From: Markus Lehtonen <markus.lehtonen@linux.intel.com>
Date: Wed, 4 Sep 2013 11:58:21 +0300
Subject: [PATCH] GitRepository: Add clean() method

Signed-off-by: Lingchaox Xin <lingchaox.xin@intel.com>
Signed-off-by: Markus Lehtonen <markus.lehtonen@linux.intel.com>
---
 gbp/git/repository.py       | 21 +++++++++++++++++++++
 tests/test_GitRepository.py | 18 ++++++++++++++++++
 2 files changed, 39 insertions(+)

diff --git a/gbp/git/repository.py b/gbp/git/repository.py
index 41e6471..6eed588 100644
--- a/gbp/git/repository.py
+++ b/gbp/git/repository.py
@@ -765,6 +765,27 @@ class GitRepository(object):
             break
         return (ret, "".join(out))
 
+    def clean(self, directories=False, force=False, dry_run=False):
+        """
+        Remove untracked files from the working tree.
+
+        @param directories: remove untracked directories, too
+        @type directories: C{bool}
+        @param force: satisfy git configuration variable clean.requireForce
+        @type force: C{bool}
+        @param dry_run: don’t actually remove anything
+        @type dry_run: C{bool}
+        """
+        options = GitArgs()
+        options.add_true(directories, '-d')
+        options.add_true(force, '-f')
+        options.add_true(dry_run, '-n')
+
+        _out, err, ret = self._git_inout('clean', options.args,
+                                         extra_env={'LC_ALL': 'C'})
+        if ret:
+            raise GitRepositoryError("Can't execute repository clean: %s" % err)
+
     def is_empty(self):
         """
         Is the repository empty?
diff --git a/tests/test_GitRepository.py b/tests/test_GitRepository.py
index 699ecd4..187712e 100644
--- a/tests/test_GitRepository.py
+++ b/tests/test_GitRepository.py
@@ -104,6 +104,24 @@ def test_branch_master():
     'master'
     """
 
+def test_clean():
+    """
+    Remove untracked files from the working tree
+
+    Methods tested:
+         - L{gbp.git.GitRepository.clean}
+
+    >>> import gbp.git, shutil, os
+    >>> repo = gbp.git.GitRepository(repo_dir)
+    >>> shutil.copy(os.path.join(repo.path, ".git/HEAD"), \
+                                 os.path.join(repo.path, "testclean"))
+    >>> repo.clean(dry_run=True)
+    >>> repo.is_clean()[0]
+    False
+    >>> repo.clean(directories=True, force=True)
+    >>> repo.is_clean()[0]
+    True
+    """
 
 def test_create_branch():
     """
-- 
2.7.4