1 #!/usr/bin/env python 2 # 3 # Copyright (C) 2017 The Android Open Source Project 4 # 5 # Licensed under the Apache License, Version 2.0 (the "License"); 6 # you may not use this file except in compliance with the License. 7 # You may obtain a copy of the License at 8 # 9 # http://www.apache.org/licenses/LICENSE-2.0 10 # 11 # Unless required by applicable law or agreed to in writing, software 12 # distributed under the License is distributed on an "AS IS" BASIS, 13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 # See the License for the specific language governing permissions and 15 # limitations under the License. 16 # 17 18 import logging 19 20 from vts.runners.host import asserts 21 from vts.runners.host import base_test 22 from vts.runners.host import const 23 from vts.runners.host import test_runner 24 from vts.utils.python.controllers import android_device 25 import re 26 27 28 class MemorySystemStressTest(base_test.BaseTestClass): 29 """Userspace Memory Stress Test.""" 30 _mem = 64 31 _DURATION_SEC = 300 32 _STRESSAPPTEST = '/data/local/tmp/32/stressapptest' 33 34 def setUpClass(self): 35 self.dut = self.registerController(android_device)[0] 36 37 # Set executable bit on stressapptest binary 38 self.dut.shell.InvokeTerminal("sat_setup") 39 self.shell = getattr(self.dut.shell, "sat_setup") 40 cmd = ['chmod +x', str(self._STRESSAPPTEST)] 41 self.shell.Execute(' '.join(cmd)) 42 43 # Read /proc/meminfo to read MemFree; use MemFree as total memory under 44 # test 45 meminfo_results = self.shell.Execute("cat /proc/meminfo") 46 matched = re.search(r'MemFree:\s+(\d+)', str(meminfo_results[const.STDOUT])) 47 if matched: 48 self._mem = int(matched.groups()[0]) / 1024 49 self._mem = max(self._mem, 64) # Minimum 64M memory under test 50 self._mem = min(self._mem, 1280) # Maximum 1280M memory under test 51 52 def testMemory32bit(self): 53 """Memory Test using stressapptest.""" 54 55 # Compose stressapptest command to execute for memory test 56 cmd = [str(self._STRESSAPPTEST), '-W ' '-M', str(self._mem), '-s', str(self._DURATION_SEC)] 57 logging.info(cmd) 58 results = self.shell.Execute(' '.join(cmd)) 59 logging.info(str(results[const.STDOUT])) 60 61 # Check for 'PASS' and exit code 62 asserts.assertTrue('Status: PASS' in results[const.STDOUT][0].strip(), "Memory test failed.") 63 64 if __name__ == "__main__": 65 test_runner.main() 66
