Add error handling for ocloud route
[pti/o2.git] / tests / locust / ocloud.py
1 # Copyright (C) 2021 Wind River Systems, Inc.
2 #
3 #  Licensed under the Apache License, Version 2.0 (the "License");
4 #  you may not use this file except in compliance with the License.
5 #  You may obtain a copy of the License at
6 #
7 #      http://www.apache.org/licenses/LICENSE-2.0
8 #
9 #  Unless required by applicable law or agreed to in writing, software
10 #  distributed under the License is distributed on an "AS IS" BASIS,
11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 #  See the License for the specific language governing permissions and
13 #  limitations under the License.
14
15 from locust import HttpUser, task, constant
16
17
18 class QuickstartUser(HttpUser):
19     wait_time = constant(0)
20
21     @task
22     def ocloud(self):
23         self.client.get("/o2ims_infrastructureInventory/v1/")
24
25     @task
26     def resource(self):
27         resp = self.client.get(
28             "/o2ims_infrastructureInventory/v1/resourcePools")
29         json_resp_dict = resp.json()
30         self.client.get(
31             "/o2ims_infrastructureInventory/v1/resourcePools/%s/resources" %
32             json_resp_dict[0]['resourcePoolId'])
33
34     @task
35     def dms(self):
36         self.client.get("/o2ims_infrastructureInventory/v1/deploymentManagers")
37
38     def on_start(self):
39         pass