Add `slurp_gns_readme` function (gd_diff).

* gd_diff.py (slurp_gns_readme): New function.
* tests/test_gd_diff.py (test_slurp_gns_readme_success)
(test_slurp_gns_readme_error): Tests for `slurp_gns_readme` function.
(setup, teardown): Update functions.

1 files changed, 24 insertions, 0 deletions

diff --git a/tests/test_gd_diff.py b/tests/test_gd_diff.py
index 7409554..f518373 100644
--- a/tests/test_gd_diff.py
+++ b/tests/test_gd_diff.py
@@ -25,6 +25,7 @@ class TestGdDiff(object):
         self.pkgs_file = "tests/files/pkgs.list"
         self.pkgs_file_ne = 'tests/nonexistent-file.list'
         self.gns_pkgs_dir = 'tests/gns-pkgs'
+        self.stderr_orig = sys.stderr
 
     def test_read_file_success(self):
         f_content = read_file(self.pkgs_file)
@@ -107,9 +108,32 @@ class TestGdDiff(object):
             save_gns_readme(readme_content, 'parkes', 'antlr', self.gns_pkgs_dir)
 
 
+    def test_slurp_gns_readme_success(self):
+        saved = slurp_gns_readme('parkes', 'antlr', self.gns_pkgs_dir)
+        assert saved == True
+
+        gns_readme_file = path.join(self.gns_pkgs_dir, 'parkes',
+                                    'antlr', 'debian',
+                                    'README.gNewSense')
+        with open(gns_readme_file, 'rb') as f:
+            assert f.read() == b'Changed-From-Debian: Removed example with non-free files.\nChange-Type: Modified\n\nFor gNewSense, the non-free unicode.IDENTs files are *actually* removed (see\nalso README.source). See gNewSense bug #34218 for details.\n'
+
+
+    def test_slurp_gns_readme_error(self):
+        saved = slurp_gns_readme('parkes', 'non-existent-pkg', self.gns_pkgs_dir)
+        assert saved == False
+
+        gns_readme_file = path.join(self.gns_pkgs_dir, 'parkes',
+                                    'non-existent-pkg', 'debian',
+                                    'README.gNewSense')
+        assert not path.exists(gns_readme_file)
+
+
     def teardown(self):
         """Teardown method for this class."""
         if(path.exists(self.gns_pkgs_dir)):
             os.chmod(self.gns_pkgs_dir, mode=0o700)
             rmtree(self.gns_pkgs_dir)
 
+        # restore sys.stderr
+        sys.stderr = self.stderr_orig