From 7c76adbd79271a35bafbd07336fc20b5fdbfaa5d Mon Sep 17 00:00:00 2001
From: eeff <33566201+eeff@users.noreply.github.com>
Date: Fri, 11 Feb 2022 15:01:26 +0800
Subject: [PATCH] Add dataset selection option for pressure testing

---
 ft/common.resource | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/ft/common.resource b/ft/common.resource
index b2ce9014..9da4bf3f 100644
--- a/ft/common.resource
+++ b/ft/common.resource
@@ -14,7 +14,6 @@ Resource    simulator.resource
 
 *** Variables ***
 ${test_data_dir}     ft/data
-${test_pdata_dir}    ${test_data_dir}/persistence
 
 ${NODE_UNKNOWN}	0
 ${NODE_DRIVER}	1
@@ -158,6 +157,10 @@ Skip If Not MQTT API
     Skip If    $NEURON_API != "mqtt"    Current API is not `mqtt`, just skip
 
 Copy Persistence Test Data
+    ${data_set} =                  Get Variable Value   ${DATASET}                       total-10k
+    ${data_dir} =                  Set Variable         ${test_data_dir}/persistence/${data_set}
+    Directory Should Exist         ${data_dir}
+
     Directory Should Exist         build
-    ${rc} =                        Run And Return Rc    cp -r ${test_pdata_dir} build/
+    ${rc} =                        Run And Return Rc    cp -r ${data_dir} build/
     Should Be Equal As Integers    ${rc}                0