def _create_mock_config(self, hashes): stanzas = [] for hash in hashes: stanza = ConfigStanza() stanza.set('package', 'apackage', 1) stanza.set('accept-origin', hash['accepted_origin'], 2) stanza.set('track-origin', hash['track_origin'], 3) stanza.set('track-version', hash['track_version'], 4) stanza.finish(4) stanzas.append(stanza) mock_config = self.struct() mock_config.package = lambda name: name == 'apackage' and stanzas or [] return mock_config
class Test_ConfigStanza(test_helper.MoxTestCase): def setUp(self): super(Test_ConfigStanza, self).setUp() self.cs = ConfigStanza() self.assert_(self.cs.is_empty()) def test_empty(self): self.assertRaisesWithMessageContaining(ValueError, 'line 2', self.cs.finish, 2) def test_all_attributes(self): self.cs.set('Package', 'dpkg', 2) self.assert_(not self.cs.is_empty()) self.cs.set('Accept-Origin', 'NonDebian Stuff', 3) self.cs.set('Track-oriGIN', 'Debian', 4) self.cs.set('track-version', '1.2.3', 5) self.assertEquals(self.cs.get('package'), 'dpkg') self.assertEquals(self.cs.get('accept-origin'), 'NonDebian Stuff') self.assert_(self.cs.matches('accept-origin', 'NonDebian Stuff')) self.assert_(not self.cs.matches('accept-origin', 'Debian')) self.assertEquals(self.cs.get('track-origin'), 'Debian') self.assertEquals(self.cs.get('track-version'), '1.2.3') self.assert_(not self.cs.is_empty()) self.assertEquals(self.cs.finish(6), self.cs) def test_invalid_attribute(self): self.assertRaisesWithMessageContaining(ValueError, 'invalid tag', self.cs.set, 'FoO', 'bar', 2) self.assertRaisesWithMessageContaining(ValueError, 'line 2', self.cs.set, 'FoO', 'bar', 2) def test_wildcard_attributes(self): self.cs.set('Package', 'dpkg', 2) self.assert_(not self.cs.is_empty()) self.cs.set('Accept-Origin', '*', 3) self.cs.set('Track-oriGIN', '*', 4) self.cs.set('track-version', '1.2.3', 5) self.assertEquals(self.cs.get('accept-origin'), '*') self.assert_(self.cs.matches('accept-origin', 'whatever')) self.assertEquals(self.cs.get('track-origin'), '*') self.assert_(self.cs.matches('track-origin', 'whatever')) self.assert_(not self.cs.is_empty()) self.assertEquals(self.cs.finish(6), self.cs)
def _create_default_stanza(self, status, base=False): """Creates a stanza used when there is no configuration for a given package. Such Stanza has a matching Package tag, Accept-Origin that matches anything, Track-Origin equal to the distributor ID according to facter, and Track-Version equal to the candidate version (or the base of the candidate version, if base argument is True). """ s = ConfigStanza() s.set('package', status.package_name, 1) s.set('accept-origin', '*', 1) s.set('track-origin', self.facter.distributors_id, 1) if base: v = self.base(status.candidate_version.string) else: v = status.candidate_version.string s.set('track-version', v, 1) return s.finish(1)
def _create_default_stanza(self, status, base = False): """Creates a stanza used when there is no configuration for a given package. Such Stanza has a matching Package tag, Accept-Origin that matches anything, Track-Origin equal to the distributor ID according to facter, and Track-Version equal to the candidate version (or the base of the candidate version, if base argument is True). """ s = ConfigStanza() s.set('package', status.package_name, 1) s.set('accept-origin', '*', 1) s.set('track-origin', self.facter.distributors_id, 1) if base: v = self.base(status.candidate_version.string) else: v = status.candidate_version.string s.set('track-version', v, 1) return s.finish(1)