constellation/internal/cloud/openstack/api_test.go

/*
Copyright (c) Edgeless Systems GmbH

SPDX-License-Identifier: AGPL-3.0-only
*/

package openstack

import (
	"context"

	"github.com/edgelesssys/constellation/v2/internal/role"
	"github.com/gophercloud/gophercloud/openstack/compute/v2/servers"
	"github.com/gophercloud/gophercloud/openstack/networking/v2/networks"
	"github.com/gophercloud/gophercloud/openstack/networking/v2/subnets"
	"github.com/gophercloud/gophercloud/pagination"
)

type stubIMDSClient struct {
	providerIDResult           string
	providerIDErr              error
	nameResult                 string
	nameErr                    error
	projectIDResult            string
	projectIDErr               error
	uidResult                  string
	uidErr                     error
	initSecretHashResult       string
	initSecretHashErr          error
	roleResult                 role.Role
	roleErr                    error
	vpcIPResult                string
	vpcIPErr                   error
	loadBalancerEndpointResult string
	loadBalancerEndpointErr    error
}

func (c *stubIMDSClient) providerID(_ context.Context) (string, error) {
	return c.providerIDResult, c.providerIDErr
}

func (c *stubIMDSClient) name(_ context.Context) (string, error) {
	return c.nameResult, c.nameErr
}

func (c *stubIMDSClient) projectID(_ context.Context) (string, error) {
	return c.projectIDResult, c.projectIDErr
}

func (c *stubIMDSClient) uid(_ context.Context) (string, error) {
	return c.uidResult, c.uidErr
}

func (c *stubIMDSClient) initSecretHash(_ context.Context) (string, error) {
	return c.initSecretHashResult, c.initSecretHashErr
}

func (c *stubIMDSClient) role(_ context.Context) (role.Role, error) {
	return c.roleResult, c.roleErr
}

func (c *stubIMDSClient) vpcIP(_ context.Context) (string, error) {
	return c.vpcIPResult, c.vpcIPErr
}

func (c *stubIMDSClient) loadBalancerEndpoint(_ context.Context) (string, error) {
	return c.loadBalancerEndpointResult, c.loadBalancerEndpointErr
}

type stubServersClient struct {
	serversPager stubPager
	netsPager    stubPager
	subnetsPager stubPager
}

func (c *stubServersClient) ListServers(_ servers.ListOptsBuilder) pagerAPI {
	return &c.serversPager
}

func (c *stubServersClient) ListNetworks(_ networks.ListOptsBuilder) pagerAPI {
	return &c.netsPager
}

func (c *stubServersClient) ListSubnets(_ subnets.ListOpts) pagerAPI {
	return &c.subnetsPager
}

type stubPager struct {
	page        pagination.Page
	allPagesErr error
}

func (p *stubPager) AllPages() (pagination.Page, error) {
	return p.page, p.allPagesErr
}