Commit 95600af3 authored by Adrian Pino's avatar Adrian Pino
Browse files

Fix import issue

parent 8bbf5647

tests/network/test_create_qod_session.py

+3 −3
Original line number Diff line number Diff line 
# -*- coding: utf-8 -*-

import pytest



from sunrise6g_opensdk.common.sdk_catalog_client import SdkCatalogClient

from sunrise6g_opensdk.common.sdk import Sdk as sdkclient

from sunrise6g_opensdk.network.clients.open5gs.client import NetworkManager



OPEN5GS_TEST_CASES = [
@@ -21,7 +21,7 @@ OPEN5GS_TEST_CASES = [ 
    ids=["open5gs"],

)

def test_valid_input_open5gs(client_specs):

    network_client: NetworkManager = SdkCatalogClient.create_clients_from(client_specs)[

    network_client: NetworkManager = sdkclient.create_clients_from(client_specs)[

        "network"

    ]
@@ -46,7 +46,7 @@ def test_valid_input_open5gs(client_specs): 
    ids=["open5gs"],

)

def test_create_qod_session_open5gs(client_specs):

    network_client: NetworkManager = SdkCatalogClient.create_clients_from(client_specs)[

    network_client: NetworkManager = sdkclient.create_clients_from(client_specs)[

        "network"

    ]

tests/network/test_create_traffic_influence.py

+3 −7
Original line number Diff line number Diff line 
# # -*- coding: utf-8 -*-

import pytest



from sunrise6g_opensdk.network.core.network_factory import NetworkClientFactory

from sunrise6g_opensdk.common.sdk import Sdk as sdkclient



test_cases = [("oai", "http://127.0.0.1/", "scs-oai")]





@pytest.mark.parametrize("client_name, base_url, scs_as_id", test_cases)

def test_valid_input(client_name, base_url, scs_as_id):

    network_client = NetworkClientFactory.create_network_client(

        client_name, base_url, scs_as_id

    )

    network_client = sdkclient.create_network_client(client_name, base_url, scs_as_id)



    ti_session = {

        "device": {
@@ -26,9 +24,7 @@ def test_valid_input(client_name, base_url, scs_as_id): 


@pytest.mark.parametrize("client_name, base_url, scs_as_id", test_cases)

def test_create_traffic_influence(client_name, base_url, scs_as_id):

    network_client = NetworkClientFactory.create_network_client(

        client_name, base_url, scs_as_id

    )

    network_client = sdkclient.create_network_client(client_name, base_url, scs_as_id)



    ti_session = {

        "device": {