#!/usr/bin/expect -f
#
#  Author(s)  : Jean-Paul Saman
#
#  Description: Expect script that handles the interactive
#               telnet session. It opens a connection then
#               test the program command (and show program)
#               and closes the connection
#
#  Arg1       : Hostname of IP address
#  Arg2       : Port address
#  Arg3       : Username
#  Arg4       : Password
#
#  Return     : 0  The session succeeded
#               -1 The session failed
#


set Site [lindex $argv 0]
set Port [lindex $argv 1]
set User [lindex $argv 2]
set Password [lindex $argv 3]

set timeout 30
spawn telnet $Site $Port
expect {
    "Login:" { }
    "unknown" {
        send "quit\r"
        send_user "Error: login expected!\n"
        exit -1
    }
    "failure" {
       send "quit\r"
       send_user "Error: could not start a telnet session\n"
       exit -1
    }
    timeout {
       send "quit\r"
       send_user "Error: (timeout) login expected!\n"
       exit -1
    }
}
#  Sometime the telnet session hangs, probably due to
#  the fast inlogging (scripting), the following sleep
#  commands solve this hangup.
sleep 1
send "$User\r"
sleep 1
expect {
    "Password:" { }
    timeout {
        send "quit\r"
        send_user "Error: (timeout) Password expected!\n"
        exit -1
    }
}

send "$Password\r"
expect {
    "$User@vls>" { }
    "incorrect" {
        send_user "Error: remote user prompt expected!\n"
        exit -1
    }
    timeout {
       send "quit\r"
       send_user "Error: (timeout) remote user prompt expected!\n"
       exit -1
    }
}

# Test input command
send "input myinput video\r"
expect {
    "Status: 0" {
       send_user "Error: Adding input should not work.\n"
       exit -1
    }
    "Error: -1" {
    }
    timeout {
       send "quit\r"
       send_user "Error: (timeout) Adding input command expected to succeed!\n"
       exit -1
    }
}

send "input myinput video --add\r"
expect {
    "Status: 0" {
       send_user "Error: Adding input should not work.\n"
       exit -1
    }
    "Error: -1" {
    }
    timeout {
       send "quit\r"
       send_user "Error: (timeout) Adding input command expected to succeed!\n"
       exit -1
    }
}

send "input myinput video /dev/video bogey --add\r"
expect {
    "Status: 0" {
       send_user "Error: Adding input should not work.\n"
       exit -1
    }
    "Error: -1" {
    }
    timeout {
       send "quit\r"
       send_user "Error: (timeout) Adding input command expected to succeed!\n"
       exit -1
    }
}

# Success adding and delete the same
for { set Index 20 } { $Index>0 } {} {
    send "input myinput video /dev/video Mpeg2-PS --add\r"
    expect {
        "Status: 0" {
        }
        "Error: -1" {
           send_user "Error: Adding input should not work.\n"
           exit -1
        }
        timeout {
           send "quit\r"
           send_user "Error: (timeout) Adding input command expected to succeed!\n"
           exit -1
        }
    }

    send "input myinput video /dev/video Mpeg2-PS --delete\r"
    expect {
        "Status: 0" {
        }
        "Error: -1" {
           send_user "Error: Adding input should work.\n"
           exit -1
        }
        timeout {
           send "quit\r"
           send_user "Error: (timeout) Adding input command expected to succeed!\n"
           exit -1
        }
    }
    incr Index -1
}

# Stress test
for { set Index 100 } { $Index>0 } {} {

    send "input myinput${Index} video /dev/video Mpeg2-PS --add\r"
    expect {
        "Status: 0" {
        }
        "Error: -1" {
           send_user "Error: Adding input should work.\n"
           exit -1
        }
        timeout {
           send "quit\r"
           send_user "Error: (timeout) Adding input command expected to succeed!\n"
           exit -1
        }
    }

    send "input myinput${Index} video /dev/video Mpeg2-PS --delete\r"
    expect {
        "Status: 0" {
        }
        "Error: -1" {
           send_user "Error: Adding input should work.\n"
           exit -1
        }
        timeout {
           send "quit\r"
           send_user "Error: (timeout) Adding input command expected to succeed!\n"
           exit -1
        }
    }

    incr Index -1
}

# logging out
send "logout\r"
expect "Closing connection\rConnection closed by foreign host.\r"

#  End script