def setUp(self): self.security_group = Mock(spec=SecurityGroup) self.security_group.name = "secGroup" self.openstack_group_service = Mock(OpenstackGroupService) self.openstack_group_service.get_group.return_value = self.security_group self.openstack_rule_transformer = Mock(OpenstackRuleTransformer) self.openstack_rule_service = OpenstackRuleService( self.openstack_group_service, self.openstack_rule_transformer)
class TestOpenstackRuleService(unittest.TestCase): def setUp(self): self.security_group = Mock(spec=SecurityGroup) self.security_group.name = "secGroup" self.openstack_group_service = Mock(OpenstackGroupService) self.openstack_group_service.get_group.return_value = self.security_group self.openstack_rule_transformer = Mock(OpenstackRuleTransformer) self.openstack_rule_service = OpenstackRuleService( self.openstack_group_service, self.openstack_rule_transformer) def test_should_get_security_group_from_group_service(self): self.security_group.rules = [] self.openstack_rule_service.get_rules_for_group( self.security_group.name) self.openstack_group_service.get_group.assert_called_once_with( self.security_group.name) def test_should_transform_rules_from_security_group(self): first_rule = Mock() second_rule = Mock() self.security_group.rules = [first_rule, second_rule] self.openstack_rule_service.get_rules_for_group( self.security_group.name) self.openstack_rule_transformer.to_rule.assert_has_calls( [call(first_rule), call(second_rule)]) def test_should_return_transformed_security_group_rules(self): first_rule = Mock() second_rule = Mock() self.security_group.rules = [first_rule, second_rule] first_transformed_rule = Mock() second_transformed_rule = Mock() self.openstack_rule_transformer.to_rule.side_effect = [ first_transformed_rule, second_transformed_rule ] actual_rules = self.openstack_rule_service.get_rules_for_group( self.security_group.name) self.assertEqual( actual_rules, set([first_transformed_rule, second_transformed_rule]))
class TestOpenstackRuleService(unittest.TestCase): def setUp(self): self.security_group = Mock(spec=SecurityGroup) self.security_group.name = "secGroup" self.openstack_group_service = Mock(OpenstackGroupService) self.openstack_group_service.get_group.return_value = self.security_group self.openstack_rule_transformer = Mock(OpenstackRuleTransformer) self.openstack_rule_service = OpenstackRuleService( self.openstack_group_service, self.openstack_rule_transformer ) def test_should_get_security_group_from_group_service(self): self.security_group.rules = [] self.openstack_rule_service.get_rules_for_group(self.security_group.name) self.openstack_group_service.get_group.assert_called_once_with(self.security_group.name) def test_should_transform_rules_from_security_group(self): first_rule = Mock() second_rule = Mock() self.security_group.rules = [first_rule, second_rule] self.openstack_rule_service.get_rules_for_group(self.security_group.name) self.openstack_rule_transformer.to_rule.assert_has_calls([call(first_rule), call(second_rule)]) def test_should_return_transformed_security_group_rules(self): first_rule = Mock() second_rule = Mock() self.security_group.rules = [first_rule, second_rule] first_transformed_rule = Mock() second_transformed_rule = Mock() self.openstack_rule_transformer.to_rule.side_effect = [first_transformed_rule, second_transformed_rule] actual_rules = self.openstack_rule_service.get_rules_for_group(self.security_group.name) self.assertEqual(actual_rules, set([first_transformed_rule, second_transformed_rule]))
def setUp(self): self.security_group = Mock(spec=SecurityGroup) self.security_group.name = "secGroup" self.openstack_group_service = Mock(OpenstackGroupService) self.openstack_group_service.get_group.return_value = self.security_group self.openstack_rule_transformer = Mock(OpenstackRuleTransformer) self.openstack_rule_service = OpenstackRuleService( self.openstack_group_service, self.openstack_rule_transformer )