97 lines
2.2 KiB
Go
97 lines
2.2 KiB
Go
package oci
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/oracle/oci-go-sdk/core"
|
|
)
|
|
|
|
// driverMock implements the Driver interface and communicates with Oracle
|
|
// OCI.
|
|
type driverMock struct {
|
|
CreateInstanceID string
|
|
CreateInstanceErr error
|
|
|
|
CreateImageID string
|
|
CreateImageErr error
|
|
|
|
DeleteImageID string
|
|
DeleteImageErr error
|
|
|
|
GetInstanceIPErr error
|
|
|
|
TerminateInstanceID string
|
|
TerminateInstanceErr error
|
|
|
|
WaitForImageCreationErr error
|
|
|
|
WaitForInstanceStateErr error
|
|
|
|
cfg *Config
|
|
}
|
|
|
|
// CreateInstance creates a new compute instance.
|
|
func (d *driverMock) CreateInstance(ctx context.Context, publicKey string) (string, error) {
|
|
if d.CreateInstanceErr != nil {
|
|
return "", d.CreateInstanceErr
|
|
}
|
|
|
|
d.CreateInstanceID = "ocid1..."
|
|
|
|
return d.CreateInstanceID, nil
|
|
}
|
|
|
|
// CreateImage creates a new custom image.
|
|
func (d *driverMock) CreateImage(ctx context.Context, id string) (core.Image, error) {
|
|
if d.CreateImageErr != nil {
|
|
return core.Image{}, d.CreateImageErr
|
|
}
|
|
d.CreateImageID = id
|
|
return core.Image{Id: &id}, nil
|
|
}
|
|
|
|
// DeleteImage mocks deleting a custom image.
|
|
func (d *driverMock) DeleteImage(ctx context.Context, id string) error {
|
|
if d.DeleteImageErr != nil {
|
|
return d.DeleteImageErr
|
|
}
|
|
|
|
d.DeleteImageID = id
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetInstanceIP returns the public or private IP corresponding to the given instance id.
|
|
func (d *driverMock) GetInstanceIP(ctx context.Context, id string) (string, error) {
|
|
if d.GetInstanceIPErr != nil {
|
|
return "", d.GetInstanceIPErr
|
|
}
|
|
if d.cfg.UsePrivateIP {
|
|
return "private_ip", nil
|
|
}
|
|
return "ip", nil
|
|
}
|
|
|
|
// TerminateInstance terminates a compute instance.
|
|
func (d *driverMock) TerminateInstance(ctx context.Context, id string) error {
|
|
if d.TerminateInstanceErr != nil {
|
|
return d.TerminateInstanceErr
|
|
}
|
|
|
|
d.TerminateInstanceID = id
|
|
|
|
return nil
|
|
}
|
|
|
|
// WaitForImageCreation waits for a provisioning custom image to reach the
|
|
// "AVAILABLE" state.
|
|
func (d *driverMock) WaitForImageCreation(ctx context.Context, id string) error {
|
|
return d.WaitForImageCreationErr
|
|
}
|
|
|
|
// WaitForInstanceState waits for an instance to reach the a given terminal
|
|
// state.
|
|
func (d *driverMock) WaitForInstanceState(ctx context.Context, id string, waitStates []string, terminalState string) error {
|
|
return d.WaitForInstanceStateErr
|
|
}
|