5991b29f76
* Add UserAgent * Change to join the user agent with a whitespace and set in the header * Remove empty entry from user agent
77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package network
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/pkg/errors"
|
|
azure "github.com/virtual-kubelet/virtual-kubelet/providers/azure/client"
|
|
"github.com/virtual-kubelet/virtual-kubelet/providers/azure/client/resourcegroups"
|
|
)
|
|
|
|
var (
|
|
location = "westus"
|
|
resourceGroup = "virtual-kubelet-tests"
|
|
testAuth *azure.Authentication
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
uid := uuid.New()
|
|
resourceGroup += "-" + uid.String()[0:6]
|
|
|
|
if err := setupAuth(); err != nil {
|
|
fmt.Fprintln(os.Stderr, "Error setting up auth:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
c, err := resourcegroups.NewClient(testAuth, "unit-test")
|
|
if err != nil {
|
|
os.Exit(1)
|
|
}
|
|
_, err = c.CreateResourceGroup(resourceGroup, resourcegroups.Group{
|
|
Name: resourceGroup,
|
|
Location: location,
|
|
})
|
|
if err != nil {
|
|
os.Exit(1)
|
|
}
|
|
|
|
code := m.Run()
|
|
|
|
if err := c.DeleteResourceGroup(resourceGroup); err != nil {
|
|
fmt.Fprintln(os.Stderr, "error removing resource group:", err)
|
|
}
|
|
|
|
os.Exit(code)
|
|
}
|
|
|
|
var authOnce sync.Once
|
|
|
|
func setupAuth() error {
|
|
var err error
|
|
authOnce.Do(func() {
|
|
testAuth, err = azure.NewAuthenticationFromFile(os.Getenv("AZURE_AUTH_LOCATION"))
|
|
if err != nil {
|
|
testAuth, err = azure.NewAuthenticationFromFile("../../../../credentials.json")
|
|
}
|
|
if err != nil {
|
|
err = errors.Wrap(err, "failed to load Azure authentication file")
|
|
}
|
|
})
|
|
return err
|
|
}
|
|
|
|
func newTestClient(t *testing.T) *Client {
|
|
if err := setupAuth(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
c, err := NewClient(testAuth, "unit-test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return c
|
|
}
|