def test_list_with_not_found(self): class NotFound(Exception): status_code = 404 neut = resources.NeutronSecurityGroup() neut.user = mock.MagicMock() neut._resource = "security_group" neut.tenant_uuid = "user_tenant" neut.user.neutron().list_security_groups.side_effect = NotFound() expected_result = [] self.assertEqual(expected_result, list(neut.list())) neut.user.neutron().list_security_groups.assert_called_once_with( tenant_id=neut.tenant_uuid)
def test_list(self, admin, admin_required): sg_list = [{"tenant_id": "user_tenant", "name": "default"}, {"tenant_id": "user_tenant", "name": "foo_sg"}] neut = resources.NeutronSecurityGroup() neut.user = mock.MagicMock() neut._resource = "security_group" neut.tenant_uuid = "user_tenant" neut.user.neutron().list_security_groups.return_value = { "security_groups": sg_list } expected_result = [sg_list[1]] self.assertEqual(expected_result, list(neut.list())) neut.user.neutron().list_security_groups.assert_called_once_with( tenant_id=neut.tenant_uuid)