def test_with_replica_conf(self):
import os
import relstorage.tests
tests_dir = relstorage.tests.__file__
replica_conf = os.path.join(os.path.dirname(tests_dir), 'replicas.conf')
options = MockOptions(replica_conf)
from relstorage.adapters.connmanager \
import AbstractConnectionManager
from relstorage.adapters.interfaces import ReplicaClosedException
cm = AbstractConnectionManager(options)
conn = MockConnection()
conn.replica = 'localhost'
cm.restart_load(conn, MockCursor())
self.assertTrue(conn.rolled_back)
conn.replica = 'other'
self.assertRaises(ReplicaClosedException,
cm.restart_load, conn, MockCursor())
conn = MockConnection()
conn.replica = 'localhost'
cm.restart_store(conn, MockCursor())
self.assertTrue(conn.rolled_back)
conn.replica = 'other'
self.assertRaises(ReplicaClosedException,
cm.restart_store, conn, MockCursor())
def test_without_replica_conf(self):
from relstorage.adapters.connmanager import AbstractConnectionManager
cm = AbstractConnectionManager(MockOptions())
conn = MockConnection()
cm.restart_load(conn, MockCursor())
self.assertTrue(conn.rolled_back)
conn = MockConnection()
cm.restart_store(conn, MockCursor())
self.assertTrue(conn.rolled_back)
def test_with_replica_conf(self):
import os
import relstorage.tests
tests_dir = relstorage.tests.__file__
replica_conf = os.path.join(os.path.dirname(tests_dir),
'replicas.conf')
options = MockOptions(replica_conf)
from relstorage.adapters.connmanager \
import AbstractConnectionManager
from relstorage.adapters.interfaces import ReplicaClosedException
cm = AbstractConnectionManager(options)
conn = MockConnection()
conn.replica = 'localhost'
cm.restart_load(conn, MockCursor())
self.assertTrue(conn.rolled_back)
conn.replica = 'other'
self.assertRaises(ReplicaClosedException, cm.restart_load, conn,
MockCursor())
conn = MockConnection()
conn.replica = 'localhost'
cm.restart_store(conn, MockCursor())
self.assertTrue(conn.rolled_back)
conn.replica = 'other'
self.assertRaises(ReplicaClosedException, cm.restart_store, conn,
MockCursor())
def test_without_replica_conf(self):
from relstorage.adapters.connmanager import AbstractConnectionManager
cm = AbstractConnectionManager(MockOptions())
conn = MockConnection()
cm.restart_load(conn, MockCursor())
self.assertTrue(conn.rolled_back)
conn = MockConnection()
cm.restart_store(conn, MockCursor())
self.assertTrue(conn.rolled_back)
def _makeOne(self, **options):
return AbstractConnectionManager(MockOptions.from_args(**options),
MockDriver())
