#!/usr/bin/env python
'''
Created on 12-Feb-2013

@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/>.		


RemoteVMDriver is the basic driver which will handle the Mininet functions
'''

import pexpect
import struct
import fcntl
import os
import signal
import re
import sys
import time

sys.path.append("../")

from drivers.common.cli.remotetestbeddriver import RemoteTestBedDriver

class RemotePoxDriver(RemoteTestBedDriver):
    '''
        RemoteVMDriver is the basic driver which will handle the Mininet functions
    '''
    def __init__(self):
        super(RemoteTestBedDriver, self).__init__()
        
    def connect(self,**connectargs):
        for key in connectargs:
            vars(self)[key] = connectargs[key]
        
        self.name = self.options['name']

        self.handle = super(RemotePoxDriver,self).connect(user_name = self.user_name, ip_address = self.ip_address,port = self.port, pwd = self.pwd)
        if self.handle :
            main.log.info(self.name+" connected successfully ")  
            
            self.execute(cmd="cd "+self.options['pox_lib_location'],prompt="/pox\$",timeout=120)
            self.execute(cmd='./pox.py samples.of_tutorial',prompt="DEBUG:",timeout=120)
            return self.handle
        return main.TRUE
        
    def disconnect(self,handle):
        if self.handle:
            self.execute(cmd="exit()",prompt="/pox\$",timeout=120)
        else :
            main.log.error("Connection failed to the host")