test_dependency_versioning.py

#!/bin/env python
from __future__ import print_function
from __future__ import unicode_literals
from __future__ import division

import unittest

import tempfile
import subprocess
import os

import yaml

import dependency_versioning as dv

def mktempdir():
    tempdir = tempfile.TemporaryDirectory()
    tempdir_name = self.tempdir.name
    subprocess.Popen(
        "OD=$PWD&&cd {dir:s}&&tar xzf $OD/test_repo.tgz".format(dir=self.tempdir_name),
        shell="True", universal_newlines=True).communicate()
    return (tempdir)

def rmtempdir(tempdir):
    tempdir.cleanup()

class TestDependencyVersioning(unittest.TestCase):
    vif = {
        "test": {
            "type": "git",
            "repository": "nowhere",
            "branch": "master",
        }
    }
    def test_init(self):
        "Test internalizing vifs"
        with tempfile.NamedTemporaryFile(mode="w", delete=False) as input_viffile:
            yaml.dump(self.vif, input_viffile)
            input_viffile.close()
            test_vif = dv.VersionInformationFile(input_viffile.name)
        self.assertDictEqual(self.vif, test_vif)

    def test_write_vif(self):
        "Test writing out a vif."
        # Write out the test vif and read it in.
        with tempfile.NamedTemporaryFile(mode="w", delete=False) as input_viffile:
            yaml.dump(self.vif, input_viffile)
            input_viffile.close()
            input_vif = dv.VersionInformationFile(input_viffile.name)
        # Dump it out.
        with tempfile.NamedTemporaryFile(mode="r", delete=False) as output_viffile:
            input_vif.dump(output_viffile.name)
            output_viffile.close()
            # Read in the dump.
            output_vif = dv.VersionInformationFile(output_viffile.name)
        self.assertDictEqual(self.vif, output_vif)

    def test_main(self):
        self.tempdir = tempfile.TemporaryDirectory()
        self.tempdir_name = self.tempdir.name
        subprocess.Popen(
            "OD=$PWD&&cd {dir:s}&&tar xzf $OD/test_repo.tgz".format(dir=self.tempdir_name),
            shell="True", universal_newlines=True).communicate()
        base_dir = os.getcwd()
        try:
            os.chdir(self.tempdir_name)
            with open("master.vif", "w") as output_vif_file:
                yaml.dump(self.vif, output_vif_file)
            output_vif = dv.main(["--file","master.vif","--output-file","new.vif", "--no-update"])
            self.assertDictEqual(self.vif, output_vif)
            with open("new.vif", "r") as written_vif_file:
                written_vif = yaml.load(written_vif_file)
            self.assertDictEqual(self.vif, written_vif)
            self.tempdir.cleanup()
        finally:
            os.chdir(base_dir)

    def test_print_version(self):
        self.tempdir = tempfile.TemporaryDirectory()
        self.tempdir_name = self.tempdir.name
        subprocess.Popen(
            "OD=$PWD&&cd {dir:s}&&tar xzf $OD/test_repo.tgz".format(dir=self.tempdir_name),
            shell="True", universal_newlines=True).communicate()
        base_dir = os.getcwd()
        try:
            os.chdir(self.tempdir_name)
            with open("master.vif", "w") as output_vif_file:
                yaml.dump(self.vif, output_vif_file)
            output_vif = dv.main(["--file","master.vif","--print-version","test", "--no-update"])
            self.assertDictEqual(self.vif, output_vif)
            self.tempdir.cleanup()
        finally:
            os.chdir(base_dir)
            

class TestGITDependencyVersioning(unittest.TestCase):
    def setUp(self):
        self.tempdir = tempfile.TemporaryDirectory()
        self.tempdir_name = self.tempdir.name
        subprocess.Popen(
            "OD=$PWD&&cd {dir:s}&&tar xzf $OD/test_repo.tgz".format(dir=self.tempdir_name),
            shell="True", universal_newlines=True).communicate()

    def tearDown(self):
        self.tempdir.cleanup()

    def test_git_clone(self):
        try:
            (stdout, stderr) = subprocess.Popen("rm -fr test_repo", shell=True, universal_newlines=True).communicate()
        except:
            pass
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": os.path.join(self.tempdir_name, "test_repo"),
                "branch": "master"})
        git_dep.update()
        branch = git_dep.get_current_branch()
        self.assertEqual(branch, "master")

    def test_git_update(self):
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": os.path.join(self.tempdir_name, "test_repo"),
                "branch": "master"})
        git_dep.update()
        self.assertEqual(git_dep.get_present_version(), "4cc3b223d2997c8bbf84bfeae420b0d1e4dad732")
        (stdout, stderr) = subprocess.Popen("cd test_repo && git reset --hard HEAD^", shell=True, universal_newlines=True).communicate()
        self.assertEqual(git_dep.get_present_version(), "395c738b6aa8b7074d7d2e533c95276cc0990876")
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": os.path.join(self.tempdir_name, "test_repo"),
                "branch": "master"})
        git_dep.update()
        self.assertEqual(git_dep.get_present_version(), "4cc3b223d2997c8bbf84bfeae420b0d1e4dad732")

    def test_git_change_branch(self):
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": os.path.join(self.tempdir_name, "test_repo"),
                "branch": "master"})
        git_dep.update()
        self.assertEqual(git_dep.get_present_version(), "4cc3b223d2997c8bbf84bfeae420b0d1e4dad732")
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": os.path.join(self.tempdir_name, "test_repo"),
                "branch": "dev"})
        git_dep.update()
        self.assertEqual(git_dep.get_present_version(), "238aacacf8d4c0a73f3395912c5114763fa195ab")

    def test_git_set_version(self):
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": os.path.join(self.tempdir_name, "test_repo"),
                "branch": "master"})
        git_dep.update()
        self.assertEqual(git_dep.get_present_version(), "4cc3b223d2997c8bbf84bfeae420b0d1e4dad732")
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": os.path.join(self.tempdir_name, "test_repo"),
                "branch": "master",
                "version": "395c738b6aa8b7074d7d2e533c95276cc0990876"})
        git_dep.update()
        self.assertEqual(git_dep.get_present_version(), "395c738b6aa8b7074d7d2e533c95276cc0990876")

    def test_git_get_info(self):
        repo = os.path.join(self.tempdir_name, "test_repo")
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": repo,
                "branch": "master",
                "version": "395c738b6aa8b7074d7d2e533c95276cc0990876"})
        git_dep = dv.GITDependency(
            name="test_repo",
            version_info = {
                "repository": repo,
                "branch": "master"})
        git_dep.update()
        self.assertDictEqual(
            git_dep.get_present_info(),
            {
                "type": "git",
                "repository": repo,
                "branch": "master",
                "version": "4cc3b223d2997c8bbf84bfeae420b0d1e4dad732"
            }
        )
        self.assertDictEqual(
            git_dep,
            {
                "type": "git",
                "repository": repo,
                "branch": "master",
                "version": "4cc3b223d2997c8bbf84bfeae420b0d1e4dad732"
            }
        )

if __name__ == "__main__":
    main()