TestON/tests/SCPFmaxIntents/Dependency/maxIntentFunctions.py

'''
    Wrapper functions for maxIntent
'''

import json
import time

def __init__( self ):
    self.default = ""

def getIntents( main, state="INSTALLED", sleep=1, timeout=120 ):
    cmd = "intents | grep " + state + " | wc -l"
    main.log.info("Sending: " + cmd)
    main.CLIs[0].handle.sendline(cmd)

    time.sleep(sleep)

    main.CLIs[0].handle.expect("onos>", timeout=timeout)
    raw = main.CLIs[0].handle.before
    intents = int(main.CLIs[0].handle.before.split()[7])
    main.log.info(state + "intents: " + str(intents))
    return intents


def getFlows( main, state="ADDED", sleep=1, timeout=120 ):
    cmd = "flows | grep " + state + " | wc -l"
    main.log.info("Sending: " + cmd)
    main.CLIs[0].handle.sendline(cmd)

    time.sleep(sleep)

    main.CLIs[0].handle.expect("onos>", timeout=timeout)
    raw = main.CLIs[0].handle.before
    flows = int(main.CLIs[0].handle.before.split()[7])
    main.log.info(state + "flows: " + str(flows))
    return flows


def pushIntents( main,
                 switch,
                 ingress,
                 egress,
                 batch,
                 offset,
                 sleep=1,
                 options="",
                 timeout=120):
    '''
        Description
    '''
    cmd = "push-test-intents " + options + " " + switch + ingress + " " +\
            switch + egress + " " + str(batch) + " " + str(offset)
    main.log.info("Installing " + str(offset+batch) + " intents")
    main.log.debug("Sending: " + cmd)
    main.CLIs[0].handle.sendline(cmd)
    time.sleep(sleep)
    main.CLIs[0].handle.expect("onos>", timeout=timeout)

    raw = main.CLIs[0].handle.before
    if "Failure:" in raw or "GC" in raw:
        return main.FALSE
    return main.TRUE

def verifyFlows( main, expectedFlows, state="ADDED", sleep=1,  timeout=120):
    '''
        This function returns main.TRUE if the number of expected flows are in
        the specified state

        @params
            expectedFlows: the flows you expect to see in the specified state
            state: the state of the flow to check for
            timeout: the timeout for pexpect
    '''
    cmd = "flows | grep " + state + " | wc -l"
    for i in range(10):
        flows = getFlows( main, state, sleep, timeout )
        if expectedFlows == flows:
            return main.TRUE

    return main.FALSE

def verifyIntents( main, expectedIntents, state="INSTALLED", sleep=1, timeout=120):
    '''
        This function returns main.TRUE if the number of expected intents are in
        the specified state

        @params
            expectedFlows: the intents you expect to see in the specified state
            state: the state of the intent to check for
            timeout: the timeout for pexpect
    '''
    cmd = "intents | grep " + state + " | wc -l"
    for i in range(10):
        intents = getIntents( main, state, sleep, timeout )
        if expectedIntents == intents:
            return main.TRUE

    return main.FALSE