From 13025ea3641252b7cf38c633e77f1c4d8376fcad Mon Sep 17 00:00:00 2001
From: Jarrod Johnson <jjohnson2@lenovo.com>
Date: Thu, 28 Mar 2019 09:43:27 -0400
Subject: [PATCH] Add token refreshes to adapter update

It was observed that an adapter update could be inflicted
with an expiring token.  Aggressively refresh to mitigate
this.

Change-Id: I7624e8c9abbfdc4e8fd3b323f868d3606cd0039e
---
 pyghmi/ipmi/oem/lenovo/imm.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pyghmi/ipmi/oem/lenovo/imm.py b/pyghmi/ipmi/oem/lenovo/imm.py
index beedf6cb..7e893ffb 100644
--- a/pyghmi/ipmi/oem/lenovo/imm.py
+++ b/pyghmi/ipmi/oem/lenovo/imm.py
@@ -1578,6 +1578,7 @@ class XCCClient(IMMClient):
             validselectors = set([])
             for loc in locations:
                 validselectors.add(loc.replace('#', '-'))
+            self._refresh_token()
             rsp = self.wc.grab_json_response(
                 '/api/function/adapter_update?params=pci_GetAdapterListAndFW')
             foundselectors = []
@@ -1590,6 +1591,7 @@ class XCCClient(IMMClient):
                         break
             else:
                 raise Exception('Could not find matching adapter for update')
+            self._refresh_token()
             rsp = self.wc.grab_json_response('/api/function', json.dumps(
                 {'pci_SetOOBFWSlots': '|'.join(foundselectors)}))
             if rsp.get('return', -1) != 0:
@@ -1598,6 +1600,7 @@ class XCCClient(IMMClient):
                     'Unexpected result from PCI select: ' + errmsg)
             self.set_property('/v2/ibmc/uefi/force-inventory', 1)
         else:
+            self._refresh_token()
             rsp = self.wc.grab_json_response(
                 '/api/dataset/imm_firmware_update')
             if rsp['items'][0]['upgrades'][0]['id'] != 1: