aries-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From dav...@apache.org
Subject aries-containers git commit: Unit tests for Marathon binding
Date Thu, 15 Jun 2017 09:49:26 GMT
Repository: aries-containers
Updated Branches:
  refs/heads/master e33916185 -> 15d9b7c21


Unit tests for Marathon binding


Project: http://git-wip-us.apache.org/repos/asf/aries-containers/repo
Commit: http://git-wip-us.apache.org/repos/asf/aries-containers/commit/15d9b7c2
Tree: http://git-wip-us.apache.org/repos/asf/aries-containers/tree/15d9b7c2
Diff: http://git-wip-us.apache.org/repos/asf/aries-containers/diff/15d9b7c2

Branch: refs/heads/master
Commit: 15d9b7c21ec5ee01c7fd6d5eb44d753f27cad469
Parents: e339161
Author: David Bosschaert <davidb@apache.org>
Authored: Thu Jun 15 10:49:18 2017 +0100
Committer: David Bosschaert <davidb@apache.org>
Committed: Thu Jun 15 10:49:18 2017 +0100

----------------------------------------------------------------------
 .../impl/MarathonServiceManagerTest.java        | 123 +++++++++++++++++++
 1 file changed, 123 insertions(+)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/aries-containers/blob/15d9b7c2/containers-marathon/src/test/java/org/apache/aries/containers/marathon/impl/MarathonServiceManagerTest.java
----------------------------------------------------------------------
diff --git a/containers-marathon/src/test/java/org/apache/aries/containers/marathon/impl/MarathonServiceManagerTest.java
b/containers-marathon/src/test/java/org/apache/aries/containers/marathon/impl/MarathonServiceManagerTest.java
new file mode 100644
index 0000000..de75ce9
--- /dev/null
+++ b/containers-marathon/src/test/java/org/apache/aries/containers/marathon/impl/MarathonServiceManagerTest.java
@@ -0,0 +1,123 @@
+/*
+ * 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.aries.containers.marathon.impl;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import org.apache.aries.containers.Service;
+import org.apache.aries.containers.ServiceConfig;
+import org.junit.Test;
+import org.mockito.Mockito;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertSame;
+
+import mesosphere.marathon.client.Marathon;
+import mesosphere.marathon.client.model.v2.App;
+import mesosphere.marathon.client.model.v2.Container;
+import mesosphere.marathon.client.model.v2.Docker;
+import mesosphere.marathon.client.model.v2.GetAppsResponse;
+import mesosphere.marathon.client.model.v2.Port;
+
+public class MarathonServiceManagerTest {
+    @Test
+    public void testGetService() throws Exception {
+        GetAppsResponse nar = Mockito.mock(GetAppsResponse.class);
+
+        List<App> appsCreated = new ArrayList<>();
+        Marathon mc = Mockito.mock(Marathon.class);
+        Mockito.when(mc.getApps(Collections.singletonMap(
+                "label", MarathonServiceManager.SERVICE_NAME + "==myservice"))).thenReturn(nar);
+        Mockito.when(mc.createApp(Mockito.isA(App.class))).then(new Answer<App>() {
+            @Override
+            public App answer(InvocationOnMock invocation) throws Throwable {
+                App a = (App) invocation.getArguments()[0];
+                appsCreated.add(a);
+                return a;
+            }
+        });
+
+        MarathonServiceManager msm = new MarathonServiceManager(mc);
+
+        ServiceConfig cfg = ServiceConfig.builder("myservice", "animage").
+                cpu(1.72).
+                env("foo", "bar").
+                memory(80.5).
+                port(8080).
+                port(8181).
+                build();
+
+        assertEquals("Precondition", 0, appsCreated.size());
+        Service svc = msm.getService(cfg);
+        assertEquals(1, appsCreated.size());
+
+        App app = appsCreated.iterator().next();
+        assertEquals("myservice", app.getId());
+        assertEquals(1.72, app.getCpus(), 0.0001);
+        assertEquals(80.5, app.getMem(), 0.0001);
+        assertEquals(1, (int) app.getInstances());
+        assertEquals(Collections.singletonMap("foo", "bar"), app.getEnv());
+        assertNull(app.getCmd());
+
+        Container ctr = app.getContainer();
+        assertEquals("DOCKER", ctr.getType());
+
+        Docker dkr = ctr.getDocker();
+        assertEquals("animage", dkr.getImage());
+        assertEquals("BRIDGE", dkr.getNetwork());
+
+        Collection<Port> ports = dkr.getPortMappings();
+        assertEquals(2, ports.size());
+        Set<Integer> foundPorts = new HashSet<>();
+        for (Port p : ports) {
+            foundPorts.add(p.getContainerPort());
+        }
+        assertEquals(new HashSet<>(Arrays.asList(8080, 8181)), foundPorts);
+
+        assertSame(cfg, svc.getConfiguration());
+    }
+
+    @Test
+    public void testGetExistingService() throws Exception {
+        App app = new App();
+        app.setLabels(Collections.singletonMap(MarathonServiceManager.SERVICE_NAME, "asvc"));
+
+        GetAppsResponse ear = Mockito.mock(GetAppsResponse.class);
+        Mockito.when(ear.getApps()).thenReturn(Collections.singletonList(app));
+
+        Marathon mc = Mockito.mock(Marathon.class);
+        Mockito.when(mc.getApps(Collections.singletonMap(
+                "label", MarathonServiceManager.SERVICE_NAME + "==asvc"))).thenReturn(ear);
+
+        MarathonServiceManager msm = new MarathonServiceManager(mc);
+
+        ServiceConfig cfg = ServiceConfig.builder("asvc", "img.1").build();
+        Service svc = msm.getService(cfg);
+        assertSame(cfg, svc.getConfiguration());
+    }
+}


Mime
View raw message