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())