blob: e851b642ff18869d4495f78c241fd8030e037f29 [file] [log] [blame]
#!/usr/bin/env python
#
# Copyright 2017 the original author or authors.
#
# 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.
#
from time import time, sleep
import logging
import os
import json
from unittest import TestCase,main
this_dir = os.path.abspath(os.path.dirname(__file__))
from tests.itests.test_utils import run_command_to_completion_with_raw_stdout
from voltha.protos.device_pb2 import Device
from google.protobuf.json_format import MessageToDict
from tests.itests.voltha.rest_base import RestBase
from common.utils.consulhelpers import get_endpoint_from_consul
log = logging.getLogger(__name__)
DOCKER_COMPOSE_FILE = "compose/docker-compose-ofagent-test.yml"
LOCAL_CONSUL = "localhost:8500"
command_defs = dict(
docker_images="docker images",
docker_stop="docker stop",
docker_rm="docker rm",
docker_voltha_logs="docker logs -f compose_voltha_1",
docker_compose_logs="docker-compose -f {} logs".format(
DOCKER_COMPOSE_FILE),
docker_stop_and_remove_all_containers="docker stop `docker ps -q` ; "
"docker rm `docker ps -a -q`",
docker_compose_start_all="docker-compose -f {} up -d "
.format(DOCKER_COMPOSE_FILE),
docker_compose_stop="docker-compose -f {} stop"
.format(DOCKER_COMPOSE_FILE),
docker_compose_rm_f="docker-compose -f {} rm -f"
.format(DOCKER_COMPOSE_FILE),
docker_compose_ps="docker-compose -f {} ps".format(DOCKER_COMPOSE_FILE),
docker_ps="docker ps",
onos_form_cluster="./tests/itests/ofagent/onos-form-cluster",
onos1_ip="docker inspect --format '{{ .NetworkSettings.Networks.compose_default.IPAddress }}' onos1",
onos2_ip ="docker inspect --format '{{ .NetworkSettings.Networks.compose_default.IPAddress }}' onos2",
onos3_ip="docker inspect --format '{{ .NetworkSettings.Networks.compose_default.IPAddress }}' onos3",
onos1_devices="curl -u karaf:karaf http://localhost:8181/onos/v1/devices",
onos2_devices="curl -u karaf:karaf http://localhost:8182/onos/v1/devices",
onos3_devices="curl -u karaf:karaf http://localhost:8183/onos/v1/devices")
class TestOFAGENT_MultiController(RestBase):
# Test OFAgent Support for Multiple controller
def setUp(self):
# Run Voltha,OFAgent,3 ONOS and form ONOS cluster.
print "Starting all containers ..."
cmd = command_defs['docker_compose_start_all']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
print "Waiting for all containers to be ready ..."
sleep(80)
cmd = command_defs['onos1_ip']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos1_ip = out
print "ONOS1 IP is {}".format(onos1_ip)
cmd = command_defs['onos2_ip']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos2_ip = out
print "ONOS2 IP is {}".format(onos2_ip)
cmd = command_defs['onos3_ip']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos3_ip = out
print "ONOS3 IP is {}".format(onos3_ip)
cmd = command_defs['onos_form_cluster'] + ' {} {} {}'.format(onos1_ip.strip(),onos2_ip.strip(),onos3_ip.strip())
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
print "Cluster Output :{} ".format(out)
self.get_rest_endpoint()
def tearDown(self):
# Stopping and Removing Voltha,OFAgent,3 ONOS.
print "Stopping and removing all containers ..."
cmd = command_defs['docker_compose_stop']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
print "Waiting for all containers to be stopped ..."
sleep(1)
cmd = command_defs['docker_compose_rm_f']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
def wait_till(self, msg, predicate, interval=0.1, timeout=5.0):
deadline = time() + timeout
while time() < deadline:
if predicate():
return
sleep(interval)
self.fail('Timed out while waiting for condition: {}'.format(msg))
def get_rest_endpoint(self):
# Retrieve details on the REST entry point
rest_endpoint = get_endpoint_from_consul(LOCAL_CONSUL, 'voltha-envoy-8443')
# Construct the base_url
self.base_url = 'https://' + rest_endpoint
def add_device(self):
print "Adding device"
device = Device(
type='simulated_olt',
mac_address='01:0c:e2:31:40:00'
)
device = self.post('/api/v1/devices', MessageToDict(device),
expected_http_code=200)
print "Added device - id:{}, type:{}".format(device['id'], device['type'])
sleep(5)
return device
def enable_device(self, device_id):
print "Enabling device - id:{}".format(device_id)
path = '/api/v1/devices/{}'.format(device_id)
self.post(path + '/enable', expected_http_code=200)
device = self.get(path)
self.assertEqual(device['admin_state'], 'ENABLED')
self.wait_till(
'admin state moves to ACTIVATING or ACTIVE',
lambda: self.get(path)['oper_status'] in ('ACTIVATING', 'ACTIVE'),
timeout=0.5)
# eventually, it shall move to active state and by then we shall have
# device details filled, connect_state set, and device ports created
self.wait_till(
'admin state ACTIVE',
lambda: self.get(path)['oper_status'] == 'ACTIVE',
timeout=0.5)
device = self.get(path)
images = device['images']
image = images['image']
image_1 = image[0]
version = image_1['version']
self.assertNotEqual(version, '')
self.assertEqual(device['connect_status'], 'REACHABLE')
ports = self.get(path + '/ports')['items']
self.assertEqual(len(ports), 2)
sleep(30)
print "Enabled device - id:{}".format(device_id)
def test_ofagent_controller_failover(self):
olt_device = self.add_device()
print "Output of ADD OLT is {} {} {}".format(olt_device, type(olt_device), olt_device['id'])
sleep(5)
self.enable_device(olt_device['id'])
print "Waiting for OLT device to be activated ..."
sleep(80)
cmd = command_defs['onos1_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos1_devices = json.loads(out)
onos1_role = onos1_devices['devices'][0]['role']
print "Role of ONOS1 is {}".format(onos1_role)
cmd = command_defs['onos2_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos2_devices = json.loads(out)
onos2_role = onos2_devices['devices'][0]['role']
print "Role of ONOS2 is {}".format(onos2_role)
cmd = command_defs['onos3_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos3_devices = json.loads(out)
onos3_role = onos3_devices['devices'][0]['role']
print "Role of ONOS3 is {}".format(onos3_role)
if onos1_role == "MASTER":
cmd = command_defs['docker_stop']+ ' onos1'
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
print "Waiting for ONOS to Elect New Master"
sleep(20)
cmd = command_defs['onos2_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos2_devices = json.loads(out)
onos2_role = onos2_devices['devices'][0]['role']
print "Role of ONOS2 is {}".format(onos2_role)
cmd = command_defs['onos3_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos3_devices = json.loads(out)
onos3_role = onos3_devices['devices'][0]['role']
print "Role of ONOS3 is {}".format(onos3_role)
assert (onos3_role == "MASTER" or onos2_role == "MASTER"), "Exception,New Master Election Failed"
elif onos2_role == "MASTER":
cmd = command_defs['docker_stop']+ ' onos2'
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
print "Waiting for ONOS to Elect New Master"
sleep(20)
cmd = command_defs['onos1_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos1_devices = json.loads(out)
onos1_role = onos1_devices['devices'][0]['role']
print "Role of ONOS1 is {}".format(onos1_role)
cmd = command_defs['onos3_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos3_devices = json.loads(out)
onos3_role = onos3_devices['devices'][0]['role']
print "Role of ONOS3 is {}".format(onos3_role)
assert (onos3_role == "MASTER" or onos1_role == "MASTER"), "Exception,New Master Election Failed"
elif onos3_role == "MASTER":
cmd = command_defs['docker_stop']+ ' onos3'
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
print "Waiting for ONOS to Elect New Master"
sleep(20)
cmd = command_defs['onos1_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos1_devices = json.loads(out)
onos1_role = onos1_devices['devices'][0]['role']
print "Role of ONOS1 is {}".format(onos1_role)
cmd = command_defs['onos2_devices']
out, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
onos2_devices = json.loads(out)
onos2_role = onos2_devices['devices'][0]['role']
print "Role of ONOS2 is {}".format(onos2_role)
assert (onos1_role == "MASTER" or onos2_role == "MASTER"), "Exception,New Master Election Failed"