/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.brooklyn.location.multi;
import static org.testng.Assert.assertTrue;
import org.apache.brooklyn.test.Asserts;
import org.apache.brooklyn.util.collections.MutableSet;
import org.apache.brooklyn.util.net.Networking;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.testng.Assert;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import org.apache.brooklyn.api.location.LocationSpec;
import org.apache.brooklyn.api.location.NoMachinesAvailableException;
import org.apache.brooklyn.core.entity.Entities;
import org.apache.brooklyn.core.location.cloud.AvailabilityZoneExtension;
import org.apache.brooklyn.core.mgmt.internal.LocalManagementContext;
import org.apache.brooklyn.core.test.entity.LocalManagementContextForTests;
import org.apache.brooklyn.location.byon.FixedListMachineProvisioningLocation;
import org.apache.brooklyn.location.multi.MultiLocation;
import org.apache.brooklyn.location.ssh.SshMachineLocation;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
public class MultiLocationTest {
private static final Logger log = LoggerFactory.getLogger(MultiLocationTest.class);
private LocalManagementContext managementContext;
private SshMachineLocation mac1a;
private SshMachineLocation mac1b;
private SshMachineLocation mac2a;
private SshMachineLocation mac2b;
private FixedListMachineProvisioningLocation<SshMachineLocation> loc1;
private FixedListMachineProvisioningLocation<SshMachineLocation> loc2;
private MultiLocation<SshMachineLocation> multiLoc;
@SuppressWarnings("unchecked")
@BeforeMethod(alwaysRun=true)
public void setUp() throws Exception {
managementContext = LocalManagementContextForTests.newInstance();
mac1a = managementContext.getLocationManager().createLocation(LocationSpec.create(SshMachineLocation.class)
.displayName("mac1a")
.configure("address", Networking.getInetAddressWithFixedName("1.1.1.1")));
mac1b = managementContext.getLocationManager().createLocation(LocationSpec.create(SshMachineLocation.class)
.displayName("mac1b")
.configure("address", Networking.getInetAddressWithFixedName("1.1.1.2")));
mac2a = managementContext.getLocationManager().createLocation(LocationSpec.create(SshMachineLocation.class)
.displayName("mac2a")
.configure("address", Networking.getInetAddressWithFixedName("1.1.1.3")));
mac2b = managementContext.getLocationManager().createLocation(LocationSpec.create(SshMachineLocation.class)
.displayName("mac2b")
.configure("address", Networking.getInetAddressWithFixedName("1.1.1.4")));
loc1 = managementContext.getLocationManager().createLocation(LocationSpec.create(FixedListMachineProvisioningLocation.class)
.displayName("loc1")
.configure("machines", MutableSet.of(mac1a, mac1b)));
loc2 = managementContext.getLocationManager().createLocation(LocationSpec.create(FixedListMachineProvisioningLocation.class)
.displayName("loc2")
.configure("machines", MutableSet.of(mac2a, mac2b)));
multiLoc = managementContext.getLocationManager().createLocation(LocationSpec.create(MultiLocation.class)
.displayName("multiLoc")
.configure("subLocations", ImmutableList.of(loc1, loc2)));
}
@AfterMethod(alwaysRun=true)
public void tearDown() throws Exception {
if (managementContext != null) Entities.destroyAll(managementContext);
}
@Test
public void testHasAvailabilityZonesAsSubLocations() throws Exception {
multiLoc.hasExtension(AvailabilityZoneExtension.class);
AvailabilityZoneExtension extension = multiLoc.getExtension(AvailabilityZoneExtension.class);
Asserts.assertEqualsIgnoringOrder(extension.getAllSubLocations(), ImmutableList.of(loc1, loc2));
Asserts.assertEqualsIgnoringOrder(extension.getSubLocations(2), ImmutableList.of(loc1, loc2));
assertTrue(ImmutableList.of(loc1, loc2).containsAll(extension.getSubLocations(1)));
}
@Test
public void testObtainAndReleaseDelegateToSubLocation() throws Exception {
SshMachineLocation obtained = multiLoc.obtain(ImmutableMap.of());
assertTrue(ImmutableList.of(mac1a, mac1b, mac2a, mac2b).contains(obtained));
multiLoc.release(obtained);
}
@Test
public void testObtainsMovesThroughSubLocations() throws Exception {
Assert.assertEquals(multiLoc.obtain().getAddress().getHostAddress(), "1.1.1.1");
Assert.assertEquals(multiLoc.obtain().getAddress().getHostAddress(), "1.1.1.2");
Assert.assertEquals(multiLoc.obtain().getAddress().getHostAddress(), "1.1.1.3");
Assert.assertEquals(multiLoc.obtain().getAddress().getHostAddress(), "1.1.1.4");
try {
multiLoc.obtain();
Assert.fail();
} catch (NoMachinesAvailableException e) {
log.info("Error when no machines available across locations is: "+e);
Assert.assertTrue(e.toString().contains("loc1"), "Message should have referred to sub-location message: "+e);
}
}
}