George Keishing | e7e9171 | 2021-09-03 11:28:44 -0500 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 2 | |
| 3 | r""" |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 4 | This module provides utilities for code updates. |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 5 | """ |
| 6 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 7 | import collections |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 8 | import os |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 9 | import re |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 10 | import sys |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 11 | import tarfile |
| 12 | import time |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 13 | |
| 14 | import bmc_ssh_utils as bsu |
| 15 | import gen_print as gp |
| 16 | import gen_robot_keyword as keyword |
| 17 | import variables as var |
| 18 | from robot.libraries.BuiltIn import BuiltIn |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 19 | |
| 20 | robot_pgm_dir_path = os.path.dirname(__file__) + os.sep |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 21 | repo_data_path = re.sub("/lib", "/data", robot_pgm_dir_path) |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 22 | sys.path.append(repo_data_path) |
| 23 | |
Charles Paul Hofer | c1fa2bc | 2017-08-18 16:44:03 -0500 | [diff] [blame] | 24 | |
Sushil Singh | 8469a48 | 2020-07-30 04:19:10 -0500 | [diff] [blame] | 25 | def get_bmc_firmware(image_type, sw_dict): |
| 26 | r""" |
| 27 | Get the dictionary of image based on image type like either BMC or Host. |
| 28 | |
| 29 | Description of argument(s): |
| 30 | image_type This value is either BMC update or Host update type. |
George Keishing | 16b3c7b | 2021-01-28 09:23:37 -0600 | [diff] [blame] | 31 | sw_dict This contain dictionary of firmware inventory properties. |
Sushil Singh | 8469a48 | 2020-07-30 04:19:10 -0500 | [diff] [blame] | 32 | """ |
| 33 | |
| 34 | temp_dict = collections.OrderedDict() |
| 35 | for key, value in sw_dict.items(): |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 36 | if value["image_type"] == image_type: |
Sushil Singh | 8469a48 | 2020-07-30 04:19:10 -0500 | [diff] [blame] | 37 | temp_dict[key] = value |
| 38 | else: |
| 39 | pass |
| 40 | return temp_dict |
| 41 | |
| 42 | |
Gunnar Mills | 096cd56 | 2018-03-26 10:19:12 -0500 | [diff] [blame] | 43 | def verify_no_duplicate_image_priorities(image_purpose): |
Charles Paul Hofer | c1fa2bc | 2017-08-18 16:44:03 -0500 | [diff] [blame] | 44 | r""" |
| 45 | Check that there are no active images with the same purpose and priority. |
| 46 | |
| 47 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 48 | image_purpose The purpose that images must have to be |
| 49 | checked for priority duplicates. |
Charles Paul Hofer | c1fa2bc | 2017-08-18 16:44:03 -0500 | [diff] [blame] | 50 | """ |
| 51 | |
| 52 | taken_priorities = {} |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 53 | _, image_names = keyword.run_key( |
| 54 | "Get Software Objects " + "version_type=" + image_purpose |
| 55 | ) |
Charles Paul Hofer | c1fa2bc | 2017-08-18 16:44:03 -0500 | [diff] [blame] | 56 | |
| 57 | for image_name in image_names: |
| 58 | _, image = keyword.run_key("Get Host Software Property " + image_name) |
| 59 | if image["Activation"] != var.ACTIVE: |
| 60 | continue |
| 61 | image_priority = image["Priority"] |
| 62 | if image_priority in taken_priorities: |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 63 | BuiltIn().fail( |
| 64 | "Found active images with the same priority.\n" |
| 65 | + gp.sprint_vars(image, taken_priorities[image_priority]) |
| 66 | ) |
Charles Paul Hofer | a567316 | 2017-08-30 09:49:16 -0500 | [diff] [blame] | 67 | taken_priorities[image_priority] = image |
Charles Paul Hofer | c1fa2bc | 2017-08-18 16:44:03 -0500 | [diff] [blame] | 68 | |
Charles Paul Hofer | c1fa2bc | 2017-08-18 16:44:03 -0500 | [diff] [blame] | 69 | |
Charles Paul Hofer | da24d0a | 2017-08-09 15:03:40 -0500 | [diff] [blame] | 70 | def get_non_running_bmc_software_object(): |
Charles Paul Hofer | da24d0a | 2017-08-09 15:03:40 -0500 | [diff] [blame] | 71 | r""" |
| 72 | Get the URI to a BMC image from software that is not running on the BMC. |
| 73 | """ |
| 74 | |
| 75 | # Get the version of the image currently running on the BMC. |
| 76 | _, cur_img_version = keyword.run_key("Get BMC Version") |
| 77 | # Remove the surrounding double quotes from the version. |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 78 | cur_img_version = cur_img_version.replace('"', "") |
Charles Paul Hofer | da24d0a | 2017-08-09 15:03:40 -0500 | [diff] [blame] | 79 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 80 | _, images = keyword.run_key( |
| 81 | "Read Properties " + var.SOFTWARE_VERSION_URI + "enumerate" |
| 82 | ) |
Charles Paul Hofer | da24d0a | 2017-08-09 15:03:40 -0500 | [diff] [blame] | 83 | |
| 84 | for image_name in images: |
| 85 | _, image_properties = keyword.run_key( |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 86 | "Get Host Software Property " + image_name |
| 87 | ) |
| 88 | if ( |
| 89 | "Purpose" in image_properties |
| 90 | and "Version" in image_properties |
| 91 | and image_properties["Purpose"] != var.VERSION_PURPOSE_HOST |
| 92 | and image_properties["Version"] != cur_img_version |
| 93 | ): |
Charles Paul Hofer | da24d0a | 2017-08-09 15:03:40 -0500 | [diff] [blame] | 94 | return image_name |
| 95 | BuiltIn().fail("Did not find any non-running BMC images.") |
| 96 | |
Charles Paul Hofer | da24d0a | 2017-08-09 15:03:40 -0500 | [diff] [blame] | 97 | |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 98 | def delete_all_pnor_images(): |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 99 | r""" |
| 100 | Delete all PNOR images from the BMC. |
| 101 | """ |
| 102 | |
Adriana Kobylak | 7eedb1d | 2019-08-12 15:24:23 -0500 | [diff] [blame] | 103 | keyword.run_key("Initiate Host PowerOff") |
| 104 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 105 | status, images = keyword.run_key( |
| 106 | "Get Software Objects " + var.VERSION_PURPOSE_HOST |
| 107 | ) |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 108 | for image_name in images: |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 109 | keyword.run_key( |
| 110 | "Delete Image And Verify " |
| 111 | + image_name |
| 112 | + " " |
| 113 | + var.VERSION_PURPOSE_HOST |
| 114 | ) |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 115 | |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 116 | |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 117 | def wait_for_activation_state_change(version_id, initial_state): |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 118 | r""" |
| 119 | Wait for the current activation state of ${version_id} to |
| 120 | change from the state provided by the calling function. |
| 121 | |
| 122 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 123 | version_id The version ID whose state change we are |
| 124 | waiting for. |
| 125 | initial_state The activation state we want to wait for. |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 126 | """ |
| 127 | |
| 128 | keyword.run_key_u("Open Connection And Log In") |
| 129 | retry = 0 |
Charles Paul Hofer | 290b8bd | 2017-09-26 10:02:22 -0500 | [diff] [blame] | 130 | num_read_errors = 0 |
| 131 | read_fail_threshold = 1 |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 132 | while retry < 60: |
| 133 | status, software_state = keyword.run_key( |
| 134 | "Read Properties " + var.SOFTWARE_VERSION_URI + str(version_id), |
| 135 | ignore=1, |
| 136 | ) |
| 137 | if status == "FAIL": |
Charles Paul Hofer | 290b8bd | 2017-09-26 10:02:22 -0500 | [diff] [blame] | 138 | num_read_errors += 1 |
| 139 | if num_read_errors > read_fail_threshold: |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 140 | message = "Read errors exceeds threshold:\n " + gp.sprint_vars( |
| 141 | num_read_errors, read_fail_threshold |
| 142 | ) |
Charles Paul Hofer | 290b8bd | 2017-09-26 10:02:22 -0500 | [diff] [blame] | 143 | BuiltIn().fail(message) |
Charles Paul Hofer | 4d26c00 | 2017-09-27 08:39:29 -0500 | [diff] [blame] | 144 | time.sleep(10) |
Charles Paul Hofer | 290b8bd | 2017-09-26 10:02:22 -0500 | [diff] [blame] | 145 | continue |
| 146 | |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 147 | current_state = (software_state)["Activation"] |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 148 | if initial_state == current_state: |
Charles Paul Hofer | 4d26c00 | 2017-09-27 08:39:29 -0500 | [diff] [blame] | 149 | time.sleep(10) |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 150 | retry += 1 |
Charles Paul Hofer | 290b8bd | 2017-09-26 10:02:22 -0500 | [diff] [blame] | 151 | num_read_errors = 0 |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 152 | else: |
| 153 | return |
| 154 | return |
| 155 | |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 156 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 157 | def get_latest_file(dir_path): |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 158 | r""" |
| 159 | Get the path to the latest uploaded file. |
| 160 | |
| 161 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 162 | dir_path Path to the dir from which the name of the |
| 163 | last updated file or folder will be |
| 164 | returned to the calling function. |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 165 | """ |
| 166 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 167 | stdout, stderr, rc = bsu.bmc_execute_command( |
| 168 | "cd " |
| 169 | + dir_path |
| 170 | + "; stat -c '%Y %n' * |" |
| 171 | + " sort -k1,1nr | head -n 1" |
| 172 | ) |
Joy Onyerikwu | 9b66897 | 2018-05-22 19:10:43 -0500 | [diff] [blame] | 173 | return stdout.split(" ")[-1] |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 174 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 175 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 176 | def get_version_tar(tar_file_path): |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 177 | r""" |
| 178 | Read the image version from the MANIFEST inside the tarball. |
| 179 | |
| 180 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 181 | tar_file_path The path to a tar file that holds the image |
| 182 | version inside the MANIFEST. |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 183 | """ |
| 184 | |
Sushil Singh | 3c77d78 | 2019-10-14 05:29:42 -0500 | [diff] [blame] | 185 | version = "" |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 186 | tar = tarfile.open(tar_file_path) |
| 187 | for member in tar.getmembers(): |
George Keishing | 42ade54 | 2020-07-22 09:51:46 -0500 | [diff] [blame] | 188 | BuiltIn().log_to_console(member.name) |
| 189 | if member.name != "MANIFEST": |
| 190 | continue |
Gunnar Mills | 096cd56 | 2018-03-26 10:19:12 -0500 | [diff] [blame] | 191 | f = tar.extractfile(member) |
| 192 | content = f.read() |
George Keishing | 36efbc0 | 2018-12-12 10:18:23 -0600 | [diff] [blame] | 193 | if content.find(b"version=") == -1: |
| 194 | # This tar member does not contain the version. |
| 195 | continue |
George Keishing | 42ade54 | 2020-07-22 09:51:46 -0500 | [diff] [blame] | 196 | content = content.decode("utf-8", "ignore").split("\n") |
George Keishing | 36efbc0 | 2018-12-12 10:18:23 -0600 | [diff] [blame] | 197 | content = [x for x in content if "version=" in x] |
| 198 | version = content[0].split("=")[-1] |
| 199 | break |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 200 | tar.close() |
| 201 | return version |
| 202 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 203 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 204 | def get_image_version(file_path): |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 205 | r""" |
| 206 | Read the file for a version object. |
| 207 | |
| 208 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 209 | file_path The path to a file that holds the image |
| 210 | version. |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 211 | """ |
| 212 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 213 | stdout, stderr, rc = bsu.bmc_execute_command( |
| 214 | "cat " + file_path + ' | grep "version="', ignore_err=1 |
| 215 | ) |
Joy Onyerikwu | 9b66897 | 2018-05-22 19:10:43 -0500 | [diff] [blame] | 216 | return (stdout.split("\n")[0]).split("=")[-1] |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 217 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 218 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 219 | def get_image_purpose(file_path): |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 220 | r""" |
| 221 | Read the file for a purpose object. |
| 222 | |
| 223 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 224 | file_path The path to a file that holds the image |
| 225 | purpose. |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 226 | """ |
| 227 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 228 | stdout, stderr, rc = bsu.bmc_execute_command( |
| 229 | "cat " + file_path + ' | grep "purpose="', ignore_err=1 |
| 230 | ) |
Joy Onyerikwu | 9b66897 | 2018-05-22 19:10:43 -0500 | [diff] [blame] | 231 | return stdout.split("=")[-1] |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 232 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 233 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 234 | def get_image_path(image_version): |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 235 | r""" |
| 236 | Query the upload image dir for the presence of image matching |
| 237 | the version that was read from the MANIFEST before uploading |
| 238 | the image. Based on the purpose verify the activation object |
| 239 | exists and is either READY or INVALID. |
| 240 | |
| 241 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 242 | image_version The version of the image that should match |
| 243 | one of the images in the upload dir. |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 244 | """ |
| 245 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 246 | stdout, stderr, rc = bsu.bmc_execute_command( |
| 247 | "ls -d " + var.IMAGE_UPLOAD_DIR_PATH + "*/" |
| 248 | ) |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 249 | |
Joy Onyerikwu | 9b66897 | 2018-05-22 19:10:43 -0500 | [diff] [blame] | 250 | image_list = stdout.split("\n") |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 251 | retry = 0 |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 252 | while retry < 10: |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 253 | for i in range(0, len(image_list)): |
| 254 | version = get_image_version(image_list[i] + "MANIFEST") |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 255 | if version == image_version: |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 256 | return image_list[i] |
| 257 | time.sleep(10) |
| 258 | retry += 1 |
| 259 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 260 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 261 | def verify_image_upload(image_version, timeout=3): |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 262 | r""" |
| 263 | Verify the image was uploaded correctly and that it created |
Charles Paul Hofer | 6b97268 | 2017-07-20 11:36:56 -0500 | [diff] [blame] | 264 | a valid d-bus object. If the first check for the image |
| 265 | fails, try again until we reach the timeout. |
| 266 | |
| 267 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 268 | image_version The version from the image's manifest file |
| 269 | (e.g. "v2.2-253-g00050f1"). |
| 270 | timeout How long, in minutes, to keep trying to |
| 271 | find the image on the BMC. Default is 3 minutes. |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 272 | """ |
| 273 | |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 274 | image_path = get_image_path(image_version) |
| 275 | image_version_id = image_path.split("/")[-2] |
| 276 | |
Charles Paul Hofer | de7d408 | 2017-08-08 14:41:01 -0500 | [diff] [blame] | 277 | keyword.run_key_u("Open Connection And Log In") |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 278 | image_purpose = get_image_purpose(image_path + "MANIFEST") |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 279 | if ( |
| 280 | image_purpose == var.VERSION_PURPOSE_BMC |
| 281 | or image_purpose == var.VERSION_PURPOSE_HOST |
| 282 | ): |
George Keishing | ff1e3ec | 2017-07-20 01:58:21 -0500 | [diff] [blame] | 283 | uri = var.SOFTWARE_VERSION_URI + image_version_id |
Charles Paul Hofer | 6b97268 | 2017-07-20 11:36:56 -0500 | [diff] [blame] | 284 | ret_values = "" |
| 285 | for itr in range(timeout * 2): |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 286 | status, ret_values = keyword.run_key( |
| 287 | "Read Attribute " + uri + " Activation" |
| 288 | ) |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 289 | |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 290 | if ( |
| 291 | (ret_values == var.READY) |
| 292 | or (ret_values == var.INVALID) |
| 293 | or (ret_values == var.ACTIVE) |
| 294 | ): |
Charles Paul Hofer | cef6199 | 2017-08-18 10:14:18 -0500 | [diff] [blame] | 295 | return True, image_version_id |
Charles Paul Hofer | 6b97268 | 2017-07-20 11:36:56 -0500 | [diff] [blame] | 296 | else: |
| 297 | time.sleep(30) |
| 298 | |
| 299 | # If we exit the for loop, the timeout has been reached |
| 300 | gp.print_var(ret_values) |
Charles Paul Hofer | cef6199 | 2017-08-18 10:14:18 -0500 | [diff] [blame] | 301 | return False, None |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 302 | else: |
Saqib Khan | bb8b63f | 2017-05-24 10:58:01 -0500 | [diff] [blame] | 303 | gp.print_var(image_purpose) |
Charles Paul Hofer | cef6199 | 2017-08-18 10:14:18 -0500 | [diff] [blame] | 304 | return False, None |
Saqib Khan | fb1f6ae | 2017-04-26 13:24:41 -0500 | [diff] [blame] | 305 | |
Charles P. Hofer | 1d20acd | 2017-07-05 15:24:40 -0500 | [diff] [blame] | 306 | |
Charles Paul Hofer | 6b97268 | 2017-07-20 11:36:56 -0500 | [diff] [blame] | 307 | def verify_image_not_in_bmc_uploads_dir(image_version, timeout=3): |
Charles P. Hofer | 1d20acd | 2017-07-05 15:24:40 -0500 | [diff] [blame] | 308 | r""" |
| 309 | Check that an image with the given version is not unpacked inside of the |
| 310 | BMCs image uploads directory. If no image is found, retry every 30 seconds |
Charles Paul Hofer | 6b97268 | 2017-07-20 11:36:56 -0500 | [diff] [blame] | 311 | until the given timeout is hit, in case the BMC takes time |
| 312 | unpacking the image. |
Charles P. Hofer | 1d20acd | 2017-07-05 15:24:40 -0500 | [diff] [blame] | 313 | |
| 314 | Description of argument(s): |
Joy Onyerikwu | 004ad3c | 2018-06-11 16:29:56 -0500 | [diff] [blame] | 315 | image_version The version of the image to look for on |
| 316 | the BMC. |
| 317 | timeout How long, in minutes, to try to find an |
| 318 | image on the BMC. Default is 3 minutes. |
Charles P. Hofer | 1d20acd | 2017-07-05 15:24:40 -0500 | [diff] [blame] | 319 | """ |
| 320 | |
Charles Paul Hofer | 6b97268 | 2017-07-20 11:36:56 -0500 | [diff] [blame] | 321 | for i in range(timeout * 2): |
Patrick Williams | 5731818 | 2022-12-08 06:18:26 -0600 | [diff] [blame^] | 322 | stdout, stderr, rc = bsu.bmc_execute_command( |
| 323 | "ls " |
| 324 | + var.IMAGE_UPLOAD_DIR_PATH |
| 325 | + "*/MANIFEST 2>/dev/null " |
| 326 | + '| xargs grep -rl "version=' |
| 327 | + image_version |
| 328 | + '"' |
| 329 | ) |
| 330 | image_dir = os.path.dirname(stdout.split("\n")[0]) |
| 331 | if "" != image_dir: |
| 332 | bsu.bmc_execute_command("rm -rf " + image_dir) |
| 333 | BuiltIn().fail("Found invalid BMC Image: " + image_dir) |
Charles P. Hofer | 1d20acd | 2017-07-05 15:24:40 -0500 | [diff] [blame] | 334 | time.sleep(30) |