From 3f8befde79b903f6bc88709b0bd58106f066eec4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ante=20Pu=C5=A1i=C4=87?= <ante.pusic@fer.hr>
Date: Wed, 23 Aug 2023 12:48:17 +0200
Subject: [PATCH] Bump PyYAML version (#1174)

---
 tests/gql_behave/requirements.txt         |  2 +-
 tests/gql_behave/steps/test_parameters.py | 18 +++++++++---------
 tests/integration/ldap/requirements.txt   |  2 +-
 tests/setup.sh                            |  2 +-
 4 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/tests/gql_behave/requirements.txt b/tests/gql_behave/requirements.txt
index 290c4cab2..7d7d3a3ab 100644
--- a/tests/gql_behave/requirements.txt
+++ b/tests/gql_behave/requirements.txt
@@ -2,5 +2,5 @@ behave==1.2.6
 neo4j-driver==4.1.1
 parse==1.18.0
 parse-type==0.5.2
-PyYAML==5.4.1
+PyYAML==6.0.1
 six==1.15.0
diff --git a/tests/gql_behave/steps/test_parameters.py b/tests/gql_behave/steps/test_parameters.py
index ee92d9ff3..1a8d74cc7 100644
--- a/tests/gql_behave/steps/test_parameters.py
+++ b/tests/gql_behave/steps/test_parameters.py
@@ -40,15 +40,15 @@ class TestParameters:
         par = dict()
         for row in table:
             par[row[0]] = self.parse_parameters(row[1])
-            if isinstance(par[row[0]], str) and par[row[0]].startswith("'") \
-                    and par[row[0]].endswith("'"):
-                par[row[0]] = par[row[0]][1:len(par[row[0]]) - 1]
+            if isinstance(par[row[0]], str) and par[row[0]].startswith("'") and par[row[0]].endswith("'"):
+                par[row[0]] = par[row[0]][1 : len(par[row[0]]) - 1]
         par[table.headings[0]] = self.parse_parameters(table.headings[1])
-        if isinstance(par[table.headings[0]], str) and \
-                par[table.headings[0]].startswith("'") and \
-                par[table.headings[0]].endswith("'"):
-            par[table.headings[0]] = \
-                par[table.headings[0]][1:len(par[table.headings[0]]) - 1]
+        if (
+            isinstance(par[table.headings[0]], str)
+            and par[table.headings[0]].startswith("'")
+            and par[table.headings[0]].endswith("'")
+        ):
+            par[table.headings[0]] = par[table.headings[0]][1 : len(par[table.headings[0]]) - 1]
 
         self.parameters = par
 
@@ -77,4 +77,4 @@ class TestParameters:
             Format readable for a database.
         """
 
-        return yaml.load(val)
+        return yaml.load(val, Loader=yaml.FullLoader)
diff --git a/tests/integration/ldap/requirements.txt b/tests/integration/ldap/requirements.txt
index 849f48f74..263196d84 100644
--- a/tests/integration/ldap/requirements.txt
+++ b/tests/integration/ldap/requirements.txt
@@ -1,2 +1,2 @@
 ldap3==2.6
-pyyaml==5.4.1
+pyyaml==6.0.1
diff --git a/tests/setup.sh b/tests/setup.sh
index bc9d614c2..a70020778 100755
--- a/tests/setup.sh
+++ b/tests/setup.sh
@@ -14,7 +14,7 @@ PIP_DEPS=(
    "parse==1.18.0"
    "parse-type==0.5.2"
    "pytest==7.3.2"
-   "pyyaml==5.4.1"
+   "pyyaml==6.0.1"
    "six==1.15.0"
    "networkx==2.4"
    "gqlalchemy==1.3.3"