示例#1
0
 def test_force_prompt_grab_inputs(self):
     """ Test the force_prompt grabbing of inputs """
     target = Manifest(StringIO(manifest_force_inputs))
     target.get_config = Mock()
     target.grab_inputs(force_prompt=True)
     target.get_config.assert_has_calls([
         call("gitroot", default="~/workspace", secret=False, force_prompt=True),
         call("username", default=None, secret=False, force_prompt=True),
         call("main_branch", default="comp_main", secret=True, force_prompt=True)
     ])
示例#2
0
class TestManifest(object):
    """
    Tests the Manifest object
    """

    def setup(self):
        self.old_manifest = Manifest(StringIO(old_manifest))
        self.new_manifest = Manifest(StringIO(new_manifest))

    def test_dependency_order(self):
        """ Test whether a proper dependency tree generated the correct output. """
        sections = self.old_manifest.formula_sections()
        assert sections.index('git') < sections.index('sub'), \
            "Dependency is out of order! git comes after sub"

    @tools.raises(ManifestException)
    def test_incorrect_dependency(self):
        """ Test whether an incorrect dependency tree returns an error. """
        Manifest(StringIO(manifest_incorrect_dependency))

    def test_equality(self):
        """ Manifest object should be equal to itself """
        tools.eq_(self.old_manifest, Manifest(StringIO(old_manifest)))

    def test_get_feature_config(self):
        """ get_feature_config should return a dictionary with the attributes """
        tools.eq_(self.old_manifest.get_feature_config("sub").to_dict(), {
            'url': 'git://github.com/Toumorokoshi/sub.git',
            'formula': 'sprinter.formulas.git',
            'depends': 'git',
            'branch': 'yusuke',
            'rc': 'temp=`pwd`; cd %(sub:root_dir)s/libexec && . sub-init2 && cd $tmp',
            'bc': 'temp=`pwd`; cd %(sub:testvar)s/libexec && . sub-init2 && cd $tmp'})

    def test_get_context_dict(self):
        """ Test getting a config dict """
        context_dict = self.old_manifest.get_context_dict()
        test_dict = {'maven:formula': 'sprinter.formulas.unpack',
                     'maven:specific_version': '2.10',
                     'ant:formula': 'sprinter.formulas.unpack',
                     'mysql:formula': 'sprinter.formulas.package',
                     'sub:rc': 'temp=`pwd`; cd %(sub:root_dir)s/libexec && . sub-init2 && cd $tmp',
                     'ant:specific_version': '1.8.4',
                     'sub:formula': 'sprinter.formulas.git',
                     'sub:branch': 'yusuke',
                     'git:apt-get': 'git-core',
                     'sub:url': 'git://github.com/Toumorokoshi/sub.git',
                     'ant:phases': 'update',
                     'sub:depends': 'git',
                     'config:namespace': 'sprinter',
                     'sub:bc': 'temp=`pwd`; cd %(sub:testvar)s/libexec && . sub-init2 && cd $tmp',
                     'mysql:apt-get': 'libmysqlclient\nlibmysqlclient-dev',
                     'mysql:brew': 'mysql',
                     'git:brew': 'git',
                     'git:formula': 'sprinter.formulas.package',
                     'config:inputs': 'sourceonly'}
        for k, v in test_dict.items():
            tools.eq_(context_dict[k], v)
        self.old_manifest.add_additional_context({"config:test": "testing this"})
        assert "config:test" in self.old_manifest.get_context_dict()

    def test_get_context_dict_escaped_character(self):
        """ Test getting a config dict with escaping filter will properly escape a character"""
        manifest = Manifest(StringIO(manifest_escaped_parameters))
        context_dict = manifest.get_context_dict()
        assert "section:escapeme|escaped" in context_dict
        tools.eq_(context_dict["section:escapeme|escaped"], "\!\@\#\$\%\^\&\*\(\)\\\"\\'\~\`\/\?\<\>")

    def test_add_additional_context(self):
        """ Test the add additonal context method """
        self.old_manifest.add_additional_context({'testme': 'testyou'})
        assert 'testme' in self.old_manifest.additional_context_variables
        self.old_manifest.add_additional_context({'testhim': 'testher'})
        assert 'testme' in self.old_manifest.additional_context_variables
        assert 'testhim' in self.old_manifest.additional_context_variables

    @httpretty.activate
    def test_source_from_url(self):
        """ When the manifest is sourced from a url, the source should be the url. """
        TEST_URI = "http://testme.com/test.cfg"
        httpretty.register_uri(httpretty.GET, TEST_URI,
                               body=http_manifest)
        m = Manifest(TEST_URI)
        assert m.source() == TEST_URI

    @patch.object(lib, 'prompt')
    def test_get_config(self, prompt):
        """ Test the get config """
        prompt.return_value = "no"
        self.new_manifest.get_config('hobopopo', default="Yes", secret=False)
        prompt.assert_called_once_with("please enter your hobopopo",
                                       default="Yes", secret=False)

    @patch.object(lib, 'prompt')
    def test_get_config_force_prompt(self, prompt):
        """ Test the get config with force_prompt """
        prompt.return_value = "no"
        self.new_manifest.set('config', 'hobopopo', 'test')
        self.new_manifest.get_config('hobopopo', default="Yes", secret=False, force_prompt=True)
        prompt.assert_called_once_with("please enter your hobopopo",
                                       default="Yes", secret=False)

    def test_grab_inputs(self):
        """ Test grabbing inputs """
        self.new_manifest.get_config = Mock()
        self.new_manifest.grab_inputs()
        self.new_manifest.get_config.assert_has_calls([
            call("gitroot", default="~/workspace", secret=False, force_prompt=False),
            call("username", default=None, secret=False, force_prompt=False),
            call("password", default=None, secret=True, force_prompt=False),
            call("main_branch", default="comp_main", secret=True, force_prompt=False)
        ])

    def test_force_prompt_grab_inputs(self):
        """ Test the force_prompt grabbing of inputs """
        target = Manifest(StringIO(manifest_force_inputs))
        target.get_config = Mock()
        target.grab_inputs(force_prompt=True)
        target.get_config.assert_has_calls([
            call("gitroot", default="~/workspace", secret=False, force_prompt=True),
            call("username", default=None, secret=False, force_prompt=True),
            call("main_branch", default="comp_main", secret=True, force_prompt=True)
        ])

    def test_write(self):
        """ Test the write command """
        temp_file = tempfile.mkstemp()[1]
        try:
            with open(temp_file, 'w+') as fh:
                self.new_manifest.write(fh)
            tools.eq_(self.new_manifest, Manifest(temp_file))
        finally:
            os.unlink(temp_file)
        
    @patch.object(lib, 'prompt')
    def test_write_with_temporary_config(self, prompt):
        """ The Write command with a temporary config value should not be written """
        prompt.return_value = "no"
        self.new_manifest.get_config('hobopopo', default="Yes", secret=False)
        new_manifest = StringIO()
        self.new_manifest.write(new_manifest)
        assert not Manifest(new_manifest).has_option('config', 'hobopopo'), \
            "A secret value was written to the config!"

    def skip_additional_context(self):
        """ Ensure that additional context variables are configured correctly """
        additional_context_variables = {"sub:root_dir": "teststring"}
        self.config.set_additional_context('source', additional_context_variables)
        sub_values = self.config.source.get_feature_config("sub")
        assert sub_values['rc'].find("teststring") != -1, "teststring is not substituted in"
        # should add to context
        additional_context_variables = {"sub:testvar": "teststring2"}
        self.config.set_additional_context('source', additional_context_variables)
        sub_values = self.config.source.get_feature_config("sub")
        assert sub_values['rc'].find("teststring") != -1, "teststring is not substituted in"
        assert sub_values['bc'].find("teststring2") != -1, "teststring2 is not substituted in"