mirror of
https://github.com/outbackdingo/patroni.git
synced 2026-08-25 14:53:37 +00:00
Alexander Kukushkin
aa486194ae
In case if it was interrupted by SIGCHLD but scheduled awake time is not reached it will continue sleeping. For all other signals behaviuor is not changed.
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
import psycopg2
|
|
import requests
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
import unittest
|
|
import yaml
|
|
|
|
from governor import Governor, main
|
|
from test_ha import true, false
|
|
from test_postgresql import Postgresql, subprocess_call, psycopg2_connect
|
|
from test_etcd import requests_get, requests_put, requests_delete
|
|
|
|
if sys.hexversion >= 0x03000000:
|
|
import http.server as BaseHTTPServer
|
|
else:
|
|
import BaseHTTPServer
|
|
|
|
|
|
def nop(*args, **kwargs):
|
|
pass
|
|
|
|
|
|
def time_sleep(_):
|
|
raise Exception()
|
|
|
|
|
|
class TestGovernor(unittest.TestCase):
|
|
|
|
def __init__(self, method_name='runTest'):
|
|
self.setUp = self.set_up
|
|
self.tearDown = self.tear_down
|
|
super(TestGovernor, self).__init__(method_name)
|
|
|
|
def set_up(self):
|
|
self.touched = False
|
|
subprocess.call = subprocess_call
|
|
psycopg2.connect = psycopg2_connect
|
|
requests.get = requests_get
|
|
requests.put = requests_put
|
|
requests.delete = requests_delete
|
|
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
|
|
with open('postgres0.yml', 'r') as f:
|
|
config = yaml.load(f)
|
|
self.g = Governor(config)
|
|
|
|
def tear_down(self):
|
|
time.sleep = self.time_sleep
|
|
Postgresql.write_pg_hba = self.write_pg_hba
|
|
Postgresql.write_recovery_conf = self.write_recovery_conf
|
|
|
|
def test_governor_main(self):
|
|
main()
|
|
sys.argv = ['governor.py', 'postgres0.yml']
|
|
time.sleep = time_sleep
|
|
self.assertRaises(Exception, main)
|
|
|
|
def touch_member(self):
|
|
if not self.touched:
|
|
self.touched = True
|
|
return False
|
|
return True
|
|
|
|
def test_governor_initialize(self):
|
|
self.g.etcd.base_client_url = 'http://remote'
|
|
self.g.etcd.client_url
|
|
self.g.postgresql.data_directory_empty = true
|
|
self.g.etcd.race = true
|
|
self.g.initialize()
|
|
self.g.etcd.race = false
|
|
self.g.initialize()
|
|
self.g.postgresql.data_directory_empty = false
|
|
self.g.touch_member = self.touch_member
|
|
self.g.initialize()
|
|
self.g.postgresql.data_directory_empty = true
|
|
time.sleep = time_sleep
|
|
self.g.postgresql.sync_from_leader = false
|
|
self.assertRaises(Exception, self.g.initialize)
|
|
|
|
def test_schedule_next_run(self):
|
|
self.g.next_run = time.time() - self.g.nap_time - 1
|
|
self.g.schedule_next_run()
|