示例#1
0
 def setUp(self):
     self._handlers = logging.getLogger().handlers[:]
     RestApiServer._BaseServer__is_shut_down = Mock()
     RestApiServer._BaseServer__shutdown_request = True
     RestApiServer.socket = 0
     os.environ['PATRONI_POSTGRESQL_DATA_DIR'] = 'data/test0'
     self.p = Patroni()
示例#2
0
 def setUp(self):
     RestApiServer._BaseServer__is_shut_down = Mock()
     RestApiServer._BaseServer__shutdown_request = True
     RestApiServer.socket = 0
     with patch.object(Client, 'machines') as mock_machines:
         mock_machines.__get__ = Mock(return_value=['http://remotehost:2379'])
         sys.argv = ['patroni.py', 'postgres0.yml']
         self.p = Patroni()
示例#3
0
 def setUp(self):
     RestApiServer._BaseServer__is_shut_down = Mock()
     RestApiServer._BaseServer__shutdown_request = True
     RestApiServer.socket = 0
     with patch.object(etcd.Client, 'machines') as mock_machines:
         mock_machines.__get__ = Mock(
             return_value=['http://remotehost:2379'])
         with open('postgres0.yml', 'r') as f:
             config = yaml.load(f)
             self.p = Patroni(config)
示例#4
0
 def setUp(self):
     self._handlers = logging.getLogger().handlers[:]
     RestApiServer._BaseServer__is_shut_down = Mock()
     RestApiServer._BaseServer__shutdown_request = True
     RestApiServer.socket = 0
     with patch.object(Client, 'machines') as mock_machines:
         mock_machines.__get__ = Mock(
             return_value=['http://remotehost:2379'])
         sys.argv = ['patroni.py', 'postgres0.yml']
         os.environ['PATRONI_POSTGRESQL_DATA_DIR'] = 'data/test0'
         self.p = Patroni()
示例#5
0
 def setUp(self, mock_machines):
     mock_machines.__get__ = Mock(return_value=['http://remotehost:2379'])
     self.touched = False
     self.init_cancelled = False
     RestApiServer._BaseServer__is_shut_down = Mock()
     RestApiServer._BaseServer__shutdown_request = True
     RestApiServer.socket = 0
     with open('postgres0.yml', 'r') as f:
         config = yaml.load(f)
         self.p = Patroni(config)
         self.p.ha.dcs.client.write = etcd_write
         self.p.ha.dcs.client.read = etcd_read
示例#6
0
 def set_up(self):
     self.touched = False
     subprocess.call = subprocess_call
     psycopg2.connect = psycopg2_connect
     self.time_sleep = time.sleep
     time.sleep = nop
     self.write_pg_hba = Postgresql.write_pg_hba
     self.write_recovery_conf = Postgresql.write_recovery_conf
     Postgresql.write_pg_hba = nop
     Postgresql.write_recovery_conf = nop
     BaseHTTPServer.HTTPServer.__init__ = nop
     RestApiServer._BaseServer__is_shut_down = Mock_BaseServer__is_shut_down(
     )
     RestApiServer._BaseServer__shutdown_request = True
     RestApiServer.socket = 0
     with open('postgres0.yml', 'r') as f:
         config = yaml.load(f)
         with patch.object(Client, 'machines') as mock_machines:
             mock_machines.__get__ = Mock(
                 return_value=['http://remotehost:2379'])
             self.p = Patroni(config)