Merge pull request #889 from dsurnin/hot-fix-upgrage-cmd

Hot fix upgrage cmd

Author: ashcherbakov

Diff for: indy_client/client/wallet/upgrade.py

@@ -31,15 +31,12 @@ def _op(self):
             ACTION: self.action,
             SHA256: self.sha256,
             FORCE: self.force,
-            PACKAGE: self.package
+            PACKAGE: self.package,
+            SCHEDULE: self.schedule,
+            TIMEOUT: self.timeout,
+            JUSTIFICATION: self.justification,
+            REINSTALL: self.reinstall,
         }
-        if self.action == START:
-            op.update({
-                SCHEDULE: self.schedule,
-                TIMEOUT: self.timeout,
-                JUSTIFICATION: self.justification,
-                REINSTALL: self.reinstall,
-            })
 
         return op
 

Diff for: indy_client/test/cli/test_pool_upgrade.py

@@ -43,8 +43,8 @@ def poolUpgradeCancelled(poolUpgradeScheduled, be, do, trusteeCli,
     cancelUpgrade[ACTION] = CANCEL
     cancelUpgrade[JUSTIFICATION] = '"not gonna give you one"'
     be(trusteeCli)
-    do('send POOL_UPGRADE name={name} version={version} sha256={sha256} '
-       'action={action} justification={justification}',
+    do('send POOL_UPGRADE name={name} version={version} sha256={sha256} action={action} '
+       'justification={justification} schedule={schedule} timeout={timeout} ',
        within=10,
        expect=['Sending pool upgrade', 'Pool Upgrade Transaction Scheduled'],
        mapper=cancelUpgrade)