#!/usr/bin/expect

if {$argc == 1} {
	set command     [lindex $argv 0]
} else {
	set command     ""
}


set timeout 2
expect_after default {puts stderr "\nFAIL"; exit 1}

spawn /bin/bash
expect "# "

send "id\r"
expect "uid=0(root) gid=0(root) groups=0(root)\r"
expect "# "

send "export PATH=bar:\$PATH\r"
expect "# "
send "echo \"PATH=\\\"\$PATH\\\"\"\r"
expect "# "

#=============================================================================
#
# Regular su to root, preserve environment
# However, PATH is reset
#
#=============================================================================
send "/bin/su $command\r"
expect "# "

send "id\n"
expect "uid=0(root) gid=0(root) groups=0(root)\r"
expect "# "

send_user "\n# PATH should be '/sbin:/bin:/usr/sbin:/usr/bin'"
send "\r"
expect "# "

send "echo \"PATH=\\\"\$PATH\\\"\"\r"
expect "PATH=\"/sbin:/bin:/usr/sbin:/usr/bin\"\r"
expect "# "

send "echo \"'\$HOME'\$USER'\$LOGNAME'\$SHELL'\"\r"
expect "'/root'root'root'/bin/bash'\r"
expect "# "

send "exit\r"
expect "# "

puts "\nPASS"
exit 0

