gh-975: Fix gafferpy alpha 1

gchq · Mar 1, 2022 · 2e8afc1 · 2e8afc1
1 parent 2ec511c
commit 2e8afc1
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 1 deletion.
diff --git a/python-shell/src/gafferpy/gaffer_operations.py b/python-shell/src/gafferpy/gaffer_operations.py
@@ -683,7 +683,7 @@ class GetTraits(Operation):
     CLASS = 'uk.gov.gchq.gaffer.store.operation.GetTraits'
 
     def __init__(self,
-                 current_traits,
+                 current_traits=True,
                  options=None):
         super().__init__(
             _class_name=self.CLASS, options=options)
@@ -696,6 +696,26 @@ def to_json(self):
         operation['currentTraits'] = self.current_traits
         return operation
 
+class HasTrait(Operation):
+    CLASS = 'uk.gov.gchq.gaffer.store.operation.HasTrait'
+
+    def __init__(self,
+                 trait,
+                 current_traits=True,
+                 options=None):
+        super().__init__(
+            _class_name=self.CLASS, options=options)
+
+        self.trait = trait
+        self.current_traits = current_traits
+
+    def to_json(self):
+        operation = super().to_json()
+
+        operation['trait'] = self.trait
+        operation['currentTraits'] = self.current_traits
+        return operation
+
 
 class AddElements(Operation):
     """

diff --git a/python-shell/src/test/test_connector.py b/python-shell/src/test/test_connector.py
@@ -202,6 +202,7 @@ def test_execute_get(self):
             "uk.gov.gchq.gaffer.operation.impl.function.Filter",
             "uk.gov.gchq.gaffer.operation.impl.function.Transform",
             "uk.gov.gchq.gaffer.operation.impl.function.Aggregate",
+            "uk.gov.gchq.gaffer.store.operation.HasTrait",
             "uk.gov.gchq.gaffer.store.operation.GetTraits",
             "uk.gov.gchq.gaffer.mapstore.operation.CountAllElementsDefaultView",
             "uk.gov.gchq.gaffer.operation.export.graph.ExportToOtherAuthorisedGraph",

diff --git a/python-shell/src/test/test_connector_requests.py b/python-shell/src/test/test_connector_requests.py
@@ -202,6 +202,7 @@ def test_execute_get(self):
             "uk.gov.gchq.gaffer.operation.impl.function.Filter",
             "uk.gov.gchq.gaffer.operation.impl.function.Transform",
             "uk.gov.gchq.gaffer.operation.impl.function.Aggregate",
+            "uk.gov.gchq.gaffer.store.operation.HasTrait",
             "uk.gov.gchq.gaffer.store.operation.GetTraits",
             "uk.gov.gchq.gaffer.mapstore.operation.CountAllElementsDefaultView",
             "uk.gov.gchq.gaffer.operation.export.graph.ExportToOtherAuthorisedGraph",

diff --git a/python-shell/src/test/test_gaffer_operations.py b/python-shell/src/test/test_gaffer_operations.py
@@ -4996,6 +4996,19 @@ class GafferOperationsTest(unittest.TestCase):
                 operation=g.GetElements()
             )
         ],
+        [
+            '''
+            {
+                "class" : "uk.gov.gchq.gaffer.store.operation.HasTrait",
+                "trait": "VISIBILITY",
+                "currentTraits" : true
+            }
+            ''',
+            g.HasTrait(
+                trait="VISIBILITY",
+                current_traits=True
+            )
+        ],
         [
             '''
             {