// +build acceptance

package openstack

import (
	blockstorage "github.com/rackspace/gophercloud/openstack/blockstorage/v1"
	"github.com/rackspace/gophercloud/openstack/blockstorage/v1/volumes"
	"github.com/rackspace/gophercloud/openstack/identity"
	"github.com/rackspace/gophercloud/openstack/utils"
	"os"
	"strconv"
	"testing"
	"time"
)

var numVols = 2

func getClient() (*blockstorage.Client, error) {
	ao, err := utils.AuthOptions()
	if err != nil {
		return nil, err
	}

	r, err := identity.Authenticate(ao)
	if err != nil {
		return nil, err
	}

	sc, err := identity.GetServiceCatalog(r)
	if err != nil {
		return nil, err
	}

	ces, err := sc.CatalogEntries()
	if err != nil {
		return nil, err
	}

	var eps []identity.Endpoint
	for _, ce := range ces {
		if ce.Type == "volume" {
			eps = ce.Endpoints
		}
	}

	region := os.Getenv("OS_REGION_NAME")
	rep := ""
	for _, ep := range eps {
		if ep.Region == region {
			rep = ep.PublicURL
		}
	}

	client := blockstorage.NewClient(rep, r, ao)

	return client, nil

}

func TestVolumes(t *testing.T) {
	client, err := getClient()
	if err != nil {
		t.Error(err)
		return
	}

	var cv volumes.Volume
	for i := 0; i < numVols; i++ {
		cv, err := volumes.Create(client, volumes.CreateOpts{
			"size":         1,
			"display_name": "test-volume" + strconv.Itoa(i),
		})
		if err != nil {
			t.Error(err)
			return
		}
		defer func() {
			time.Sleep(10000 * time.Millisecond)
			err = volumes.Delete(client, volumes.DeleteOpts{
				"id": cv.Id,
			})
			if err != nil {
				t.Error(err)
				return
			}
		}()
	}

	vols, err := volumes.List(client, volumes.ListOpts{
		"full": true,
	})
	if err != nil {
		t.Error(err)
		return
	}
	if len(vols) != numVols {
		t.Errorf("Expected %d volumes, got %d", numVols, len(vols))
		return
	}

	vols, err = volumes.List(client, volumes.ListOpts{
		"full": false,
	})
	if err != nil {
		t.Error(err)
		return
	}
	if len(vols) != numVols {
		t.Errorf("Expected %d volumes, got %d", numVols, len(vols))
		return
	}

	_, err = volumes.Get(client, volumes.GetOpts{
		"id": cv.Id,
	})
	if err != nil {
		t.Error(err)
		return
	}

}