X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;ds=sidebyside;f=run_tests.sh;h=1f985b6c18acfc46bc859d593343440fd2ca3a79;hb=caf81bafc6132d79674f4c31bbd22d97f4aee961;hp=972f6a27f24511590e604a92419d4a8366b4a462;hpb=f42719d1347b60bf7d99226221f3a5b51a3b0a25;p=yardstick.git

diff --git a/run_tests.sh b/run_tests.sh
index 972f6a27f..1f985b6c1 100755
--- a/run_tests.sh
+++ b/run_tests.sh
@@ -33,7 +33,33 @@ run_flake8() {
     fi
 }
 
+get_external_libs() {
+    cd $(dirname ${BASH_SOURCE[0]})
+    TREX_DOWNLOAD="https://trex-tgn.cisco.com/trex/release/v2.05.tar.gz"
+    TREX_DIR=$PWD/trex/scripts
+    if [ ! -d "$TREX_DIR" ]; then
+        rm -rf ${TREX_DOWNLOAD##*/}
+        if [ ! -e ${TREX_DOWNLOAD##*/} ] ; then
+            wget -nv $TREX_DOWNLOAD
+        fi
+        tar xf ${TREX_DOWNLOAD##*/}
+        pushd .
+        rm -rf trex && mkdir -p trex
+        mv v2.05 trex/scripts
+        rm -rf v2.05.tar.gz
+        touch "$PWD/trex/scripts/automation/trex_control_plane/stl/__init__.py"
+        popd
+    fi
+    echo "Done."
+    export PYTHONPATH=$PYTHONPATH:"$PWD/trex/scripts/automation/trex_control_plane"
+    export PYTHONPATH=$PYTHONPATH:"$PWD/trex/scripts/automation/trex_control_plane/stl"
+    echo $PYTHONPATH
+}
+
 run_tests() {
+    echo "Get external libs needed for unit test"
+    get_external_libs
+
     echo "Running unittest ... "
     if [ $FILE_OPTION == "f" ]; then
         python -m unittest discover -v -s tests/unit > $logfile 2>&1