upload android base code part6

android/test/vts-testcase/kernel/memory/MemorySystemStressTest.py

@@ -0,0 +1,65 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2017 The Android Open Source Project
+#
+# Licensed 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.
+#
+
+import logging
+
+from vts.runners.host import asserts
+from vts.runners.host import base_test
+from vts.runners.host import const
+from vts.runners.host import test_runner
+from vts.utils.python.controllers import android_device
+import re
+
+
+class MemorySystemStressTest(base_test.BaseTestClass):
+    """Userspace Memory Stress Test."""
+    _mem = 64
+    _DURATION_SEC = 300
+    _STRESSAPPTEST = '/data/local/tmp/32/stressapptest'
+
+    def setUpClass(self):
+        self.dut = self.registerController(android_device)[0]
+
+        # Set executable bit on stressapptest binary
+        self.dut.shell.InvokeTerminal("sat_setup")
+        self.shell = getattr(self.dut.shell, "sat_setup")
+        cmd = ['chmod +x', str(self._STRESSAPPTEST)]
+        self.shell.Execute(' '.join(cmd))
+
+        # Read /proc/meminfo to read MemFree; use MemFree as total memory under
+        # test
+        meminfo_results = self.shell.Execute("cat /proc/meminfo")
+        matched = re.search(r'MemFree:\s+(\d+)', str(meminfo_results[const.STDOUT]))
+        if matched:
+            self._mem = int(matched.groups()[0]) / 1024
+            self._mem = max(self._mem, 64)   # Minimum 64M memory under test
+            self._mem = min(self._mem, 1280) # Maximum 1280M memory under test
+
+    def testMemory32bit(self):
+        """Memory Test using stressapptest."""
+
+        # Compose stressapptest command to execute for memory test
+        cmd = [str(self._STRESSAPPTEST), '-W ' '-M', str(self._mem), '-s', str(self._DURATION_SEC)]
+        logging.info(cmd)
+        results = self.shell.Execute(' '.join(cmd))
+        logging.info(str(results[const.STDOUT]))
+
+        # Check for 'PASS' and exit code
+        asserts.assertTrue('Status: PASS' in results[const.STDOUT][0].strip(), "Memory test failed.")
+
+if __name__ == "__main__":
+    test_runner.main()