xos/xos_client/tests/vtr_crud.py

# Copyright 2017-present Open Networking Foundation
#
# 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 __future__ import print_function
from xosapi import xos_grpc_client
import sys

sys.path.append("..")


def test_callback():
    print("TEST: vtr_crud")

    c = xos_grpc_client.coreclient

    sr = c.xos_orm.CordSubscriberRoot.objects.first()
    if not sr:
        print("No subscriber roots!")
        return

    vt = c.xos_orm.VTRTenant.objects.new()
    vt.target = sr
    vt.test = "ping"
    vt.scope = "vm"
    vt.argument = "8.8.8.8"
    vt.save()

    assert vt.id is not None
    assert vt.id > 0

    # Check and make sure we can read it back, pay particular attention to
    # the generic foreign key.
    vt2 = c.xos_orm.VTRTenant.objects.get(id=vt.id)
    assert vt2.target_id == sr.id
    assert vt2.target_type_id == sr.self_content_type_id
    assert "TenantRoot" in vt2.target.class_names

    vt2.delete()

    # now, make sure it has been deleted
    vt3 = c.xos_orm.VTRTenant.objects.filter(id=vt.id)
    assert not vt3

    print("    okay")


xos_grpc_client.start_api_parseargs(test_callback)