From 3ffebbde5266f643fb791ef6728a88ef7fbdd1f5 Mon Sep 17 00:00:00 2001
From: Alessio Ababilov <aababilo@yahoo-inc.com>
Date: Tue, 22 Jan 2013 15:43:29 +0200
Subject: Add tests for setup

Test write_git_changelog, generate_authors, get_cmdclass, and
parse_dependency_links. Also test private _run_shell_command
function.

Change-Id: I583d679afc1dc805c920b4beffc2f664833aabd0
---
 tests/unit/test_setup.py | 114 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 114 insertions(+)

diff --git a/tests/unit/test_setup.py b/tests/unit/test_setup.py
index dfd8841..8f6324d 100644
--- a/tests/unit/test_setup.py
+++ b/tests/unit/test_setup.py
@@ -17,6 +17,7 @@
 
 import os
 import sys
+import StringIO
 from tempfile import mkstemp
 
 import fixtures
@@ -25,6 +26,22 @@ from openstack.common.setup import *
 from tests import utils
 
 
+class DiveDir(fixtures.Fixture):
+    """Dive into given directory and return back on cleanup.
+
+    :ivar path: The target directory.
+    """
+
+    def __init__(self, path):
+        self.path = path
+
+    def setUp(self):
+        super(DiveDir, self).setUp()
+        self.old_path = os.getcwd()
+        os.chdir(self.path)
+        self.addCleanup(os.chdir, self.old_path)
+
+
 class EmailTestCase(utils.BaseTestCase):
 
     def test_str_dict_replace(self):
@@ -60,6 +77,77 @@ class MailmapTestCase(utils.BaseTestCase):
                          parse_mailmap(self.mailmap))
 
 
+class GitLogsTest(utils.BaseTestCase):
+
+    def setUp(self):
+        super(GitLogsTest, self).setUp()
+        temp_path = self.useFixture(fixtures.TempDir()).path
+        self.useFixture(DiveDir(temp_path))
+        self.useFixture(fixtures.MonkeyPatch("os.path.isdir",
+                                             lambda path: path == ".git"))
+
+    def test_write_git_changelog(self):
+        self.useFixture(fixtures.FakePopen(lambda _: {
+            "stdout": StringIO.StringIO("Author: Foo Bar <email@bar.com>\n")
+        }))
+        with open(".mailmap", "w") as mm_fh:
+            mm_fh.write("Foo Bar <email@foo.com> <email@bar.com>\n")
+
+        write_git_changelog()
+
+        with open("ChangeLog", "r") as ch_fh:
+            self.assertTrue("email@foo.com" in ch_fh.read())
+
+    def test_generate_authors(self):
+        author_old = "Foo Foo <email@foo.com>"
+        author_new = "Bar Bar <email@bar.com>"
+
+        self.useFixture(fixtures.FakePopen(lambda proc_args: {
+            "stdout": StringIO.StringIO(
+                author_new
+                if proc_args["args"][2].startswith("git log")
+                else "")
+        }))
+
+        with open("AUTHORS.in", "w") as auth_fh:
+            auth_fh.write(author_old)
+
+        generate_authors()
+
+        with open("AUTHORS", "r") as auth_fh:
+            authors = auth_fh.read()
+            self.assertTrue(author_old in authors)
+            self.assertTrue(author_new in authors)
+
+
+class GetCmdClassTest(utils.BaseTestCase):
+
+    def test_get_cmdclass(self):
+        cmdclass = get_cmdclass()
+
+        self.assertTrue("sdist" in cmdclass)
+        build_sphinx = cmdclass.get("build_sphinx")
+        if build_sphinx:
+            self.useFixture(fixtures.MonkeyPatch(
+                "sphinx.setup_command.BuildDoc.run", lambda self: None))
+            from distutils.dist import Distribution
+            distr = Distribution()
+            distr.packages = ("fake_package",)
+            distr.command_options["build_sphinx"] = {"source_dir": ["a", "."]}
+            pkg_fixture = fixtures.PythonPackage(
+                "fake_package", [("fake_module.py", "")])
+            self.useFixture(pkg_fixture)
+            self.useFixture(DiveDir(pkg_fixture.base))
+
+            build_doc = build_sphinx(distr)
+            build_doc.run()
+
+            self.assertTrue(
+                os.path.exists("api/autoindex.rst"))
+            self.assertTrue(
+                os.path.exists("api/fake_package.fake_module.rst"))
+
+
 class ParseRequirementsTest(utils.BaseTestCase):
 
     def setUp(self):
@@ -93,3 +181,29 @@ class ParseRequirementsTest(utils.BaseTestCase):
             fh.write("argparse")
         if sys.version_info >= (2, 7):
             self.assertEqual([], parse_requirements([self.tmp_file]))
+
+    def test_get_requirement_from_file_empty(self):
+        actual = get_reqs_from_files([])
+        self.assertEqual([], actual)
+
+
+class ParseDependencyLinksTest(utils.BaseTestCase):
+
+    def setUp(self):
+        super(ParseDependencyLinksTest, self).setUp()
+        self.useFixture(fixtures.NestedTempfile())
+        (fd, self.tmp_file) = mkstemp(prefix="openstack", suffix=".setup")
+
+    def test_parse_dependency_normal(self):
+        with open(self.tmp_file, "w") as fh:
+            fh.write("http://test.com\n")
+        self.assertEqual(
+            ["http://test.com"],
+            parse_dependency_links([self.tmp_file]))
+
+    def test_parse_dependency_with_git_egg_url(self):
+        with open(self.tmp_file, "w") as fh:
+            fh.write("-e git://foo.com/zipball#egg=bar")
+        self.assertEqual(
+            ["git://foo.com/zipball#egg=bar"],
+            parse_dependency_links([self.tmp_file]))
-- 
cgit