#!/usr/bin/env python3
# SPDX-License-Identifier: LGPL-2.1+
# ~~~
#   runtest.sh of /CoreOS/iproute/Sanity/ip-fou-sanity-test
#   Description: Test basic ip fou funcionality
#
#   Author: Susant Sahani <susant@redhat.com>
#   Copyright (c) 2018 Red Hat, Inc.
# ~~~

import errno
import os
import sys
import time
import unittest
import subprocess
import signal
import shutil

def setUpModule():

    if shutil.which('ip') is None:
        raise OSError(errno.ENOENT, 'ip not found')

class IPFOUTests(unittest.TestCase):

    def test_configure_fou_receive_port_gre(self):
        ''' Configure a FOU receive port for GRE bound to 7777'''

        subprocess.call(" ip fou add port 7777 ipproto 47", shell=True)
        output=subprocess.check_output(['ip', 'fou', 'show']).rstrip().decode('utf-8')
        print(output)
        self.assertRegex(output, "port 7777 ipproto 47")

        subprocess.call("ip fou del port 7777", shell=True)

    def test_configure_fou_receive_port_ipip(self):
        ''' Configure a FOU receive port for IPIP bound to 8888'''

        subprocess.call("ip fou add port 8888 ipproto 4", shell=True)
        output=subprocess.check_output(['ip', 'fou', 'show']).rstrip().decode('utf-8')
        print(output)
        self.assertRegex(output, "port 8888 ipproto 4")

        subprocess.call("ip fou del port 8888", shell=True)

    def test_configure_fou_receive_port_gue(self):
        ''' Configure a GUE receive port bound to 9999 '''

        subprocess.call("ip fou add port 9999 gue", shell=True)
        output=subprocess.check_output(['ip', 'fou', 'show']).rstrip().decode('utf-8')
        print(output)
        self.assertRegex(output, "port 9999 gue")

        subprocess.call("ip fou del port 9999", shell=True)

    def test_configure_fou_with_ipip(self):
        ''' IP over UDP tunnel '''

        subprocess.call("ip fou add port 9000 ipproto 4", shell=True)
        output=subprocess.check_output(['ip', 'fou', 'show']).rstrip().decode('utf-8')
        print(output)
        self.assertRegex(output, "port 9000 ipproto 4")

        subprocess.call("ip link add name tunudp type ipip remote 192.168.2.2 local 192.168.2.1 ttl 225 encap fou encap-sport auto encap-dport 9000", shell=True)
        output=subprocess.check_output(['ip', '-d', 'link', 'show', 'tunudp']).rstrip().decode('utf-8')
        self.assertRegex(output, "encap fou")

        subprocess.call("ip link del tunudp", shell=True)
        subprocess.call("ip fou del port 9000", shell=True)

if __name__ == '__main__':
    unittest.main(testRunner=unittest.TextTestRunner(stream=sys.stdout,
                                                     verbosity=2))