#!/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()