def test__prepare_open_secgroup_rules(self, mock_clients): fake_neutron = mock_clients.return_value.neutron.return_value fake_neutron.list_security_groups.return_value = { "security_groups": [{ "id": "id", "name": "foo", "security_group_rules": [] }] } allow_ssh._prepare_open_secgroup("credential", self.secgroup_name) allow_ssh._prepare_open_secgroup("credential", "foo")
def test__prepare_open_secgroup(self, mock_clients): fake_nova = fakes.FakeNovaClient() self.assertEqual(len(fake_nova.security_groups.list()), 1) mock_cl = mock.MagicMock() mock_cl.nova.return_value = fake_nova mock_clients.return_value = mock_cl ret = allow_ssh._prepare_open_secgroup("endpoint", self.secgroup_name) self.assertEqual(self.secgroup_name, ret["name"]) self.assertEqual(2, len(fake_nova.security_groups.list())) self.assertIn(self.secgroup_name, [sg.name for sg in fake_nova.security_groups.list()]) # run prep again, check that another security group is not created allow_ssh._prepare_open_secgroup("endpoint", self.secgroup_name) self.assertEqual(2, len(fake_nova.security_groups.list()))
def test__prepare_open_secgroup(self, mock_clients): fake_nova = fakes.FakeNovaClient() self.assertEqual(len(fake_nova.security_groups.list()), 1) mock_cl = mock.MagicMock() mock_cl.nova.return_value = fake_nova mock_clients.return_value = mock_cl ret = allow_ssh._prepare_open_secgroup("endpoint", self.secgroup_name) self.assertEqual(self.secgroup_name, ret["name"]) self.assertEqual(2, len(fake_nova.security_groups.list())) self.assertIn( self.secgroup_name, [sg.name for sg in fake_nova.security_groups.list()]) # run prep again, check that another security group is not created allow_ssh._prepare_open_secgroup("endpoint", self.secgroup_name) self.assertEqual(2, len(fake_nova.security_groups.list()))
def test__prepare_open_secgroup_rules(self, mock_clients): fake_nova = fakes.FakeNovaClient() # NOTE(hughsaunders) Default security group is precreated self.assertEqual(1, len(fake_nova.security_groups.list())) mock_cl = mock.MagicMock() mock_cl.nova.return_value = fake_nova mock_clients.return_value = mock_cl allow_ssh._prepare_open_secgroup("endpoint", self.secgroup_name) self.assertEqual(2, len(fake_nova.security_groups.list())) rally_open = fake_nova.security_groups.find(self.secgroup_name) self.assertEqual(3, len(rally_open.rules)) # run prep again, check that extra rules are not created allow_ssh._prepare_open_secgroup("endpoint", self.secgroup_name) rally_open = fake_nova.security_groups.find(self.secgroup_name) self.assertEqual(3, len(rally_open.rules))
def test__prepare_open_secgroup_rules(self, mock_clients): fake_nova = fakes.FakeNovaClient() # NOTE(hughsaunders) Default security group is precreated self.assertEqual(1, len(fake_nova.security_groups.list())) mock_cl = mock.MagicMock() mock_cl.nova.return_value = fake_nova mock_clients.return_value = mock_cl allow_ssh._prepare_open_secgroup("credential", self.secgroup_name) self.assertEqual(2, len(fake_nova.security_groups.list())) rally_open = fake_nova.security_groups.find(self.secgroup_name) self.assertEqual(3, len(rally_open.rules)) # run prep again, check that extra rules are not created allow_ssh._prepare_open_secgroup("credential", self.secgroup_name) rally_open = fake_nova.security_groups.find(self.secgroup_name) self.assertEqual(3, len(rally_open.rules))