1 /*
   2  * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 
  25 /*
  26  * @test
  27  * @summary Test JVM's memory resource awareness when running inside docker container
  28  * @requires docker.support
  29  * @library /test/lib
  30  * @modules java.base/jdk.internal.misc
  31  *          java.management
  32  *          jdk.jartool/sun.tools.jar
  33  * @build Common AttemptOOM sun.hotspot.WhiteBox PrintContainerInfo
  34  * @run driver ClassFileInstaller -jar whitebox.jar sun.hotspot.WhiteBox sun.hotspot.WhiteBox$WhiteBoxPermission
  35  * @run driver TestMemoryAwareness
  36  */
  37 import jdk.test.lib.containers.docker.DockerRunOptions;
  38 import jdk.test.lib.containers.docker.DockerTestUtils;
  39 
  40 
  41 public class TestMemoryAwareness {
  42     private static final String imageName = Common.imageName("memory");
  43 
  44     public static void main(String[] args) throws Exception {
  45         if (!DockerTestUtils.canTestDocker()) {
  46             return;
  47         }
  48 
  49         Common.prepareWhiteBox();
  50         DockerTestUtils.buildJdkDockerImage(imageName, "Dockerfile-BasicTest", "jdk-docker");
  51 
  52         try {
  53             testMemoryLimit("100m", "104857600");
  54             testMemoryLimit("500m", "524288000");
  55             testMemoryLimit("1g", "1073741824");
  56             testMemoryLimit("4g", "4294967296");
  57 
  58             testMemorySoftLimit("500m", "524288000");
  59             testMemorySoftLimit("1g", "1073741824");
  60 
  61             // Add extra 10 Mb to allocator limit, to be sure to cause OOM
  62             testOOM("256m", 256 + 10);
  63 
  64         } finally {
  65             DockerTestUtils.removeDockerImage(imageName);
  66         }
  67     }
  68 
  69 
  70     private static void testMemoryLimit(String valueToSet, String expectedTraceValue)
  71             throws Exception {
  72 
  73         Common.logNewTestCase("memory limit: " + valueToSet);
  74 
  75         DockerRunOptions opts = Common.newOpts(imageName)
  76             .addDockerOpts("--memory", valueToSet);
  77 
  78         Common.run(opts)
  79             .shouldMatch("Memory Limit is:.*" + expectedTraceValue);
  80     }
  81 
  82 
  83     private static void testMemorySoftLimit(String valueToSet, String expectedTraceValue)
  84             throws Exception {
  85         Common.logNewTestCase("memory soft limit: " + valueToSet);
  86 
  87         DockerRunOptions opts = Common.newOpts(imageName, "PrintContainerInfo");
  88         Common.addWhiteBoxOpts(opts);
  89         opts.addDockerOpts("--memory-reservation=" + valueToSet);
  90 
  91         Common.run(opts)
  92             .shouldMatch("Memory Soft Limit.*" + expectedTraceValue);
  93     }
  94 
  95 
  96     // provoke OOM inside the container, see how VM reacts
  97     private static void testOOM(String dockerMemLimit, int sizeToAllocInMb) throws Exception {
  98         Common.logNewTestCase("OOM");
  99 
 100         DockerRunOptions opts = Common.newOpts(imageName, "AttemptOOM")
 101             .addDockerOpts("--memory", dockerMemLimit, "--memory-swap", dockerMemLimit);
 102         opts.classParams.add("" + sizeToAllocInMb);
 103 
 104         DockerTestUtils.dockerRunJava(opts)
 105             .shouldHaveExitValue(1)
 106             .shouldContain("Entering AttemptOOM main")
 107             .shouldNotContain("AttemptOOM allocation successful")
 108             .shouldContain("java.lang.OutOfMemoryError");
 109     }
 110 
 111 }