def __add(self, stanza, lineno, ret, config): """Private method for gathering stanzas.""" if not stanza.is_empty(): ret.append(stanza.finish(lineno)) if config: config.add(stanza) stanza = ConfigStanza() return stanza
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 load(self, file, config=None): """Parses an open file object, and returns a list of stanzas found. If an optional config argument (a Config object) is provided, it also adds the stanzas to it. Parsing errors are raised as exceptions. """ ret = [] stanza = ConfigStanza() lineno = 0 for line in file: lineno += 1 line = line.strip() if line == '': stanza = self.__add(stanza, lineno, ret, config) continue colon = line.find(':') if colon != -1: key, val = line.split(':', 1) stanza.set(key.strip(), val.strip(), lineno) else: raise ValueError('invalid line %d: %s' % (lineno, repr(line))) stanza = self.__add(stanza, lineno, ret, config) return ret
def load(self, file, config = None): """Parses an open file object, and returns a list of stanzas found. If an optional config argument (a Config object) is provided, it also adds the stanzas to it. Parsing errors are raised as exceptions. """ ret = [] stanza = ConfigStanza() lineno = 0 for line in file: lineno += 1 line = line.strip() if line == '': stanza = self.__add(stanza, lineno, ret, config) continue colon = line.find(':') if colon != -1: key, val = line.split(':', 1) stanza.set(key.strip(), val.strip(), lineno) else: raise ValueError('invalid line %d: %s' % (lineno, repr(line))) stanza = self.__add(stanza, lineno, ret, config) return ret
def setUp(self): super(Test_ConfigStanza, self).setUp() self.cs = ConfigStanza() self.assert_(self.cs.is_empty())
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_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
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)