adjusted tests

This commit is contained in:
Chris Lu 2012-09-01 13:17:30 -07:00
parent c51884ce23
commit a467d5081c
2 changed files with 3 additions and 6 deletions

View file

@ -161,11 +161,9 @@ func TestRemoveDataCenter(t *testing.T) {
func TestReserveOneVolume(t *testing.T) {
topo := setup()
rand.Seed(time.Now().UnixNano())
ret, vid := topo.RandomlyReserveOneVolume()
ret, node, vid := topo.RandomlyReserveOneVolume()
fmt.Println("topology:", topo.Node)
fmt.Println("assigned :", ret)
fmt.Println("assigned node :", node)
fmt.Println("assigned volume id:", vid)
if topo.reservedVolumeCount != 1 {
t.Fail()
}
}

View file

@ -3,7 +3,7 @@ package topology
import (
"math/rand"
"pkg/storage"
"fmt"
_ "fmt"
)
type Topology struct {
@ -19,7 +19,6 @@ func NewTopology(id NodeId) *Topology{
func (t *Topology) RandomlyReserveOneVolume() (bool, *Node, storage.VolumeId) {
slots := t.Node.maxVolumeCount-t.Node.activeVolumeCount
r := rand.Intn(slots)
fmt.Println("slots:", slots, "random :", r)
vid := t.nextVolumeId()
ret, node := t.Node.ReserveOneVolume(r,vid)
return ret, node, vid