#!/usr/bin/env python
'''
Created on 26-Oct-2012

@author: Anil Kumar (anilkumar.s@paxterrasolutions.com)


    TestON is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 2 of the License, or
    (at your option) any later version.

    TestON is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with TestON.  If not, see <http://www.gnu.org/licenses/>.		


fvtapidriver is the basic driver which will handle the fvtapidriver functions
'''
'''
There are two changes required in flowvisor-test framework :

1. In ~/flowvisortests/tests/templatetest.py line : 15 comment 'basic_logger = None'
2. In ~/flowvisortests/tests/testutils.py line : 50 specify config file path CONFIG_FILE = "~/flowvisor-test/tests/tests-base.json"

'''
import pexpect
import struct
import fcntl
import os
import signal
import re
import sys
sys.path.append("../")
from common.apidriver import API
import logging

sys.path.append(path+"/lib/flowvisor-test/tests")
sys.path.append(path+"/lib/flowvisor-test/src/python/")

import templatetest
import testutils
import oftest.cstruct as ofp
import oftest.message as message
import oftest.parse as parse
import oftest.action as action
import oftest.error as error
import socket
import __builtin__

config_default = {
    "param"              : None,
    "fv_cmd"             : "/home/openflow/flowvisor/scripts/flowvisor.sh",
    "platform"           : "local",
    "controller_host"    : "127.0.0.1",
    "controller_port"    : 6633,
    "timeout"            : 3,
    "port_count"         : 4,
    "base_of_port"       : 1,
    "base_if_index"      : 1,
    "test_spec"          : "all",
    "test_dir"           : ".",
    "log_file"           : "/home/openflow/fvt.log",
    "list"               : False,
    "debug"              : "debug",
    "dbg_level"          : logging.DEBUG,
    "port_map"           : {},
    "test_params"        : "None"
}

def test_set_init(config):
    """
    Set up function for basic test classes
    @param config The configuration dictionary; see fvt
    """
    global basic_port_map
    global basic_fv_cmd
    global basic_logger
    global basic_timeout
    global basic_config
    global baisc_logger

    basic_fv_cmd = config["fv_cmd"]
    basic_timeout = config["timeout"]
    basic_port_map = config["port_map"]
    basic_config = config

class FvtApiDriver(API,templatetest.TemplateTest):

    def __init__(self):
        super(API, self).__init__()
        print 'init'
                                                

    def connect(self,**connectargs):
        for key in connectargs:
            vars(self)[key] = connectargs[key]
        
        self.name = self.options['name']
        connect_result = super(API,self).connect()
        self.logFileName = main.logdir+"/"+self.name+".session"
        config_default["log_file"] = self.logFileName
        test_set_init(config_default)
        __builtin__.basic_logger = vars(main)[self.name+'log']
        __builtin__.basic_logger.info("Calling my test setup")
        self.setUp(basic_logger)

        (self.fv, self.sv, sv_ret, ctl_ret, sw_ret) = testutils.setUpTestEnv(self, fv_cmd=basic_fv_cmd)
        
        self.chkSetUpCondition(self.fv, sv_ret, ctl_ret, sw_ret)
        return main.TRUE

    def simplePacket(self,dl_src):
        dl_src = vars(testutils)[dl_src]
        return testutils.simplePacket(dl_src = dl_src)
   
    def genPacketIn(self, in_port, pkt):
        return testutils.genPacketIn(in_port=in_port, pkt=pkt)
     
    def ofmsgSndCmp(self, snd_list, exp_list, xid_ignore=True, hdr_only=True):
        return testutils.ofmsgSndCmp(self, snd_list, exp_list, xid_ignore, hdr_only)
    
    def setRule(self,sv,rule,num_try) :
        return testutils.setRule(self,sv,rule,num_try)
    
    def chkFlowdb(self,controller_number,switch_number,exp_count,exp_rewrites) :
        return testutils.chkFlowdb(self,controller_number,switch_number,exp_count,exp_rewrites)
    
    def chkSwitchStats(self, switch_number, ofproto, exp_snd_count, exp_rcv_count):
        return testutils.chkSwitchStats(self, switch_number, ofproto, exp_snd_count, exp_rcv_count)
    
    def chkSliceStats(self,controller_number,ofproto,exp_snd_count,exp_rcv_count) :
        return testutils.chkSliceStats(self,controller_number,ofproto,exp_snd_count,exp_rcv_count)
    
    def recvStats(self,swId,typ) :
        return testutils.recvStats(self,swId,typ)
    
    def ofmsgSndCmpWithXid(self,snd_list,exp_list,xid_ignore,hdr_only) :
        return testutils.ofmsgSndCmpWithXid(self,snd_list,exp_list,xid_ignore,hdr_only)
    
    def genPacketOut(self,xid,buffer_id,in_port,action_ports,pkt) :
        return testutils.genPacketOut(self,xid,buffer_id,in_port,action_ports,pkt)
    
    def genFlowModFlush(self) :
        return testutils.genFlowModFlush()
    
    def genPhyPort(self,name,addr,port_no) :
        return testutils.genPhyPort(name,addr,port_no)
    
    def disconnect(self,handle):
        response = ''
        '''
        if self.handle:
            self.handle = handle
            response = self.execute(cmd="exit",prompt="(.*)",timeout=120)
        else :
            main.log.error("Connection failed to the host")
            response = main.FALSE
        '''
        return response  
    
    def setUp(self,basic_logger):
        self.logger = basic_logger
        #basic_logger.info("** START TEST CASE " + str(self))
        if basic_timeout == 0:
            self.timeout = None
        else:
            self.timeout = basic_timeout
        self.fv = None
        self.sv = None
        self.controllers = []
        self.switches = []
    
    def close_log_handles(self) :
        self.tearDown() 
        vars(main)[self.name+'log'].removeHandler(self.log_handler)
        #if self.logfile_handler:
        #    self.logfile_handler.close()
        
        return main.TRUE