#
# Copyright 2016-present Ciena Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import unittest
import os,sys
import keystoneclient.v2_0.client as ksclient
import keystoneclient.apiclient.exceptions
import neutronclient.v2_0.client as nclient
import neutronclient.common.exceptions
from novaclient import client as nova_client
from neutronclient.v2_0 import client as neutron_client
import neutronclient.v2_0.client as neutronclient
from nose.tools import assert_equal
from CordTestUtils import get_mac, log_test
from onosclidriver import OnosCliDriver
from OnosCtrl import OnosCtrl
from OnosFlowCtrl import OnosFlowCtrl
from OnboardingServiceUtils import OnboardingServiceUtils
from SSHTestAgent import SSHTestAgent
from CordTestUtils import running_on_pod, getstatusoutput
from CordLogger import CordLogger
from CordTestUtils import log_test as log
import requests
import time
import json
from VSGAccess import VSGAccess
from CordTestConfig import setup_module, running_on_ciab
from vsgTest import *
log.setLevel('INFO')

class onboarding_exchange(CordLogger):
    ONOS_INSTANCES = 3
    V_INF1 = 'veth0'
    device_id = 'of:' + get_mac()
    TEST_IP = '8.8.8.8'
    HOST = "10.1.0.1"
    USER = "vagrant"
    PASS = "vagrant"
    head_node = os.getenv('HEAD_NODE', 'prod')
    HEAD_NODE = head_node + '.cord.lab' if len(head_node.split('.')) == 1 else head_node
    test_path = os.path.dirname(os.path.realpath(__file__))
    on_pod = running_on_pod()
    vcpe_dhcp = 'vcpe0.222.111'
    vsg_exchange = vsg_exchange()
    vm_name = 'mysite_exampleservice'

    @classmethod
    def setUpClass(cls):
        OnboardingServiceUtils.setUp()

    @classmethod
    def tearDownClass(cls):
        OnboardingServiceUtils.tearDown()

    def cliEnter(self,  controller = None):
        retries = 0
        while retries < 30:
            self.cli = OnosCliDriver(controller = controller, connect = True)
            if self.cli.handle:
                break
            else:
                retries += 1
                time.sleep(2)

    def cliExit(self):
        self.cli.disconnect()

    def onos_shutdown(self,  controller = None):
        status = True
        self.cliEnter(controller = controller)
        try:
            self.cli.shutdown(timeout = 10)
        except:
            log.info('Graceful shutdown of ONOS failed for controller: %s' %controller)
            status = False

        self.cliExit()
        return status

    def get_exampleservice_vm_public_ip(self, vm_name = 'mysite_exampleservice'):
	if not vm_name:
		vm_name = self.vm_name
	exampleservices = OnboardingServiceUtils.get_exampleservices()
	for service in exampleservices:
		if vm_name in service.name:
			return service.get_public_ip()
	return None

    def test_exampleservice_health(self):
        """
        Algo:
        1. Login to compute node VM
        2. Get all exampleservice
        3. Ping to all exampleservice
        4. Verifying Ping success
        """
        status = OnboardingServiceUtils.health_check()
        assert_equal(status, True)

    def test_exampleservice_for_login(self):
        if self.on_pod is False:
            return
        exampleservices = OnboardingServiceUtils.get_exampleservices()
	exampleservice_access_status = map(lambda exampleservice: exampleservice.check_access(), exampleservices)
        status = filter(lambda st: st == False, exampleservice_access_status)
        assert_equal(len(status), 0)

    def test_exampleservice_for_default_route_through_testclient(self):
        if self.on_pod is False:
           return
        ssh_agent = SSHTestAgent(host = self.HEAD_NODE, user = self.USER, password = self.PASS)
        cmd = "sudo lxc exec testclient -- route | grep default"
        status, output = ssh_agent.run_cmd(cmd)
        assert_equal(status, True)

    def test_exampleservice_for_service_access_through_testclient(self):
        if self.on_pod is False:
            return
        ssh_agent = SSHTestAgent(host = self.HEAD_NODE, user = self.USER, password = self.PASS)
        cmd = "sudo lxc exec testclient -- ping -c 3 8.8.8.8"
        status, output = ssh_agent.run_cmd(cmd)
        assert_equal( status, True)

    def test_exampleservice_for_service_reachability_from_cord_tester(self, vcpe_intf=None):
        if self.on_pod is False:
            return
	if not vcpe_intf:
		vcpe_intf = self.dhcp_vcpes_reserved[0]
	vm_public_ip = self.get_exampleservice_vm_public_ip()
	self.vsg_exchange.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
        vm_public_ip = self.get_exampleservice_vm_public_ip()
	st, _ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
        assert_equal(st, False)
	self.vsg_exchange.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)

    def test_exampleservice_operational_status_from_testclient(self):
	vm_public_ip = self.get_exampleservice_vm_public_ip()
        ssh_agent = SSHTestAgent(host = self.HEAD_NODE, user = self.USER, password = self.PASS)
        cmd = 'sudo lxc exec testclient -- apt-get install -y curl'
        status, _  = ssh_agent.run_cmd(cmd)
        assert_equal(status, True)
        #Test connectivity to ExampleService from test client
        cmd = 'sudo lxc exec testclient -- curl -s http://{}'.format(vm_public_ip)
        status, output = ssh_agent.run_cmd(cmd)
        assert_equal(status, True)

    def test_exampleservice_operational_access_from_cord_tester(self, vcpe_intf=None):
        if self.on_pod is False:
            return
        if not vcpe_intf:
                vcpe_intf = self.vcpe_dhcp
	vm_public_ip = self.get_exampleservice_vm_public_ip()
        self.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
        st, _ = getstatusoutput('curl -s http://{}'.format(vm_public_ip))
        assert_equal(st, False)

    def test_exampleservice_for_service_message(self, service_message='"'+'hello'+'"'):
	vm_public_ip = self.get_exampleservice_vm_public_ip()
	vcpe_intf = self.vcpe_dhcp
	try:
	    self.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
            st,out = getstatusoutput('curl -s http://{}'.format(vm_public_ip))
            assert_equal(st, False)
            output = out.split('\n')
	    srvs_msg = ''
            for line in output:
                line = line.split(':')
                if line[0].strip() == 'Service Message':
                    srvs_msg = line[1].strip()
	    assert_equal(service_message, srvs_msg)
        finally:
            self.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)

    def test_exampleservice_for_tenant_message(self, tenant_message='"'+'world'+'"'):
	vcpe_intf = self.vcpe_dhcp
        vm_public_ip = self.get_exampleservice_vm_public_ip()
        try:
            self.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
            st,out = getstatusoutput('curl -s http://10.6.1.194')
            assert_equal(st, False)
            output = out.split('\n')
            tnt_msg = ''
            for line in output:
                line = line.split(':')
                if line[0].strip() == 'Tenant Message':
                    tnt_msg = line[1].strip()
            assert_equal(tenant_message, tnt_msg)
        finally:
            self.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)

    def test_exampleservice_access_after_subscriber_interface_toggle(self, vcpe_intf=None):
        if not vcpe_intf:
                vcpe_intf = self.vcpe_dhcp
        vm_public_ip = self.get_exampleservice_vm_public_ip()
        try:
            self.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
            #curl request from test container
            cmd = 'curl -s http://{}'.format(vm_public_ip)
            st,_ = getstatusoutput(cmd)
            assert_equal(st, False)
            st,_ = getstatusoutput('ifconfig {} down'.format(vcpe_intf))
	    time.sleep(1)
            assert_equal(st, False)
            st,_ = getstatusoutput(cmd)
            assert_equal(st, False)
            st,_ = getstatusoutput('ifconfig {} up'.format(vcpe_intf))
            time.sleep(1)
	    self.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
	    st,_ = getstatusoutput(cmd)
            assert_equal(st, False)
        finally:
            self.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)

    def test_exampleservice_access_after_service_paused(self, vcpe_intf=None,vm_name=None):
        if not vcpe_intf:
                vcpe_intf = self.vcpe_dhcp
	if not vm_name:
		vm_name = self.vm_name
	vm_public_ip = self.get_exampleservice_vm_public_ip()
        self.vsg_exchange.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
        st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
        assert_equal(st, False)
        exampleservices = OnboardingServiceUtils.get_exampleservices()
	status = False
        for service in exampleservices:
                if self.vm_name in service.name:
                        log.info('pausing mysite-example-server')
			service.pause()
			time.sleep(1)
        		st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
        		assert_equal(st, True)
			service.unpause()
			self.vsg_exchange.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
			status = True
	assert_equal(status, True)

    def test_exampleservice_access_after_service_is_suspended(self, vcpe_intf=None,vm_name=None):
        if not vcpe_intf:
                vcpe_intf = self.vcpe_dhcp
        if not vm_name:
                vm_name = self.vm_name
        vm_public_ip = self.get_exampleservice_vm_public_ip()
        self.vsg_exchange.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
        st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
        assert_equal(st, False)
        exampleservices = OnboardingServiceUtils.get_exampleservices()
        status = False
        for service in exampleservices:
                if self.vm_name in service.name:
                        log.info('suspending mysite-example-server')
                        service.suspend()
                        time.sleep(5)
                        st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
                        assert_equal(st, True)
                        service.resume()
                        self.vsg_exchange.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
                        status = True
        assert_equal(status, True)

    def test_exampleservice_access_after_service_restart(self, vcpe_intf=None,vm_name=None):
        if not vcpe_intf:
                vcpe_intf = self.vcpe_dhcp
        if not vm_name:
                vm_name = self.vm_name
        vm_public_ip = self.get_exampleservice_vm_public_ip()
        self.vsg_exchange.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
        st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
        assert_equal(st, False)
        exampleservices = OnboardingServiceUtils.get_exampleservices()
        status = False
        for service in exampleservices:
                if self.vm_name in service.name:
                        log.info('restarting mysite-example-server')
                        service.reboot()
                        time.sleep(30)
			self.vsg_exchange.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
                        st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
                        assert_equal(st, False)
                        self.vsg_exchange.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
                        status = True
        assert_equal(status, True)

    def test_exampleservice_access_after_service_stop(self, vcpe_intf=None,vm_name=None):
        if not vcpe_intf:
                vcpe_intf = self.vcpe_dhcp
        if not vm_name:
                vm_name = self.vm_name
        vm_public_ip = self.get_exampleservice_vm_public_ip()
        self.vsg_exchange.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
        st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
        assert_equal(st, False)
        exampleservices = OnboardingServiceUtils.get_exampleservices()
        status = False
        for service in exampleservices:
                if self.vm_name in service.name:
                        log.info('restarting mysite-example-server')
                        service.stop()
                        time.sleep(1)
                        st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
                        assert_equal(st, True)
			service.start()
                        self.vsg_exchange.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
                        status = True
        assert_equal(status, True)

    def test_exampleservice_for_service_message_after_service_stop_and_start(self, service_message='"'+'hello'+'"'):
        vm_public_ip = self.get_exampleservice_vm_public_ip()
        vcpe_intf = self.vcpe_dhcp
        try:
            self.vsg_exchange.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
            st,out = getstatusoutput('curl -s http://{}'.format(vm_public_ip))
            assert_equal(st, False)
	    exampleservices = OnboardingServiceUtils.get_exampleservices()
	    status = False
            for service in exampleservices:
                if self.vm_name in service.name:
                        log.info('stopping mysite-example-server')
                        service.stop()
                        time.sleep(5)
                        st,_ = getstatusoutput('ping -c 1 {}'.format(vm_public_ip))
                        assert_equal(st, True)
                        service.start()
			self.vsg_exchange.add_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
			time.sleep(50)
                        st,out = getstatusoutput('curl -s http://{}'.format(vm_public_ip))
                        assert_equal(st, False)
                        output = out.split('\n')
                        srvs_msg = ''
                        for line in output:
                            line = line.split(':')
                            if line[0].strip() == 'Service Message':
                                srvs_msg = line[1].strip()
				break
                        assert_equal(service_message, srvs_msg)
		        self.vsg_exchange.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)
                        status = True
			break
	    assert_equal(status,True)
        finally:
            self.vsg_exchange.del_static_route_via_vcpe_interface([vm_public_ip],vcpe=vcpe_intf)

    def test_multiple_subcriber_access_for_same_exampleservice(self):
        ssh_agent = SSHTestAgent(host = self.HEAD_NODE, user = self.USER, password = self.PASS)
        cmd = "nova list --all-tenants|grep mysite_exampleservice|cut -d '|' -f 2"
        status, nova_id = ssh_agent.run_cmd(cmd)
        assert_equal(status, True)
        #Get public IP of VM
        cmd = 'nova interface-list {} |grep -o -m 1 10\.6\.[[:digit:]]*\.[[:digit:]]*'.format(nova_id)
        status, public_ip = ssh_agent.run_cmd(cmd)
        assert_equal(status, True)
        for vcpe in self.dhcp_vcpes:
            self.add_static_route_via_vcpe_interface([public_ip],vcpe=vcpe)
            #curl request from test container
            curl_cmd = 'curl -s http://{}'.format(public_ip)
            st,_ = getstatusoutput(curl_cmd)
            assert_equal(st, True)
            self.del_static_route_via_vcpe_interface([public_ip],vcpe=vcpe)
            time.sleep(1)

    def test_exampleservice_after_vcpe_instance_restart(self, vcpe_intf=None,vcpe_name=None):
        if not vcpe_intf:
                vcpe_intf = self.dhcp_vcpes_reserved[0]
        if not vcpe_name:
                vcpe_name = self.container_vcpes_reserved[0]
        public_ip = self.get_exampleservice_vm_public_ip()
        vsg = VSGAccess.get_vcpe_vsg(vcpe_name)
        try:
            self.add_static_route_via_vcpe_interface([public_ip],vcpe=vcpe_intf)
            #curl request from test container
            curl_cmd = 'curl -s http://{}'.format(public_ip)
            st,_ = getstatusoutput(curl_cmd)
            assert_equal(st, True)
            #restarting example service VM
            cmd = 'sudo docker restart {}'.format(vcpe_name)
            status, _ = vsg.run_cmd(cmd)
            assert_equal(status, True)
            time.sleep(10)
            st,_ = getstatusoutput(curl_cmd)
            assert_equal(st, True)
        finally:
            self.del_static_route_via_vcpe_interface([public_ip],vcpe=vcpe_intf)

    def test_exampleservice_after_firewall_rule_added_to_drop_service_running_server_ip_in_vcpe(self):
        if not vcpe_intf:
                vcpe_intf = self.dhcp_vcpes_reserved[0]
        if not vcpe_name:
                vcpe_name = self.container_vcpes_reserved[0]
        public_ip = self.get_exampleservice_vm_public_ip()
        vsg = VSGAccess.get_vcpe_vsg(vcpe_name)
        try:
            self.add_static_route_via_vcpe_interface([public_ip],vcpe=vcpe_intf)
            #curl request from test container
            curl_cmd = 'curl -s http://{}'.format(public_ip)
            st,_ = getstatusoutput(curl_cmd)
            assert_equal(st, True)
            #restarting example service VM
            st,_ = vsg.run_cmd('sudo docker exec {} iptables -I FORWARD -d {} -j ACCEPT'.format(vcpe_name,public_ip))
            time.sleep(1)
            st,_ = getstatusoutput(curl_cmd)
            assert_equal(st, False)
        finally:
            st,_ = vsg.run_cmd('sudo docker exec {} iptables -D FORWARD -d {} -j DROP'.format(vcpe_name,public_ip))
            self.del_static_route_via_vcpe_interface([public_ip],vcpe=vcpe_intf)