From d3cb3504615a31f713abc9cd6c90aeaebaa1d718 Mon Sep 17 00:00:00 2001 From: Marc Bonnici Date: Wed, 20 Jun 2018 14:51:40 +0100 Subject: [PATCH] commands/templates: Update to add optional `setup_rerun` method --- wa/commands/templates/apk_workload | 6 ++++++ wa/commands/templates/apkrevent_workload | 6 ++++++ wa/commands/templates/apkuiauto_workload | 6 ++++++ 3 files changed, 18 insertions(+) diff --git a/wa/commands/templates/apk_workload b/wa/commands/templates/apk_workload index d9e90416..f3544fae 100644 --- a/wa/commands/templates/apk_workload +++ b/wa/commands/templates/apk_workload @@ -40,6 +40,12 @@ class ${class_name}(ApkWorkload): super(${class_name}, self).setup(context) # Perform any necessary setup before starting the application + def setup_rerun(self, context): + super(${class_name}, self).setup(context) + # If the workloads has the `requires_rerun` attribute set to `True` this + # method may be used to perform any necessary setup for the rerun of the + # application. + def extract_results(self, context): super(${class_name}, self).extract_results(context) # Extract results on the target diff --git a/wa/commands/templates/apkrevent_workload b/wa/commands/templates/apkrevent_workload index 888ab85c..e92f1aab 100644 --- a/wa/commands/templates/apkrevent_workload +++ b/wa/commands/templates/apkrevent_workload @@ -40,6 +40,12 @@ class ${class_name}(ApkReventWorkload): super(${class_name}, self).setup(context) # Perform any necessary setup before starting the UI automation + def setup_rerun(self, context): + super(${class_name}, self).setup(context) + # If the workloads has the `requires_rerun` attribute set to `True` this + # method may be used to perform any necessary setup for the rerun of the + # application. + def extract_results(self, context): super(${class_name}, self).extract_results(context) # Extract results on the target diff --git a/wa/commands/templates/apkuiauto_workload b/wa/commands/templates/apkuiauto_workload index 7a890f25..e5fde1b3 100644 --- a/wa/commands/templates/apkuiauto_workload +++ b/wa/commands/templates/apkuiauto_workload @@ -40,6 +40,12 @@ class ${class_name}(ApkUiautoWorkload): super(${class_name}, self).setup(context) # Perform any necessary setup before starting the UI automation + def setup_rerun(self, context): + super(${class_name}, self).setup(context) + # If the workloads has the `requires_rerun` attribute set to `True` this + # method may be used to perform any necessary setup for the rerun of the + # application. + def extract_results(self, context): super(${class_name}, self).extract_results(context) # Extract results on the target