Mercurial > treepkg
view test/test_util.py @ 570:44c0f8404983
Refactor git pull command out of update
Tag MUST NOT use update because therefore it always changes the current local branch!
For listing the tags it's enough to pull the latest repo changes
author | Bjoern Ricks <bricks@intevation.de> |
---|---|
date | Fri, 02 Sep 2011 11:46:29 +0000 |
parents | ca95be9d033a |
children |
line wrap: on
line source
# Copyright (C) 2007, 2008 by Intevation GmbH # Authors: # Bernhard Herzog <bh@intevation.de> # Bjoern Ricks <bjoern.ricks@intevation.de> # # This program is free software under the GPL (>=v2) # Read the file COPYING coming with the software for details. """Tests for the treepkg.util module""" import os import unittest import shutil from filesupport import FileTestMixin from treepkg.util import replace_in_file, listdir_abs, md5sum, \ remove_trailing_slashes, expand_filename, \ compress_all_logs class TestReplaceInFile(unittest.TestCase, FileTestMixin): def runtest(self, orig_contents, expected_contents, pattern, replacement): filename = self.create_temp_file("orig", orig_contents) changed = replace_in_file(filename, pattern, replacement) self.assertEquals(changed, orig_contents != expected_contents) self.check_file_contents(filename, expected_contents) def test_version_replacement(self): template = ("project foo version 1.0-svn%(rev)d" "Some filler" "text that sometimes" "looks similar to the pattern" "1.0-" "foo 1.2-svn2" "echo foo version 1.0-svn%(rev)d" "" "and more filler") self.runtest(template % dict(rev=0), template % dict(rev=321), r"1\.0-svn0", "1.0-svn321") def test_no_matches(self): """Tests replace_in_file when no matches are found""" template = ("project foo version 1.0-svn%(rev)d" "Some filler" "text that sometimes" "looks similar to the pattern" "1.0-" "foo 1.2-svn2" "echo foo version 1.0-svn%(rev)d" "" "and more filler") self.runtest(template % dict(rev=0), template % dict(rev=0), r"0\.9-svn0", "1.0-svn321") class TestListDirAbs(unittest.TestCase, FileTestMixin): def setUp(self): self.directory = self.create_temp_dir("a_directory") def test_listdir_abs(self): directory = self.create_files("dir", [("foo.orig.tgz", ""), ("foo.dsc", ""), ("foo.diff.gz", ""),]) self.assertEquals(sorted(listdir_abs(directory)), sorted([os.path.join(directory, d) for d in ["foo.orig.tgz", "foo.dsc", "foo.diff.gz"]])) def test_listdir_abs_pattern(self): directory = self.create_files("dir", [("foo.orig.tgz", ""), ("foo.dsc", ""), ("foo.diff.gz", ""),]) self.assertEquals(sorted(listdir_abs(directory, '*.dsc')), [os.path.join(directory, "foo.dsc")]) class TestMd5sum(unittest.TestCase, FileTestMixin): content = "this is a test content" def setUp(self): self.testfile = self.create_temp_file("testmd5.txt", self.content) def test_md5sum(self): sum = md5sum(self.testfile) self.assertEquals("a12511153555c1f0f0a1eda200733a3f", sum) class TestRemoveTrailingSlashes(unittest.TestCase): def test_remove_trailing_slashes(self): dir_w_slash = "/tmp/dir/" dir_wo_slash = "/tmp/dir" self.assertEquals(dir_wo_slash, remove_trailing_slashes(dir_w_slash)) self.assertEquals(dir_wo_slash, remove_trailing_slashes(dir_wo_slash)) class TestExpandFilename(unittest.TestCase): def test_expand_filenam(self): os.environ['MY_TEST_VAR'] = "def" path = "/abc/${MY_TEST_VAR}/" self.assertEquals("/abc/def/", expand_filename(path)) class TestCompressAllLogs(unittest.TestCase, FileTestMixin): def test_compress_all_logs(self): log_dir = self.create_test_specific_temp_dir() try: for i in range(15): f = open(os.path.join(log_dir, "log_%d.txt" % i), "w") print >> f, "World domination is right at hand!" f.close() ref_log = os.path.join(log_dir, "log_0.txt") compress_all_logs(ref_log) for i in range(15): self.assertTrue(os.path.isfile(os.path.join( log_dir, "log_%d.txt.gz" % i))) finally: shutil.rmtree(log_dir, ignore_errors=True)