test.py

#!/usr/bin/env python3
from __future__ import print_function
from future import standard_library
standard_library.install_aliases()
from builtins import str
import os
import signal
import sys
import subprocess
import _thread
import time
import atexit

includeos_src = os.environ.get('INCLUDEOS_SRC',
                               os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__))).split('/test')[0])
sys.path.insert(0,includeos_src)

from vmrunner import vmrunner
import requests
expected_string = "#" * 1024 * 1024 * 50

def validateRequest(addr):
    response = requests.get('https://10.0.0.68:443', verify=False, timeout=5)
    #print len(response.content)
    return (response.content) == str(addr) + expected_string

# start nodeJS
pro = subprocess.Popen(["node", "server.js"], stdout=subprocess.PIPE)

requests_completed = False
def startBenchmark(line):
    print("<test.py> starting test ")
    assert validateRequest(6001)
    assert validateRequest(6002)
    assert validateRequest(6003)
    assert validateRequest(6004)

    assert validateRequest(6001)
    assert validateRequest(6002)
    assert validateRequest(6003)
    assert validateRequest(6004)
    print("Waiting for TCP MSL end...")
    global requests_completed
    requests_completed = True
    return True
def mslEnded(line):
    if requests_completed:
        vm._on_success("SUCCESS")
    return True

@atexit.register
def cleanup():
    print("<test.py> Stopping node server")
    # stop nodeJS
    pro.kill()


# Get an auto-created VM from the vmrunner
vm = vmrunner.vms[0]

# Add custom event for testing server
vm.on_output("MicroLB ready for test", startBenchmark)
vm.on_output("TCP MSL ended", mslEnded)

if len(sys.argv) > 1:
    vm.boot(image_name=str(sys.argv[1]))
else:
    # Boot the VM, taking a timeout as parameter
    vm.boot(20,image_name='net_microLB.elf.bin')