def test__select(self):
from relstorage.adapters.replica import ReplicaSelector
rs = ReplicaSelector(self.fn, 600.0)
rs._select(0)
self.assertEqual(rs._current_replica, 'example.com:1234')
self.assertEqual(rs._current_index, 0)
self.assertEqual(rs._expiration, None)
rs._select(1)
self.assertEqual(rs._current_replica, 'localhost:4321')
self.assertEqual(rs._current_index, 1)
self.assertNotEqual(rs._expiration, None)
def test__select(self):
from relstorage.adapters.replica import ReplicaSelector
rs = ReplicaSelector(self.fn, 600.0)
rs._select(0)
self.assertEqual(rs._current_replica, 'example.com:1234')
self.assertEqual(rs._current_index, 0)
self.assertEqual(rs._expiration, None)
rs._select(1)
self.assertEqual(rs._current_replica, 'localhost:4321')
self.assertEqual(rs._current_index, 1)
self.assertNotEqual(rs._expiration, None)
def test_current(self):
from relstorage.adapters.replica import ReplicaSelector
rs = ReplicaSelector(self.fn, 600.0)
self.assertEqual(rs.current(), 'example.com:1234')
# change the file and get the new current replica
with open(self.fn, 'w') as f:
f.write('localhost\nalternate\n')
rs._config_checked = 0
rs._config_modified = 0
self.assertEqual(rs.current(), 'localhost')
# switch to the alternate
rs._select(1)
self.assertEqual(rs.current(), 'alternate')
# expire the alternate
rs._expiration = 0
self.assertEqual(rs.current(), 'localhost')
def test_current(self):
from relstorage.adapters.replica import ReplicaSelector
rs = ReplicaSelector(self.fn, 600.0)
self.assertEqual(rs.current(), 'example.com:1234')
# change the file and get the new current replica
f = open(self.fn, 'w')
f.write('localhost\nalternate\n')
f.close()
rs._config_checked = 0
rs._config_modified = 0
self.assertEqual(rs.current(), 'localhost')
# switch to the alternate
rs._select(1)
self.assertEqual(rs.current(), 'alternate')
# expire the alternate
rs._expiration = 0
self.assertEqual(rs.current(), 'localhost')
