def test_create_db(self): mock_status = MagicMock() App = VerticaApp(status=mock_status) App._get_database_password = MagicMock(return_value='text') App.create_db() self.assertTrue(App._get_database_password.called) self.assertEqual(subprocess.check_call.call_count, 1)
def test_get_database_password(self): mock_status = MagicMock() App = VerticaApp(status=mock_status) passwd = App._get_database_password() self.assertEqual(utils.execute_with_timeout.call_count, 1) self.assertEqual(passwd, "password")