class Test(unittest.TestCase): def setUp (self): self._queryenv = QueryEnvService("http://ec2farm-dev.bsd2.webta.local/query-env/","127","c+/g0PyouaqXMbuJ5Vtux34Mft7jLe5H5u8tUmyhldjwTfgm7BI6MOA8F6BwkzQnpWEOcHx+A+TRJh0u3PElQQ0SiwdwrlgpQMbj8NBxbxBgfxA9WisgvfQu5ZPYou6Gz3oUAQdWfFlFdY2ACOjmqa3DGogge+TlXtV2Xagm0rw=") def test_get_service_configuration_response(self): xmlfile = get_xml_file("get-service-configuration_response.xml") xml = xml_strip(parseString(open(xmlfile, "r").read())) ret = self._queryenv._read_get_service_configuration_response(xml, 'app') self.assertFalse(ret is None) self.assertTrue(type(ret.settings) is dict) self.assertTrue(len(ret.settings)) self.assertTrue('__defaults__' in ret.settings) self.assertTrue(type(ret.name) is type("")) self.assertEqual(ret.name, "app-test") def test_get_latest_version_response(self): xmlfile = get_xml_file("get_latest_version_response.xml") xml = xml_strip(parseString(open(xmlfile, "r").read())) version = self._queryenv._read_get_latest_version_response(xml) self.assertFalse(version is None) self.assertEqual(version, "2009-03-05") def test_get_https_certificate_response(self): xmlfile = get_xml_file("get_https_certificate_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) cert = self._queryenv._read_get_https_certificate_response(xml) self.assertFalse(cert is (None,None)) self.assertEqual(cert[0], """-----BEGIN CERTIFICATE----- MIICATCCAWoCCQDVWoPxl2kzdzANBgkqhkiG9w0BAQUFADBFMQswCQYDVQQGEwJB VTETMBEGA1UECBMKU29tZS1TdGF0ZTEhMB8GA1UEChMYSW50ZXJuZXQgV2lkZ2l0 cyBQdHkgTHRkMB4XDTEwMDYwNDEyMDcyNloXDTExMDYwNDEyMDcyNlowRTELMAkG A1UEBhMCQVUxEzARBgNVBAgTClNvbWUtU3RhdGUxITAfBgNVBAoTGEludGVybmV0 IFdpZGdpdHMgUHR5IEx0ZDCBnzANBgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEA5DeJ DnVd/Jgb3bPh296l78V/T7kRG8IAejawlR25ikUXpG9DoaV05nDwDR+e3Hm/wvmY 7EwqOiVFtFc1Uoc6av34zIroEcGmwi+nnUAptpHOC/863VhuSehKf7lWkUbLz8OS NAJOj5c6jxKfasnkqxYEakhvXCe/N4NQWdnSaVUCAwEAATANBgkqhkiG9w0BAQUF AAOBgQAJ318EcccEM34pFsF5thGQfYU4yrTb2P+Xyg/bhNgByCc+agWh9MAXBF1G EEH5rvsHc1ocVXi69a45D+m0pV5ZJwSIrwo6ssViMpWmUfIStkmYm7qsRbNnIWkZ TuMUjy2djQJdeAadKNinJ5YXk2iU7XvxTzaqZAzdpv2/9G5Nbg== -----END CERTIFICATE----- """) self.assertEqual(cert[1], """-----BEGIN RSA PRIVATE KEY----- MIICXgIBAAKBgQDkN4kOdV38mBvds+Hb3qXvxX9PuREbwgB6NrCVHbmKRRekb0Oh pXTmcPANH57ceb/C+ZjsTCo6JUW0VzVShzpq/fjMiugRwabCL6edQCm2kc4L/zrd WG5J6Ep/uVaRRsvPw5I0Ak6PlzqPEp9qyeSrFgRqSG9cJ783g1BZ2dJpVQIDAQAB AoGBAKNtDY24CIUBHDtoPF4qE6QfRur9OT4qcBMUpOEztxFIJwUO1FymUo9O4uhS 830pBmSGPrdAV6Dp3f+lz754riBj1Hzk5kQuF6rAxoBspbqXQB30Pz1r6qWt9Sf2 DZqQ9278UZEtUQq90QzEjh3xAV5BxG1Qv+d3yyIVy1K5Pt7BAkEA957XKSdGpQW5 c2V6rUnfvLi/WeIx8xFFL1ohnPSBLhvP8Hc4zJGVVBnuX9RKEVVLCizRBLFK93uI nh0aGSa9hQJBAOvwmamWMkFH4aYymqZY5CRooxCG6Wv8MTQBhgGaLyGYX/CaHpRf Y5RCEWnXZJr/rAwnowLr8kh/MiZGIBWVHZECQQDBeSRYDU4PRke+OD4AA8aC6D7q defdKVNLSjsVLZ15b1WrZxvECsQIcDJmQbKVlHULQDUYW4Zdk/IMyGRJ3pEZAkEA jWaz4RQX6FHZJY7cameJy1w+phAE4ufQ4TcshddO+dZlYUAspYWJm3gBEaq6K76g 8OPsaTrZCKPafV+3qNemUQJACZ7FDJKmO9SLccpYIDTcMIhqgu2QseZwjJPjAMbg 0xHR3hyeMnkrjP9amXrbAxOwndCD10I6Tuw4Qj7t20p+hw== -----END RSA PRIVATE KEY----- """) def test_list_roles(self): xmlfile = get_xml_file("list_roles_response.xml") xml = parseString(open(xmlfile, "r").read()) #print open(xmlfile, "r").read() xml = xml_strip(xml) roles = self._queryenv._read_list_roles_response(xml) role = roles[0] self.assertFalse(roles is None) self.assertEqual(len(roles), 2) self.assertEqual(role.behaviour, ['app', "mysql"]) self.assertEqual(role.name, "lamp-custom") hosts = role.hosts host = hosts[0] self.assertFalse(hosts is None) self.assertEqual(host.internal_ip, "211.31.14.198") self.assertEqual(host.external_ip, "211.31.14.198") self.assertTrue(host.replication_master) self.assertEqual(host.index, 1) def test_read_list_ebs_mountpoints_response(self): xmlfile = get_xml_file("list_ebs_mountpoints_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) mountpoints = self._queryenv._read_list_ebs_mountpoints_response(xml) mountpoint = mountpoints[0] volumes = mountpoint.volumes volume = volumes[0] self.assertFalse(mountpoints is None) self.assertEqual(len(mountpoints), 2) self.assertEqual(mountpoint.name, "some_name_for_LVM") self.assertEqual(mountpoint.dir, "/mnt/storage1") self.assertTrue(mountpoint.create_fs) self.assertFalse(mountpoint.is_array) self.assertFalse(volumes is None) self.assertEqual(volume.volume_id, "vol-123451") self.assertEqual(volume.device, "/dev/sdb") def test_list_role_params(self): xmlfile = get_xml_file("list_role_params_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) parameters = self._queryenv._read_list_role_params_response(xml) self.assertFalse(parameters is None) #self.assertTrue(parametres.has_key("external_ips_to_allow_access_from")) self.assertEqual(parameters["external_ips_to_allow_access_from"].strip(), '') def test_read_list_scripts_response(self): xmlfile = get_xml_file("list_scripts_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) scripts = self._queryenv._read_list_scripts_response(xml) script = scripts[0] self.assertFalse(scripts is None) self.assertEqual(len(scripts), 1) self.assertTrue(script.asynchronous) self.assertEqual(script.exec_timeout, 100) self.assertEqual(script.name, 'script_name') self.assertEqual(script.body.strip(), '') def test_read_list_virtualhosts_response(self): xmlfile = get_xml_file("list_virtualhosts_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) vhosts = self._queryenv._read_list_virtualhosts_response(xml) vhost = vhosts[0] self.assertFalse(vhosts is None) self.assertEqual(len(vhosts), 2) self.assertEqual(vhost.hostname, 'gpanel.net') self.assertEqual(vhost.type, 'apache') self.assertEqual(vhost.raw, ''' ''') self.assertTrue(vhosts[1].https) def test__read_list_farm_role_params_response(self): xmlfile = get_xml_file("list_farm_role_params_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) params = self._queryenv._read_list_farm_role_params_response(xml) print params def test_xml2dict(self): result = {'mysql': {'root_password': '******', 'stat_password': '******', 'volume_config': {'device': '/dev/xvdp', 'mpoint': '/mnt/dbstorage', 'type': 'ebs', 'id': 'vol-12345678', 'size': '100'}, 'log_file': 'binlog.000003', 'repl_password': '******', 'log_pos': '106'}} xmlfile = get_xml_file('xml2dict.xml') self.assertEqual(xml_file_to_dict(xmlfile), result) print 'done'
class Test(unittest.TestCase): def setUp(self): self._queryenv = QueryEnvService( "http://ec2farm-dev.bsd2.webta.local/query-env/", "127", "c+/g0PyouaqXMbuJ5Vtux34Mft7jLe5H5u8tUmyhldjwTfgm7BI6MOA8F6BwkzQnpWEOcHx+A+TRJh0u3PElQQ0SiwdwrlgpQMbj8NBxbxBgfxA9WisgvfQu5ZPYou6Gz3oUAQdWfFlFdY2ACOjmqa3DGogge+TlXtV2Xagm0rw=" ) def test_get_service_configuration_response(self): xmlfile = get_xml_file("get-service-configuration_response.xml") xml = xml_strip(parseString(open(xmlfile, "r").read())) ret = self._queryenv._read_get_service_configuration_response( xml, 'app') self.assertFalse(ret is None) self.assertTrue(type(ret.settings) is dict) self.assertTrue(len(ret.settings)) self.assertTrue('__defaults__' in ret.settings) self.assertTrue(type(ret.name) is type("")) self.assertEqual(ret.name, "app-test") def test_get_latest_version_response(self): xmlfile = get_xml_file("get_latest_version_response.xml") xml = xml_strip(parseString(open(xmlfile, "r").read())) version = self._queryenv._read_get_latest_version_response(xml) self.assertFalse(version is None) self.assertEqual(version, "2009-03-05") def test_get_https_certificate_response(self): xmlfile = get_xml_file("get_https_certificate_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) cert = self._queryenv._read_get_https_certificate_response(xml) self.assertFalse(cert is (None, None)) self.assertEqual( cert[0], """-----BEGIN CERTIFICATE----- MIICATCCAWoCCQDVWoPxl2kzdzANBgkqhkiG9w0BAQUFADBFMQswCQYDVQQGEwJB VTETMBEGA1UECBMKU29tZS1TdGF0ZTEhMB8GA1UEChMYSW50ZXJuZXQgV2lkZ2l0 cyBQdHkgTHRkMB4XDTEwMDYwNDEyMDcyNloXDTExMDYwNDEyMDcyNlowRTELMAkG A1UEBhMCQVUxEzARBgNVBAgTClNvbWUtU3RhdGUxITAfBgNVBAoTGEludGVybmV0 IFdpZGdpdHMgUHR5IEx0ZDCBnzANBgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEA5DeJ DnVd/Jgb3bPh296l78V/T7kRG8IAejawlR25ikUXpG9DoaV05nDwDR+e3Hm/wvmY 7EwqOiVFtFc1Uoc6av34zIroEcGmwi+nnUAptpHOC/863VhuSehKf7lWkUbLz8OS NAJOj5c6jxKfasnkqxYEakhvXCe/N4NQWdnSaVUCAwEAATANBgkqhkiG9w0BAQUF AAOBgQAJ318EcccEM34pFsF5thGQfYU4yrTb2P+Xyg/bhNgByCc+agWh9MAXBF1G EEH5rvsHc1ocVXi69a45D+m0pV5ZJwSIrwo6ssViMpWmUfIStkmYm7qsRbNnIWkZ TuMUjy2djQJdeAadKNinJ5YXk2iU7XvxTzaqZAzdpv2/9G5Nbg== -----END CERTIFICATE----- """) self.assertEqual( cert[1], """-----BEGIN RSA PRIVATE KEY----- MIICXgIBAAKBgQDkN4kOdV38mBvds+Hb3qXvxX9PuREbwgB6NrCVHbmKRRekb0Oh pXTmcPANH57ceb/C+ZjsTCo6JUW0VzVShzpq/fjMiugRwabCL6edQCm2kc4L/zrd WG5J6Ep/uVaRRsvPw5I0Ak6PlzqPEp9qyeSrFgRqSG9cJ783g1BZ2dJpVQIDAQAB AoGBAKNtDY24CIUBHDtoPF4qE6QfRur9OT4qcBMUpOEztxFIJwUO1FymUo9O4uhS 830pBmSGPrdAV6Dp3f+lz754riBj1Hzk5kQuF6rAxoBspbqXQB30Pz1r6qWt9Sf2 DZqQ9278UZEtUQq90QzEjh3xAV5BxG1Qv+d3yyIVy1K5Pt7BAkEA957XKSdGpQW5 c2V6rUnfvLi/WeIx8xFFL1ohnPSBLhvP8Hc4zJGVVBnuX9RKEVVLCizRBLFK93uI nh0aGSa9hQJBAOvwmamWMkFH4aYymqZY5CRooxCG6Wv8MTQBhgGaLyGYX/CaHpRf Y5RCEWnXZJr/rAwnowLr8kh/MiZGIBWVHZECQQDBeSRYDU4PRke+OD4AA8aC6D7q defdKVNLSjsVLZ15b1WrZxvECsQIcDJmQbKVlHULQDUYW4Zdk/IMyGRJ3pEZAkEA jWaz4RQX6FHZJY7cameJy1w+phAE4ufQ4TcshddO+dZlYUAspYWJm3gBEaq6K76g 8OPsaTrZCKPafV+3qNemUQJACZ7FDJKmO9SLccpYIDTcMIhqgu2QseZwjJPjAMbg 0xHR3hyeMnkrjP9amXrbAxOwndCD10I6Tuw4Qj7t20p+hw== -----END RSA PRIVATE KEY----- """) def test_list_roles(self): xmlfile = get_xml_file("list_roles_response.xml") xml = parseString(open(xmlfile, "r").read()) #print open(xmlfile, "r").read() xml = xml_strip(xml) roles = self._queryenv._read_list_roles_response(xml) role = roles[0] self.assertFalse(roles is None) self.assertEqual(len(roles), 2) self.assertEqual(role.behaviour, ['app', "mysql"]) self.assertEqual(role.name, "lamp-custom") hosts = role.hosts host = hosts[0] self.assertFalse(hosts is None) self.assertEqual(host.internal_ip, "211.31.14.198") self.assertEqual(host.external_ip, "211.31.14.198") self.assertTrue(host.replication_master) self.assertEqual(host.index, 1) def test_read_list_ebs_mountpoints_response(self): xmlfile = get_xml_file("list_ebs_mountpoints_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) mountpoints = self._queryenv._read_list_ebs_mountpoints_response(xml) mountpoint = mountpoints[0] volumes = mountpoint.volumes volume = volumes[0] self.assertFalse(mountpoints is None) self.assertEqual(len(mountpoints), 2) self.assertEqual(mountpoint.name, "some_name_for_LVM") self.assertEqual(mountpoint.dir, "/mnt/storage1") self.assertTrue(mountpoint.create_fs) self.assertFalse(mountpoint.is_array) self.assertFalse(volumes is None) self.assertEqual(volume.volume_id, "vol-123451") self.assertEqual(volume.device, "/dev/sdb") def test_list_role_params(self): xmlfile = get_xml_file("list_role_params_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) parameters = self._queryenv._read_list_role_params_response(xml) self.assertFalse(parameters is None) #self.assertTrue(parametres.has_key("external_ips_to_allow_access_from")) self.assertEqual( parameters["external_ips_to_allow_access_from"].strip(), '') def test_read_list_scripts_response(self): xmlfile = get_xml_file("list_scripts_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) scripts = self._queryenv._read_list_scripts_response(xml) script = scripts[0] self.assertFalse(scripts is None) self.assertEqual(len(scripts), 1) self.assertTrue(script.asynchronous) self.assertEqual(script.exec_timeout, 100) self.assertEqual(script.name, 'script_name') self.assertEqual(script.body.strip(), '') def test_read_list_virtualhosts_response(self): xmlfile = get_xml_file("list_virtualhosts_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) vhosts = self._queryenv._read_list_virtualhosts_response(xml) vhost = vhosts[0] self.assertFalse(vhosts is None) self.assertEqual(len(vhosts), 2) self.assertEqual(vhost.hostname, 'gpanel.net') self.assertEqual(vhost.type, 'apache') self.assertEqual(vhost.raw, ''' ''') self.assertTrue(vhosts[1].https) def test__read_list_farm_role_params_response(self): xmlfile = get_xml_file("list_farm_role_params_response.xml") xml = parseString(open(xmlfile, "r").read()) xml = xml_strip(xml) params = self._queryenv._read_list_farm_role_params_response(xml) print params def test_xml2dict(self): result = { 'mysql': { 'root_password': '******', 'stat_password': '******', 'volume_config': { 'device': '/dev/xvdp', 'mpoint': '/mnt/dbstorage', 'type': 'ebs', 'id': 'vol-12345678', 'size': '100' }, 'log_file': 'binlog.000003', 'repl_password': '******', 'log_pos': '106' } } xmlfile = get_xml_file('xml2dict.xml') self.assertEqual(xml_file_to_dict(xmlfile), result) print 'done'