func (s *AvailabilityZoneSuite) TestBestAvailabilityZoneAllocationsAllInstancesErrors(c *gc.C) {
resultErr := fmt.Errorf("oh noes")
s.PatchValue(&s.env.allInstances, func() ([]instance.Instance, error) {
return nil, resultErr
})
best, err := common.BestAvailabilityZoneAllocations(&s.env, nil)
c.Assert(err, gc.Equals, resultErr)
c.Assert(best, gc.HasLen, 0)
}
func (s *AvailabilityZoneSuite) TestBestAvailabilityZoneAllocationsPartialInstances(c *gc.C) {
var called int
s.PatchValue(&s.env.instanceAvailabilityZoneNames, func(ids []instance.Id) ([]string, error) {
c.Assert(ids, gc.DeepEquals, []instance.Id{"nichts", "inst1", "null", "inst2"})
called++
return []string{"", "az1", "", "az1"}, environs.ErrPartialInstances
})
best, err := common.BestAvailabilityZoneAllocations(&s.env, []instance.Id{"nichts", "inst1", "null", "inst2"})
c.Assert(called, gc.Equals, 1)
c.Assert(err, gc.IsNil)
// All known instances are in az1 and az0 is unavailable, so az2 is the best.
c.Assert(best, gc.DeepEquals, map[string][]instance.Id{"az2": nil})
}
func (s *AvailabilityZoneSuite) TestBestAvailabilityZoneAllocationsInstanceAvailabilityZonesErrors(c *gc.C) {
var returnErr error
var called int
s.PatchValue(&s.env.instanceAvailabilityZoneNames, func(ids []instance.Id) ([]string, error) {
called++
return nil, returnErr
})
errors := []error{environs.ErrNoInstances, fmt.Errorf("whatever")}
for i, err := range errors {
returnErr = err
best, err := common.BestAvailabilityZoneAllocations(&s.env, nil)
c.Assert(called, gc.Equals, i+1)
c.Assert(err, gc.Equals, returnErr)
c.Assert(best, gc.HasLen, 0)
}
}
func (s *AvailabilityZoneSuite) TestBestAvailabilityZoneAllocationsAllInstances(c *gc.C) {
var called int
s.PatchValue(&s.env.instanceAvailabilityZoneNames, func(ids []instance.Id) ([]string, error) {
c.Assert(ids, gc.DeepEquals, []instance.Id{"inst0", "inst1", "inst2"})
called++
return []string{"az0", "az1", "az2"}, nil
})
best, err := common.BestAvailabilityZoneAllocations(&s.env, nil)
c.Assert(called, gc.Equals, 1)
c.Assert(err, gc.IsNil)
// az0 is unavailable, so az1 and az2 come out as equal best.
c.Assert(best, gc.DeepEquals, map[string][]instance.Id{
"az1": []instance.Id{"inst1"},
"az2": []instance.Id{"inst2"},
})
}
func (s *AvailabilityZoneSuite) TestBestAvailabilityZoneAllocationsNoZones(c *gc.C) {
var calls []string
s.PatchValue(&s.env.instanceAvailabilityZoneNames, func(ids []instance.Id) ([]string, error) {
c.Assert(ids, gc.DeepEquals, []instance.Id{"inst0", "inst1", "inst2"})
calls = append(calls, "InstanceAvailabilityZoneNames")
return []string{"", "", ""}, nil
})
s.PatchValue(&s.env.availabilityZones, func() ([]common.AvailabilityZone, error) {
calls = append(calls, "AvailabilityZones")
return []common.AvailabilityZone{}, nil
})
best, err := common.BestAvailabilityZoneAllocations(&s.env, nil)
c.Assert(calls, gc.DeepEquals, []string{"InstanceAvailabilityZoneNames", "AvailabilityZones"})
c.Assert(err, gc.IsNil)
c.Assert(best, gc.HasLen, 0)
}