Byon dsl config - fix tests failures

Author: ivanayov

camp/camp-brooklyn/src/test/java/org/apache/brooklyn/camp/brooklyn/WinRmMachineLocationExternalConfigYamlTest.java

@@ -120,7 +120,7 @@ public void testWindowsMachinesNoExternalIPProvider() throws Exception {
             waitForApplicationTasks(app);
             Asserts.shouldHaveFailedPreviously();
         } catch (Exception e) {
-            Asserts.expectedFailureOfType(e, NullPointerException.class);
+            Asserts.expectedFailureContains(e, "Must specify exactly one of 'ssh' or 'winrm' for machine");
         }
     }
 

core/src/main/java/org/apache/brooklyn/location/byon/FixedListMachineProvisioningLocation.java

@@ -29,6 +29,7 @@
 import java.util.Map;
 import java.util.Set;
 
+import com.google.common.base.Preconditions;
 import com.google.common.net.HostAndPort;
 import org.apache.brooklyn.api.location.Location;
 import org.apache.brooklyn.api.location.LocationSpec;
@@ -384,7 +385,10 @@ protected void updateMachineConfig(T machine, Map<?, ?> flags) {
             // For backwards compatibility, where peristed state did not have this.
             origConfigs = Maps.newLinkedHashMap();
         }
-        parseMachineConfig((AbstractLocation) machine);
+        if (((AbstractLocation) machine).config().getBag().getAllConfig().get("provider") != null &&
+                ((AbstractLocation) machine).config().getBag().getAllConfig().get("provider").equals("byon")) {
+            parseMachineConfig((AbstractLocation) machine);
+        }
         Map<String, Object> strFlags = ConfigBag.newInstance(flags).getAllConfig();
         Map<String, Object> origConfig = ((ConfigurationSupportInternal)machine.config()).getLocalBag().getAllConfig();
         origConfigs.put(machine, origConfig);
@@ -399,12 +403,14 @@ private void parseMachineConfig(AbstractLocation machine) {
         String winrm =  machine.config().get(ConfigKeys.newStringConfigKey("winrm"));
         machine.config().removeKey(ConfigKeys.newStringConfigKey("winrm"));
 
-        Map<Integer, String> tcpPortMappings = (Map<Integer, String>) machine.config().get(ConfigKeys.newConfigKey(Map.class, "tcpPortMappings"));
-
-        if (ssh == null && winrm == null) {
-            throw new IllegalArgumentException("Must specify exactly one of 'ssh' or 'winrm' for machine: "+machine);
+        if (ssh ==null && winrm == null && machine.config().get(ConfigKeys.newStringConfigKey("address")) != null) {
+            return;
         }
 
+        Preconditions.checkArgument(ssh != null || winrm != null, "Must specify exactly one of 'ssh' or 'winrm' for machine: " + machine);
+
+        Map<Integer, String> tcpPortMappings = (Map<Integer, String>) machine.config().get(ConfigKeys.newConfigKey(Map.class, "tcpPortMappings"));
+
         UserAndHostAndPort userAndHostAndPort;
         String host;
         int port;