4 # Copyright 2016 RIFT.IO Inc
6 # Licensed under the Apache License, Version 2.0 (the "License");
7 # you may not use this file except in compliance with the License.
8 # You may obtain a copy of the License at
10 # http://www.apache.org/licenses/LICENSE-2.0
12 # Unless required by applicable law or agreed to in writing, software
13 # distributed under the License is distributed on an "AS IS" BASIS,
14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 # See the License for the specific language governing permissions and
16 # limitations under the License.
19 @file lp_3vnfs_test.py
20 @author Austin Cormier (Austin.Cormier@riftio.com)
22 @brief Launchpad Module Test ExtVNF
36 gi
.require_version('RwIwpYang', '1.0')
37 gi
.require_version('RwNsrYang', '1.0')
38 gi
.require_version('RwVnfdYang', '1.0')
39 gi
.require_version('RwCloudYang', '1.0')
40 gi
.require_version('RwBaseYang', '1.0')
41 gi
.require_version('RwResourceMgrYang', '1.0')
42 gi
.require_version('RwConmanYang', '1.0')
43 gi
.require_version('RwNsmYang', '1.0')
47 from gi
.repository
import RwIwpYang
, NsdYang
, NsrYang
, RwNsrYang
, VldYang
, RwVnfdYang
, RwCloudYang
, RwBaseYang
, RwResourceMgrYang
, RwConmanYang
, RwNsmYang
49 logging
.basicConfig(level
=logging
.DEBUG
)
52 RW_VROUTER_PKG_INSTALL_DIR
= os
.path
.join(
53 os
.environ
["RIFT_INSTALL"],
54 "usr/rift/mano/vnfds/vrouter"
56 RW_TRAFGEN_PKG_INSTALL_DIR
= os
.path
.join(
57 os
.environ
["RIFT_INSTALL"],
58 "usr/rift/mano/vnfds/trafgen"
60 RW_TRAFSINK_PKG_INSTALL_DIR
= os
.path
.join(
61 os
.environ
["RIFT_INSTALL"],
62 "usr/rift/mano/vnfds/trafsink"
64 RW_TG_2VROUTER_TS_NSD_PKG_INSTALL_DIR
= os
.path
.join(
65 os
.environ
["RIFT_INSTALL"],
66 "usr/rift/mano/nsds/tg_2vrouter_ts"
70 class PackageError(Exception):
74 def raise_package_error():
75 raise PackageError("Could not find ns packages")
78 @pytest.fixture(scope
='module')
79 def iwp_proxy(request
, mgmt_session
):
80 return mgmt_session
.proxy(RwIwpYang
)
82 @pytest.fixture(scope
='module')
83 def resource_mgr_proxy(request
, mgmt_session
):
84 return mgmt_session
.proxy(RwResourceMgrYang
)
87 @pytest.fixture(scope
='module')
88 def cloud_proxy(request
, mgmt_session
):
89 return mgmt_session
.proxy(RwCloudYang
)
92 @pytest.fixture(scope
='module')
93 def vnfd_proxy(request
, mgmt_session
):
94 return mgmt_session
.proxy(RwVnfdYang
)
97 @pytest.fixture(scope
='module')
98 def vld_proxy(request
, mgmt_session
):
99 return mgmt_session
.proxy(VldYang
)
102 @pytest.fixture(scope
='module')
103 def nsd_proxy(request
, mgmt_session
):
104 return mgmt_session
.proxy(NsdYang
)
107 @pytest.fixture(scope
='module')
108 def nsr_proxy(request
, mgmt_session
):
109 return mgmt_session
.proxy(NsrYang
)
112 @pytest.fixture(scope
='module')
113 def rwnsr_proxy(request
, mgmt_session
):
114 return mgmt_session
.proxy(RwNsrYang
)
117 @pytest.fixture(scope
='module')
118 def base_proxy(request
, mgmt_session
):
119 return mgmt_session
.proxy(RwBaseYang
)
121 @pytest.fixture(scope
='module')
122 def so_proxy(request
, mgmt_session
):
123 return mgmt_session
.proxy(RwConmanYang
)
125 @pytest.fixture(scope
='module')
126 def nsm_proxy(request
, mgmt_session
):
127 return mgmt_session
.proxy(RwNsmYang
)
129 @pytest.fixture(scope
='session')
130 def vrouter_vnfd_package_file():
131 vrouter_pkg_file
= os
.path
.join(
132 RW_VROUTER_PKG_INSTALL_DIR
,
133 "vrouter_vnfd.tar.gz",
135 if not os
.path
.exists(vrouter_pkg_file
):
136 raise_package_error()
138 return vrouter_pkg_file
140 @pytest.fixture(scope
='session')
141 def tg_vnfd_package_file():
142 tg_pkg_file
= os
.path
.join(
143 RW_TRAFGEN_PKG_INSTALL_DIR
,
144 "trafgen_vnfd.tar.gz",
146 if not os
.path
.exists(tg_pkg_file
):
147 raise_package_error()
151 @pytest.fixture(scope
='session')
152 def ts_vnfd_package_file():
153 ts_pkg_file
= os
.path
.join(
154 RW_TRAFSINK_PKG_INSTALL_DIR
,
155 "trafsink_vnfd.tar.gz",
157 if not os
.path
.exists(ts_pkg_file
):
158 raise_package_error()
162 @pytest.fixture(scope
='session')
163 def tg_2vrouter_ts_nsd_package_file():
164 tg_2vrouter_ts_nsd_pkg_file
= os
.path
.join(
165 RW_TG_2VROUTER_TS_NSD_PKG_INSTALL_DIR
,
166 "tg_2vrouter_ts_nsd.tar.gz",
168 if not os
.path
.exists(tg_2vrouter_ts_nsd_pkg_file
):
169 raise_package_error()
171 return tg_2vrouter_ts_nsd_pkg_file
174 def create_nsr_from_nsd_id(nsd_id
):
175 nsr
= NsrYang
.YangData_Nsr_NsInstanceConfig_Nsr()
176 nsr
.id = str(uuid
.uuid4())
177 nsr
.name
= "TG-2Vrouter-TS EPA"
178 nsr
.short_name
= "TG-2Vrouter-TS EPA"
179 nsr
.description
= "4 VNFs with Trafgen, 2 Vrouters and Trafsink EPA"
181 nsr
.admin_status
= "ENABLED"
186 def upload_descriptor(logger
, descriptor_file
, host
="127.0.0.1"):
187 curl_cmd
= 'curl -F "descriptor=@{file}" http://{host}:4567/api/upload'.format(
188 file=descriptor_file
,
191 logger
.debug("Uploading descriptor %s using cmd: %s", descriptor_file
, curl_cmd
)
192 stdout
= subprocess
.check_output(shlex
.split(curl_cmd
), universal_newlines
=True)
194 json_out
= json
.loads(stdout
)
195 transaction_id
= json_out
["transaction_id"]
197 return transaction_id
200 class DescriptorOnboardError(Exception):
204 def wait_unboard_transaction_finished(logger
, transaction_id
, timeout_secs
=600, host
="127.0.0.1"):
205 logger
.info("Waiting for onboard trans_id %s to complete",
207 start_time
= time
.time()
208 while (time
.time() - start_time
) < timeout_secs
:
210 'http://{host}:4567/api/upload/{t_id}/state'.format(
211 host
=host
, t_id
=transaction_id
215 if state
["status"] == "pending":
219 elif state
["status"] == "success":
220 logger
.info("Descriptor onboard was successful")
224 raise DescriptorOnboardError(state
)
226 if state
["status"] != "success":
227 raise DescriptorOnboardError(state
)
229 @pytest.mark
.incremental
230 class TestLaunchpadStartStop(object):
231 def test_configure_cloud_account(self
, cloud_proxy
, logger
):
232 cloud_account
= RwCloudYang
.CloudAccountConfig()
233 #cloud_account.name = "cloudsim_proxy"
234 #cloud_account.account_type = "cloudsim_proxy"
235 cloud_account
.name
= "riftuser1"
236 cloud_account
.account_type
= "openstack"
237 cloud_account
.openstack
.key
= 'pluto'
238 cloud_account
.openstack
.secret
= 'mypasswd'
239 cloud_account
.openstack
.auth_url
= 'http://10.66.4.xx:5000/v3/'
240 cloud_account
.openstack
.tenant
= 'demo'
241 cloud_account
.openstack
.mgmt_network
= 'private'
243 cloud_proxy
.merge_config("/rw-cloud:cloud-account", cloud_account
)
245 def test_configure_pools(self
, resource_mgr_proxy
):
246 pools
= RwResourceMgrYang
.ResourcePools
.from_dict({
247 "pools": [{ "name": "vm_pool_a",
248 "resource_type": "compute",
249 "pool_type" : "dynamic"},
250 {"name": "network_pool_a",
251 "resource_type": "network",
252 "pool_type" : "dynamic",}]})
254 resource_mgr_proxy
.merge_config('/rw-resource-mgr:resource-mgr-config/rw-resource-mgr:resource-pools', pools
)
256 def test_configure_resource_orchestrator(self
, so_proxy
):
257 cfg
= RwConmanYang
.RoEndpoint
.from_dict({'ro_ip_address': '127.0.0.1',
259 'ro_username' : 'admin',
260 'ro_password' : 'admin'})
261 so_proxy
.merge_config('/rw-conman:cm-config', cfg
)
263 def test_configure_service_orchestrator(self
, nsm_proxy
):
264 cfg
= RwNsmYang
.SoEndpoint
.from_dict({'cm_ip_address': '127.0.0.1',
266 'cm_username' : 'admin',
267 'cm_password' : 'admin'})
268 nsm_proxy
.merge_config('/rw-nsm:ro-config/rw-nsm:cm-endpoint', cfg
)
271 def test_onboard_tg_vnfd(self
, logger
, vnfd_proxy
, tg_vnfd_package_file
):
272 logger
.info("Onboarding trafgen_vnfd package: %s", tg_vnfd_package_file
)
273 trans_id
= upload_descriptor(logger
, tg_vnfd_package_file
)
274 wait_unboard_transaction_finished(logger
, trans_id
)
276 catalog
= vnfd_proxy
.get_config('/vnfd-catalog')
278 assert len(vnfds
) == 1, "There should be one vnfds"
279 assert "trafgen_vnfd" in [vnfds
[0].name
]
281 def test_onboard_vrouter_vnfd(self
, logger
, vnfd_proxy
, vrouter_vnfd_package_file
):
282 logger
.info("Onboarding vrouter_vnfd package: %s", vrouter_vnfd_package_file
)
283 trans_id
= upload_descriptor(logger
, vrouter_vnfd_package_file
)
284 wait_unboard_transaction_finished(logger
, trans_id
)
286 catalog
= vnfd_proxy
.get_config('/vnfd-catalog')
288 assert len(vnfds
) == 2, "There should be two vnfds"
289 assert "vrouter_vnfd" in [vnfds
[0].name
, vnfds
[1].name
]
291 def test_onboard_ts_vnfd(self
, logger
, vnfd_proxy
, ts_vnfd_package_file
):
292 logger
.info("Onboarding trafsink_vnfd package: %s", ts_vnfd_package_file
)
293 trans_id
= upload_descriptor(logger
, ts_vnfd_package_file
)
294 wait_unboard_transaction_finished(logger
, trans_id
)
296 catalog
= vnfd_proxy
.get_config('/vnfd-catalog')
298 assert len(vnfds
) == 3, "There should be three vnfds"
299 assert "trafsink_vnfd" in [vnfds
[0].name
, vnfds
[1].name
, vnfds
[2].name
]
301 def test_onboard_tg_2vrouter_ts_nsd(self
, logger
, nsd_proxy
, tg_2vrouter_ts_nsd_package_file
):
302 logger
.info("Onboarding tg_2vrouter_ts nsd package: %s", tg_2vrouter_ts_nsd_package_file
)
303 trans_id
= upload_descriptor(logger
, tg_2vrouter_ts_nsd_package_file
)
304 wait_unboard_transaction_finished(logger
, trans_id
)
306 catalog
= nsd_proxy
.get_config('/nsd-catalog')
308 assert len(nsds
) == 1, "There should only be a single nsd"
310 assert nsd
.name
== "tg_vrouter_ts_nsd"
311 assert nsd
.short_name
== "tg_2vrouter_ts_nsd"
313 def test_instantiate_tg_2vrouter_ts_nsr(self
, logger
, nsd_proxy
, nsr_proxy
, rwnsr_proxy
, base_proxy
):
314 catalog
= nsd_proxy
.get_config('/nsd-catalog')
317 nsr
= create_nsr_from_nsd_id(nsd
.id)
318 nsr_proxy
.merge_config('/ns-instance-config', nsr
)
320 nsr_opdata
= rwnsr_proxy
.get('/ns-instance-opdata')
321 nsrs
= nsr_opdata
.nsr
322 assert len(nsrs
) == 1
323 assert nsrs
[0].ns_instance_config_ref
== nsr
.id