class TestCertGeneration(TestCase): @patch.object(OSCheck, "os_distribution", new=MagicMock(return_value=os_distro_value)) def setUp(self): self.tmpdir = tempfile.mkdtemp() config = AmbariConfig.AmbariConfig() #config.add_section('server') config.set('server', 'hostname', 'example.com') config.set('server', 'url_port', '777') #config.add_section('security') config.set('security', 'keysdir', self.tmpdir) config.set('security', 'server_crt', 'ca.crt') server_hostname = config.get('server', 'hostname') self.certMan = CertificateManager(config, server_hostname) @patch.object(os, "chmod") def test_generation(self, chmod_mock): self.certMan.genAgentCrtReq('/dummy_dir/hostname.key') self.assertTrue(chmod_mock.called) self.assertTrue(os.path.exists(self.certMan.getAgentKeyName())) self.assertTrue(os.path.exists(self.certMan.getAgentCrtReqName())) def tearDown(self): shutil.rmtree(self.tmpdir)
def test_genAgentCrtReq(self, communicate_mock, popen_mock): man = CertificateManager(self.config) p = MagicMock(spec=subprocess.Popen) p.communicate = communicate_mock popen_mock.return_value = p man.genAgentCrtReq() self.assertTrue(popen_mock.called) self.assertTrue(communicate_mock.called)
def test_genAgentCrtReq(self, communicate_mock, popen_mock): man = CertificateManager(self.config) p = MagicMock(spec=subprocess.Popen) p.communicate = communicate_mock popen_mock.return_value = p man.genAgentCrtReq() self.assertTrue(popen_mock.called) self.assertTrue(communicate_mock.called)
def test_genAgentCrtReq(self, chmod_mock, communicate_mock, popen_mock): man = CertificateManager(self.config, "active_server") p = MagicMock(spec=subprocess32.Popen) p.communicate = communicate_mock popen_mock.return_value = p man.genAgentCrtReq('/dummy-keysdir/hostname.key') self.assertTrue(chmod_mock.called) self.assertTrue(popen_mock.called) self.assertTrue(communicate_mock.called)
class TestCertGeneration(TestCase): @patch.object(OSCheck, "os_distribution", new = MagicMock(return_value = os_distro_value)) def setUp(self): self.tmpdir = tempfile.mkdtemp() config = AmbariConfig.AmbariConfig() config.set('server', 'hostname', 'example.com') config.set('server', 'url_port', '777') config.set('security', 'keysdir', self.tmpdir) config.set('security', 'server_crt', 'ca.crt') server_hostname = config.get('server', 'hostname') self.certMan = CertificateManager(config, server_hostname) @patch.object(os, "chmod") def test_generation(self, chmod_mock): self.certMan.genAgentCrtReq('/dummy_dir/hostname.key') self.assertTrue(chmod_mock.called) self.assertTrue(os.path.exists(self.certMan.getAgentKeyName())) self.assertTrue(os.path.exists(self.certMan.getAgentCrtReqName())) def tearDown(self): shutil.rmtree(self.tmpdir)