def setUp(self): # create a temp file for use as a config file. This should get cleaned # up magically at the end of the run. self.fid = NamedTemporaryFile(mode='w+b', suffix='.tmp') self.fid.write(TEST_CONFIG) self.fid.seek(0) self.cfgParser = RhsmConfigParser(self.fid.name)
def test_auto_persist(self): config = Config(self.parser, auto_persist=True) self.assertEqual('baz', config['foo']['quux']) config['foo']['quux'] = 'fizz' self.assertEqual('fizz', config['foo']['quux']) reparsed = RhsmConfigParser(self.fid.name) self.assertEqual('fizz', reparsed.get('foo', 'quux'))
def test_auto_persist(self): config = Config(self.parser, auto_persist=True) self.assertEqual("baz", config["foo"]["quux"]) config["foo"]["quux"] = "fizz" self.assertEqual("fizz", config["foo"]["quux"]) reparsed = RhsmConfigParser(self.fid.name) self.assertEqual("fizz", reparsed.get("foo", "quux"))
def test_persist_cascades(self): config = Config(self.parser, auto_persist=False) self.assertEqual('baz', config['foo']['quux']) config['foo']['quux'] = 'fizz' config.persist() self.assertEqual('fizz', config['foo']['quux']) reparsed = RhsmConfigParser(self.fid.name) self.assertEqual('fizz', reparsed.get('foo', 'quux'))
def test_persist_cascades(self): config = Config(self.parser, auto_persist=False) self.assertEqual("baz", config["foo"]["quux"]) config["foo"]["quux"] = "fizz" config.persist() self.assertEqual("fizz", config["foo"]["quux"]) reparsed = RhsmConfigParser(self.fid.name) self.assertEqual("fizz", reparsed.get("foo", "quux"))
def upload_tracer_profile(conduit=False): data = json.dumps({"traces": get_apps(conduit)}) headers = {"Content-type": "application/json"} conn = httplib.HTTPSConnection( RhsmConfigParser.get(initConfig(), "server", "hostname"), RhsmConfigParser.get(initConfig(), "server", "port"), key_file=ConsumerIdentity.keypath(), cert_file=ConsumerIdentity.certpath(), ) conn.request("PUT", "/rhsm/consumers/%s/tracer" % (ConsumerIdentity.read().getConsumerId()), data, headers=headers) response = conn.getresponse()
def upload_tracer_profile(conduit=False): data = json.dumps({ "traces": get_apps(conduit) }) headers = { "Content-type": "application/json" } conn = httplib.HTTPSConnection( RhsmConfigParser.get(initConfig() ,'server', 'hostname'), RhsmConfigParser.get(initConfig() ,'server', 'port'), key_file = ConsumerIdentity.keypath(), cert_file = ConsumerIdentity.certpath() ) conn.request('PUT', '/rhsm/consumers/%s/tracer' % (ConsumerIdentity.read().getConsumerId()), data, headers=headers) response = conn.getresponse()
class ConfigTests(unittest.TestCase): def setUp(self): # create a temp file for use as a config file. This should get cleaned # up magically at the end of the run. self.fid = NamedTemporaryFile(mode='w+b', suffix='.tmp') self.fid.write(TEST_CONFIG) self.fid.seek(0) self.cfgParser = RhsmConfigParser(self.fid.name) def testRead(self): self.assertEquals(self.cfgParser.get('server', 'hostname'), 'server.example.conf') def testSet(self): self.cfgParser.set('rhsm', 'baseurl', 'cod') self.assertEquals(self.cfgParser.get('rhsm', 'baseurl'), 'cod') def test_get(self): value = self.cfgParser.get("rhsm", "baseurl") self.assertEquals("https://content.example.com", value) def test_get_empty(self): value = self.cfgParser.get("foo", "bar") self.assertEquals("", value) def test_get_int(self): value = self.cfgParser.get_int("server", "port") self.assertTrue(isinstance(value, types.IntType)) self.assertEquals(8443, value) def test_get_item_does_not_exist(self): self.assertRaises(NoOptionError, self.cfgParser.get, "rhsm", "this_isnt_a_thing") def test_get_int_un_set(self): value = self.cfgParser.get_int("server", "proxy_port") self.assertEquals(None, value) def test_get_int_does_not_exist(self): self.assertRaises(NoOptionError, self.cfgParser.get_int, "rhsm", "this_isnt_a_thing") def test_get_int_not_an_int(self): self.assertRaises(ValueError, self.cfgParser.get_int, "rhsm", "baseurl") def test_get_int_big_int(self): value = self.cfgParser.get_int("foo", "bigger_than_32_bit") self.assertEquals(21474836470, value) value = self.cfgParser.get_int("foo", "bigger_than_64_bit") self.assertEquals(123456789009876543211234567890, value)
def getRHSMUuid(): """ Tries to get UUID of of this system if it's registered into Subscription manager.""" if RhsmConfigParser and os.path.isfile(DEFAULT_RHSM_CONFIG_FILE): cfg = RhsmConfigParser(config_file=DEFAULT_RHSM_CONFIG_FILE) cert_dir = cfg.get('rhsm', 'consumerCertDir') cert_path = os.path.join(cert_dir, 'cert.pem') if os.path.isfile(cert_path): f = open(cert_path, 'r') cert = X509.load_cert_string(f.read()) f.close() subject = cert.get_subject() return subject.CN return None
class ConfigTests(unittest.TestCase): def setUp(self): # create a temp file for use as a config file. This should get cleaned # up magically at the end of the run. self.fid = NamedTemporaryFile(mode="w+b", suffix=".tmp") self.fid.write(TEST_CONFIG) self.fid.seek(0) self.cfgParser = RhsmConfigParser(self.fid.name) def testRead(self): self.assertEquals(self.cfgParser.get("server", "hostname"), "server.example.conf") def testSet(self): self.cfgParser.set("rhsm", "baseurl", "cod") self.assertEquals(self.cfgParser.get("rhsm", "baseurl"), "cod") def test_get(self): value = self.cfgParser.get("rhsm", "baseurl") self.assertEquals("https://content.example.com", value) def test_get_empty(self): value = self.cfgParser.get("foo", "bar") self.assertEquals("", value) def test_get_int(self): value = self.cfgParser.get_int("server", "port") self.assertTrue(isinstance(value, types.IntType)) self.assertEquals(8443, value) def test_get_item_does_not_exist(self): self.assertRaises(NoOptionError, self.cfgParser.get, "rhsm", "this_isnt_a_thing") def test_get_int_un_set(self): value = self.cfgParser.get_int("server", "proxy_port") self.assertEquals(None, value) def test_get_int_does_not_exist(self): self.assertRaises(NoOptionError, self.cfgParser.get_int, "rhsm", "this_isnt_a_thing") def test_get_int_not_an_int(self): self.assertRaises(ValueError, self.cfgParser.get_int, "rhsm", "baseurl") def test_get_int_big_int(self): value = self.cfgParser.get_int("foo", "bigger_than_32_bit") self.assertEquals(21474836470, value) value = self.cfgParser.get_int("foo", "bigger_than_64_bit") self.assertEquals(123456789009876543211234567890, value)
def setUp(self): # create a temp file for use as a config file. This should get cleaned # up magically at the end of the run. self.fid = NamedTemporaryFile(mode="w+b", suffix=".tmp") self.fid.write(TEST_CONFIG) self.fid.seek(0) self.cfgParser = RhsmConfigParser(self.fid.name)
class ConfigTests(unittest.TestCase): def setUp(self): # create a temp file for use as a config file. This should get cleaned # up magically at the end of the run. self.fid = NamedTemporaryFile(mode='w+b', suffix='.tmp') self.fid.write(TEST_CONFIG) self.fid.seek(0) self.cfgParser = RhsmConfigParser(self.fid.name) def testRead(self): self.assertEquals(self.cfgParser.get('server', 'hostname'), 'server.example.conf') def testSetd(self): self.cfgParser.set('rhsm', 'baseurl', 'cod') self.assertEquals(self.cfgParser.get('rhsm', 'baseurl'), 'cod')
class TestConfigDBusObject(DBusObjectTest, TestUtilsMixin): def setUp(self): super(TestConfigDBusObject, self).setUp() self.proxy = self.proxy_for(ConfigDBusObject.default_dbus_path) self.interface = dbus.Interface(self.proxy, constants.CONFIG_INTERFACE) def dbus_objects(self): self.fid = self.write_temp_file(TEST_CONFIG) self.addCleanup(self.fid.close) self.parser = RhsmConfigParser(self.fid.name) return [(ConfigDBusObject, {"parser": self.parser})] def test_get_all(self): def assertions(*args): result = args[0] self.assertIn("server", result) dbus_method_args = [""] self.dbus_request(assertions, self.interface.GetAll, dbus_method_args) def test_get_property(self): def assertions(*args): result = args[0] self.assertIn("server.example.com", result) dbus_method_args = ["server.hostname", ""] self.dbus_request(assertions, self.interface.Get, dbus_method_args) def test_get_section(self): def assertions(*args): result = args[0] self.assertIn("hostname", result) dbus_method_args = ["server", ""] self.dbus_request(assertions, self.interface.Get, dbus_method_args) def test_set(self): def assertions(*args): self.assertEqual("new", self.parser.get("server", "hostname")) dbus_method_args = ["server.hostname", "new", ""] self.dbus_request(assertions, self.interface.Set, dbus_method_args) def test_set_section_fails(self): dbus_method_args = ["server", "new", ""] with self.assertRaisesRegex(dbus.DBusException, r"Setting an entire section is not.*"): self.dbus_request(None, self.interface.Set, dbus_method_args)
class TestConfigDBusObject(DBusObjectTest, TestUtilsMixin): def setUp(self): super(TestConfigDBusObject, self).setUp() self.proxy = self.proxy_for(ConfigDBusObject.default_dbus_path) self.interface = dbus.Interface(self.proxy, constants.CONFIG_INTERFACE) def dbus_objects(self): self.fid = self.write_temp_file(TEST_CONFIG) self.addCleanup(self.fid.close) self.parser = RhsmConfigParser(self.fid.name) return [(ConfigDBusObject, {'parser': self.parser})] def test_get_all(self): def assertions(*args): result = args[0] self.assertIn("server", result) dbus_method_args = [''] self.dbus_request(assertions, self.interface.GetAll, dbus_method_args) def test_get_property(self): def assertions(*args): result = args[0] self.assertIn('server.example.com', result) dbus_method_args = ['server.hostname', ''] self.dbus_request(assertions, self.interface.Get, dbus_method_args) def test_get_section(self): def assertions(*args): result = args[0] self.assertIn('hostname', result) dbus_method_args = ['server', ''] self.dbus_request(assertions, self.interface.Get, dbus_method_args) def test_set(self): def assertions(*args): self.assertEqual('new', self.parser.get('server', 'hostname')) dbus_method_args = ['server.hostname', 'new', ''] self.dbus_request(assertions, self.interface.Set, dbus_method_args) def test_set_section_fails(self): dbus_method_args = ['server', 'new', ''] with six.assertRaisesRegex(self, dbus.DBusException, r'Setting an entire section is not.*'): self.dbus_request(None, self.interface.Set, dbus_method_args)
def setUp(self): self.fid = write_temp_file(self.cfgfile_data) self.cfgParser = RhsmConfigParser(self.fid.name)
def setUp(self): super(BaseConfigTest, self).setUp() self.fid = self.write_temp_file(TEST_CONFIG) self.parser = RhsmConfigParser(self.fid.name) self.config = Config(self.parser) self.addCleanup(self.fid.close)
def dbus_objects(self): self.fid = self.write_temp_file(TEST_CONFIG) self.addCleanup(self.fid.close) self.parser = RhsmConfigParser(self.fid.name) return [(ConfigDBusObject, {"parser": self.parser})]
def test_persist(self): self.config['foo'] = {'hello': 'world'} self.config.persist() reparsed = RhsmConfigParser(self.fid.name) self.assertEqual('world', reparsed.get('foo', 'hello')) self.assertRaises(NoOptionError, reparsed.get, 'foo', 'quux')
def test_auto_persists(self): config = Config(self.parser, auto_persist=True) config['foo'] = {'hello': 'world'} reparsed = RhsmConfigParser(self.fid.name) self.assertEqual('world', reparsed.get('foo', 'hello')) self.assertRaises(NoOptionError, reparsed.get, 'foo', 'quux')
def test_does_not_auto_persist_by_default(self): config = Config(self.parser, auto_persist=False) config["foo"] = {"hello": "world"} reparsed = RhsmConfigParser(self.fid.name) self.assertEqual("baz", reparsed.get("foo", "quux")) self.assertRaises(NoOptionError, reparsed.get, "foo", "hello")
def test_auto_persists(self): config = Config(self.parser, auto_persist=True) config["foo"] = {"hello": "world"} reparsed = RhsmConfigParser(self.fid.name) self.assertEqual("world", reparsed.get("foo", "hello")) self.assertRaises(NoOptionError, reparsed.get, "foo", "quux")
def test_persist(self): self.config["foo"] = {"hello": "world"} self.config.persist() reparsed = RhsmConfigParser(self.fid.name) self.assertEqual("world", reparsed.get("foo", "hello")) self.assertRaises(NoOptionError, reparsed.get, "foo", "quux")
def test_does_not_auto_persist_by_default(self): config = Config(self.parser, auto_persist=False) config['foo'] = {'hello': 'world'} reparsed = RhsmConfigParser(self.fid.name) self.assertEqual('baz', reparsed.get('foo', 'quux')) self.assertRaises(NoOptionError, reparsed.get, 'foo', 'hello')
def dbus_objects(self): self.fid = self.write_temp_file(TEST_CONFIG) self.addCleanup(self.fid.close) self.parser = RhsmConfigParser(self.fid.name) return [(ConfigDBusObject, {'parser': self.parser})]