Fix writing data in a Boolean field using the Fluent API

Author: aboudreault

CHANGELOG.rst

@@ -9,6 +9,7 @@ Features
 Bug Fixes
 ---------
 * Unable to connect to a cloud cluster using Ubuntu 20.04 (PYTHON-1238)
+* [GRAPH] Can't write data in a Boolean field using the Fluent API (PYTHON-1239)
 
 Others
 ------

cassandra/datastax/graph/graphson.py

@@ -52,7 +52,7 @@
 DSE Graph      GraphSON 2.0     GraphSON 3.0   |  Python Driver
 ------------ | -------------- | -------------- | ------------
 text         | string         | string         | str
-boolean      | g:Boolean      | g:Boolean      | bool
+boolean      |                |                | bool
 bigint       | g:Int64        | g:Int64        | long
 int          | g:Int32        | g:Int32        | int
 double       | g:Double       | g:Double       | float
@@ -125,7 +125,7 @@ class TextTypeIO(GraphSONTypeIO):
 
 
 class BooleanTypeIO(GraphSONTypeIO):
-    graphson_base_type = 'Boolean'
+    graphson_base_type = None
     cql_type = 'boolean'
 
     @classmethod

tests/integration/advanced/graph/__init__.py

@@ -418,6 +418,8 @@ class ClassicGraphFixtures(GraphFixtures):
     @staticmethod
     def datatypes():
         data = {
+            "boolean1": ["Boolean()", True, None],
+            "boolean2": ["Boolean()", False, None],
             "point1": ["Point()", Point(.5, .13), GraphSON1Deserializer.deserialize_point],
              "point2": ["Point()", Point(-5, .0), GraphSON1Deserializer.deserialize_point],
 

tests/integration/advanced/graph/test_graph_datatype.py

@@ -28,6 +28,7 @@
 from cassandra.graph.query import GraphProtocol
 from cassandra.graph.types import VertexProperty
 
+from tests.util import wait_until
 from tests.integration.advanced.graph import BasicGraphUnitTestCase, ClassicGraphFixtures, \
     ClassicGraphSchema, CoreGraphSchema
 from tests.integration.advanced.graph import VertexLabel, GraphTestConfiguration, GraphUnitTestCase
@@ -94,22 +95,26 @@ def _test_all_datatypes(self, schema, graphson):
             schema.create_vertex_label(self.session, vertex_label, execution_profile=ep)
             vertex = list(schema.add_vertex(self.session, vertex_label, property_name, value, execution_profile=ep))[0]
 
-            vertex_properties = list(schema.get_vertex_properties(
-                self.session, vertex, execution_profile=ep))
+            def get_vertex_properties():
+                return list(schema.get_vertex_properties(
+                    self.session, vertex, execution_profile=ep))
 
+            prop_returned = 1 if DSE_VERSION < Version('5.1') else 2  # include pkid >=5.1
+            wait_until(
+                lambda: len(get_vertex_properties()) == prop_returned, 0.2, 15)
+
+            vertex_properties = get_vertex_properties()
             if graphson == GraphProtocol.GRAPHSON_1_0:
                 vertex_properties = [vp.as_vertex_property() for vp in vertex_properties]
 
-            prop_returned = 1 if DSE_VERSION < Version('5.1') else 2  # include pkid >=5.1
-            self.assertEqual(len(vertex_properties), prop_returned)
             for vp in vertex_properties:
                 if vp.label == 'pkid':
                     continue
 
                 self.assertIsInstance(vp, VertexProperty)
                 self.assertEqual(vp.label, property_name)
                 if graphson == GraphProtocol.GRAPHSON_1_0:
-                    deserialized_value = deserializer(vp.value)
+                    deserialized_value = deserializer(vp.value) if deserializer else vp.value
                     self.assertEqual(deserialized_value, value)
                 else:
                     self.assertEqual(vp.value, value)
@@ -171,10 +176,15 @@ def __test_udt(self, schema, graphson, address_class, address_with_tags_class,
             schema.create_vertex_label(self.session, vertex_label, execution_profile=ep)
 
             vertex = list(schema.add_vertex(self.session, vertex_label, property_name, value, execution_profile=ep))[0]
-            vertex_properties = list(schema.get_vertex_properties(
-                self.session, vertex, execution_profile=ep))
 
-            self.assertEqual(len(vertex_properties), 2)  # include pkid
+            def get_vertex_properties():
+                return list(schema.get_vertex_properties(
+                    self.session, vertex, execution_profile=ep))
+
+            wait_until(
+                lambda: len(get_vertex_properties()) == 2, 0.2, 15)
+
+            vertex_properties = get_vertex_properties()
             for vp in vertex_properties:
                 if vp.label == 'pkid':
                     continue