From 981267a35e44fea8804ccf249ff89d7850f87703 Mon Sep 17 00:00:00 2001 From: rsiddharth Date: Wed, 9 Nov 2016 04:19:25 +0000 Subject: Add `write_file` function. * gd_diff.py (write_file): New function. (save_gns_readme, slurp_gns_readme): Update functions. * tests/test_gd_diff.py (test_write_file): New method. (setup, test_save_gns_readme, test_save_gns_readme_double) (teardown): Update methods --- tests/test_gd_diff.py | 22 ++++++++++++++++++++-- 1 file changed, 20 insertions(+), 2 deletions(-) (limited to 'tests') diff --git a/tests/test_gd_diff.py b/tests/test_gd_diff.py index a1be47e..cb36a7c 100644 --- a/tests/test_gd_diff.py +++ b/tests/test_gd_diff.py @@ -28,20 +28,35 @@ class TestGdDiff(object): self.small_pkgs_file = 'tests/files/small-parkes-pkgs.list' self.pkgs_file_ne = 'tests/nonexistent-file.list' self.gns_pkgs_dir = 'tests/gns-pkgs' + self.test_home = 'tests/HOME' + self.test_w_file = os.path.join(self.test_home, 'w_file') + self.stderr_orig = sys.stderr + # make test home + os.mkdir(self.test_home, mode=0o700) + + def test_read_file_success(self): f_content = read_file(self.pkgs_file) assert isinstance(f_content, str) assert_equal(len(f_content.split('\n')), 82) + @raises(SystemExit) def test_read_file_error(self): with open(os.devnull, 'w') as sys.stderr: f_content = read_file(self.pkgs_file_ne) + def test_write_file(self): + content = 'One Goodbye\n Stealing Romance' + write_file(self.test_w_file, content) + + assert read_file(self.test_w_file) == content + + def test_execute_success(self): cmd = 'python --version' cp = execute(cmd, out=subprocess.PIPE) @@ -92,7 +107,7 @@ class TestGdDiff(object): def test_save_gns_readme(self): cmd = 'bzr cat bzr://bzr.sv.gnu.org/gnewsense/packages-parkes/antlr/debian/README.gNewSense' cp = execute(cmd, out=subprocess.PIPE) - readme_content = cp.stdout + readme_content = cp.stdout.decode() # convert to str # save it save_gns_readme(readme_content, 'parkes', 'antlr', self.gns_pkgs_dir) @@ -105,7 +120,7 @@ class TestGdDiff(object): def test_save_gns_readme_double(self): cmd = 'bzr cat bzr://bzr.sv.gnu.org/gnewsense/packages-parkes/antlr/debian/README.gNewSense' cp = execute(cmd, out=subprocess.PIPE) - readme_content = cp.stdout + readme_content = cp.stdout.decode() # convert to str # save it twice save_gns_readme(readme_content, 'parkes', 'antlr', self.gns_pkgs_dir) @@ -220,5 +235,8 @@ class TestGdDiff(object): os.chmod(self.gns_pkgs_dir, mode=0o700) rmtree(self.gns_pkgs_dir) + if(path.exists(self.test_home)): + rmtree(self.test_home) + # restore sys.stderr sys.stderr = self.stderr_orig -- cgit v1.2.3