def test_remote_sessions_can_continue_on_a_different_netman(self):
with NetmanTestApp() as partial_client1, NetmanTestApp() as partial_client2:
switch_descriptor = get_available_switch("juniper")
client1 = partial_client1(switch_descriptor)
first_netman_url = "{}:{}".format(client1.host, client1.port)
client2 = partial_client2(switch_descriptor)
second_netman_url = "{}:{}".format(client2.host, client2.port)
remote_switch = RemoteSwitch(switch_descriptor)
remote_switch._proxy = first_netman_url
switch = FlowControlSwitch(remote_switch, ThreadingLockFactory().new_lock())
with switch.transaction():
switch.add_vlan(1498, "one")
remote_switch._proxy = second_netman_url
switch.add_vlan(1499, "two")
assert_that(client1.get("/switches/{hostname}/vlans/1498").json()["name"], is_("one"))
assert_that(client1.get("/switches/{hostname}/vlans/1499").json()["name"], is_("two"))
def setUp(self):
self.wrapped_switch = flexmock(
SwitchBase(SwitchDescriptor("cisco", "name")))
self.lock = flexmock()
self.switch = FlowControlSwitch(self.wrapped_switch, self.lock)
class FlowControlSwitchTest(TestCase):
def setUp(self):
self.wrapped_switch = flexmock(
SwitchBase(SwitchDescriptor("cisco", "name")))
self.lock = flexmock()
self.switch = FlowControlSwitch(self.wrapped_switch, self.lock)
def tearDown(self):
flexmock_teardown()
def test_a_get_method_connects_and_executes(self):
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"get_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.switch.get_vlan(1000)
def test_a_get_method_still_dc_if_raising(self):
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("get_vlan").once().ordered(
).with_args(1000).and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
with self.assertRaises(NetmanException):
self.switch.get_vlan(1000)
def test_a_get_method_connect_fail_gives_up(self):
self.wrapped_switch.should_receive(
"_connect").once().ordered().and_raise(NetmanException)
with self.assertRaises(NetmanException):
self.switch.get_vlan(1000)
def test_a_get_method_does_not_connect_if_already_connected(self):
self.wrapped_switch.connected = True
self.wrapped_switch.should_receive(
"get_vlan").once().ordered().with_args(1000)
self.switch.get_vlan(1000)
def test_an_operation_method_locks_connects_and_executes_in_a_transaction(
self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive(
"commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.add_vlan(1000)
def test_an_operation_method_unlocks_and_rollback_if_raising(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered(
).with_args(1000).and_raise(NetmanException)
self.wrapped_switch.should_receive(
"rollback_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_start_transaction_rollbacks_unlocks_dc(
self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_end_transaction_unlocks_dc(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive(
"commit_transaction").once().ordered()
self.wrapped_switch.should_receive(
"_end_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_commit_transaction_unlocks_dc(
self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive(
"commit_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive(
"rollback_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_disconnect_unlocks(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive(
"commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive(
"_disconnect").once().ordered().and_raise(NetmanException)
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_connect_unlocks(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive(
"_connect").once().ordered().and_raise(NetmanException)
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_lock_gives_up(self):
self.lock.should_receive("acquire").once().ordered().and_raise(
NetmanException)
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_already_connected_locks_and_executes_in_a_transaction(
self):
self.wrapped_switch.connected = True
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").never()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive(
"commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").never()
self.lock.should_receive("release").once().ordered()
self.switch.add_vlan(1000)
def test_an_operation_method_already_connected_and_in_transaction_just_executes(
self):
self.wrapped_switch.connected = True
self.wrapped_switch.in_transaction = True
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
self.switch.add_vlan(1000)
def test_transaction_context_locks_connect_and_execute(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive(
"commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.switch.transaction():
self.switch.add_vlan(1000)
def test_transaction_context_unlocks_rollback_and_dc_if_fails(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered(
).with_args(1000).and_raise(NetmanException)
self.wrapped_switch.should_receive(
"rollback_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
with self.switch.transaction():
self.switch.add_vlan(1000)
def test_transaction_context_already_connected_only_locks_connect_and_execute(
self):
self.wrapped_switch.connected = True
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive(
"commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.switch.transaction():
self.switch.add_vlan(1000)
def test_transaction_context_already_in_transaction_only_executes(self):
self.wrapped_switch.connected = True
self.wrapped_switch.in_transaction = True
self.wrapped_switch.should_receive(
"add_vlan").once().ordered().with_args(1000)
with self.switch.transaction():
self.switch.add_vlan(1000)
def test_switch_contract_compliance_start_transaction(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_failing_to_start_rollback_dc_unlocks(
self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_failing_to_connect_unlocks(
self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive(
"_connect").once().ordered().and_raise(NetmanException)
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_failing_to_lock_gives_up(
self):
self.lock.should_receive("acquire").once().ordered().and_raise(
NetmanException)
with self.assertRaises(NetmanException):
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_already_connected_don_t_connect(
self):
self.wrapped_switch.connected = True
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_already_in_transaction_does_nothing(
self):
self.wrapped_switch.connected = True
self.wrapped_switch.in_transaction = True
self.switch.start_transaction()
def test_switch_contract_compliance_end_transaction(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.end_transaction()
def test_switch_contract_compliance_end_transaction_dont_dc_for_a_legit_connection_after_a_managed_one(
self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.end_transaction()
self.wrapped_switch.connected = True
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.end_transaction()
def test_switch_contract_compliance_end_transaction_if_not_connected_by_start_dont_dc(
self):
self.wrapped_switch.connected = True
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.end_transaction()
def test_switch_contract_compliance_end_transaction_failing_end_unlocks_and_dc(
self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive(
"_end_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.end_transaction()
def test_switch_contract_compliance_end_transaction_failing_dc_unlocks(
self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive(
"_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive(
"_disconnect").once().ordered().and_raise(NetmanException)
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.end_transaction()
def test_switch_contract_compliance_connect(self):
self.wrapped_switch.should_receive("connect").once().ordered()
self.switch.connect()
def test_switch_contract_compliance_disconnect(self):
self.wrapped_switch.should_receive("disconnect").once().ordered()
self.switch.disconnect()
def test_switch_contract_compliance_commit_transaction(self):
self.wrapped_switch.should_receive(
"commit_transaction").once().ordered()
self.switch.commit_transaction()
def test_switch_contract_compliance_rollback_transaction(self):
self.wrapped_switch.should_receive(
"rollback_transaction").once().ordered()
self.switch.rollback_transaction()
def test_switch_contract_compliance_switch_descriptor(self):
assert_that(self.switch.switch_descriptor,
is_(self.wrapped_switch.switch_descriptor))
def get_switch_by_descriptor(self, switch_descriptor):
real_switch = super(FlowControlSwitchFactory,
self).get_switch_by_descriptor(switch_descriptor)
return FlowControlSwitch(real_switch,
lock=self._get_lock(switch_descriptor))
def setUp(self):
self.wrapped_switch = flexmock(SwitchBase(SwitchDescriptor("cisco", "name")))
self.lock = flexmock()
self.switch = FlowControlSwitch(self.wrapped_switch, self.lock)
class FlowControlSwitchTest(TestCase):
def setUp(self):
self.wrapped_switch = flexmock(SwitchBase(SwitchDescriptor("cisco", "name")))
self.lock = flexmock()
self.switch = FlowControlSwitch(self.wrapped_switch, self.lock)
def tearDown(self):
flexmock_teardown()
def test_a_get_method_connects_and_executes(self):
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("get_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.switch.get_vlan(1000)
def test_a_get_method_still_dc_if_raising(self):
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("get_vlan").once().ordered().with_args(1000).and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
with self.assertRaises(NetmanException):
self.switch.get_vlan(1000)
def test_a_get_method_connect_fail_gives_up(self):
self.wrapped_switch.should_receive("_connect").once().ordered().and_raise(NetmanException)
with self.assertRaises(NetmanException):
self.switch.get_vlan(1000)
def test_a_get_method_does_not_connect_if_already_connected(self):
self.wrapped_switch.connected = True
self.wrapped_switch.should_receive("get_vlan").once().ordered().with_args(1000)
self.switch.get_vlan(1000)
def test_an_operation_method_locks_connects_and_executes_in_a_transaction(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.add_vlan(1000)
def test_an_operation_method_unlocks_and_rollback_if_raising(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000).and_raise(NetmanException)
self.wrapped_switch.should_receive("rollback_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_start_transaction_rollbacks_unlocks_dc(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_end_transaction_unlocks_dc(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_commit_transaction_unlocks_dc(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("commit_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("rollback_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_disconnect_unlocks(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered().and_raise(NetmanException)
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_connect_unlocks(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered().and_raise(NetmanException)
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_failing_to_lock_gives_up(self):
self.lock.should_receive("acquire").once().ordered().and_raise(NetmanException)
with self.assertRaises(NetmanException):
self.switch.add_vlan(1000)
def test_an_operation_method_already_connected_locks_and_executes_in_a_transaction(self):
self.wrapped_switch.connected = True
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").never()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").never()
self.lock.should_receive("release").once().ordered()
self.switch.add_vlan(1000)
def test_an_operation_method_already_connected_and_in_transaction_just_executes(self):
self.wrapped_switch.connected = True
self.wrapped_switch.in_transaction = True
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
self.switch.add_vlan(1000)
def test_transaction_context_locks_connect_and_execute(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.switch.transaction():
self.switch.add_vlan(1000)
def test_transaction_context_unlocks_rollback_and_dc_if_fails(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000).and_raise(NetmanException)
self.wrapped_switch.should_receive("rollback_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
with self.switch.transaction():
self.switch.add_vlan(1000)
def test_transaction_context_already_connected_only_locks_connect_and_execute(self):
self.wrapped_switch.connected = True
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
self.wrapped_switch.should_receive("commit_transaction").once().ordered()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.switch.transaction():
self.switch.add_vlan(1000)
def test_transaction_context_already_in_transaction_only_executes(self):
self.wrapped_switch.connected = True
self.wrapped_switch.in_transaction = True
self.wrapped_switch.should_receive("add_vlan").once().ordered().with_args(1000)
with self.switch.transaction():
self.switch.add_vlan(1000)
def test_switch_contract_compliance_start_transaction(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_failing_to_start_rollback_dc_unlocks(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_failing_to_connect_unlocks(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered().and_raise(NetmanException)
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_failing_to_lock_gives_up(self):
self.lock.should_receive("acquire").once().ordered().and_raise(NetmanException)
with self.assertRaises(NetmanException):
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_already_connected_don_t_connect(self):
self.wrapped_switch.connected = True
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.switch.start_transaction()
def test_switch_contract_compliance_start_transaction_already_in_transaction_does_nothing(self):
self.wrapped_switch.connected = True
self.wrapped_switch.in_transaction = True
self.switch.start_transaction()
def test_switch_contract_compliance_end_transaction(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.end_transaction()
def test_switch_contract_compliance_end_transaction_dont_dc_for_a_legit_connection_after_a_managed_one(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.end_transaction()
self.wrapped_switch.connected = True
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.end_transaction()
def test_switch_contract_compliance_end_transaction_if_not_connected_by_start_dont_dc(self):
self.wrapped_switch.connected = True
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.lock.should_receive("release").once().ordered()
self.switch.end_transaction()
def test_switch_contract_compliance_end_transaction_failing_end_unlocks_and_dc(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered().and_raise(NetmanException)
self.wrapped_switch.should_receive("_disconnect").once().ordered()
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.end_transaction()
def test_switch_contract_compliance_end_transaction_failing_dc_unlocks(self):
self.lock.should_receive("acquire").once().ordered()
self.wrapped_switch.should_receive("_connect").once().ordered()
self.wrapped_switch.should_receive("_start_transaction").once().ordered()
self.switch.start_transaction()
self.wrapped_switch.should_receive("_end_transaction").once().ordered()
self.wrapped_switch.should_receive("_disconnect").once().ordered().and_raise(NetmanException)
self.lock.should_receive("release").once().ordered()
with self.assertRaises(NetmanException):
self.switch.end_transaction()
def test_switch_contract_compliance_connect(self):
self.wrapped_switch.should_receive("connect").once().ordered()
self.switch.connect()
def test_switch_contract_compliance_disconnect(self):
self.wrapped_switch.should_receive("disconnect").once().ordered()
self.switch.disconnect()
def test_switch_contract_compliance_commit_transaction(self):
self.wrapped_switch.should_receive("commit_transaction").once().ordered()
self.switch.commit_transaction()
def test_switch_contract_compliance_rollback_transaction(self):
self.wrapped_switch.should_receive("rollback_transaction").once().ordered()
self.switch.rollback_transaction()
def test_switch_contract_compliance_switch_descriptor(self):
assert_that(self.switch.switch_descriptor, is_(self.wrapped_switch.switch_descriptor))
