Add indexes to test SchemaGraph (kensho-technologies#315)

* Add indexes to test SchemaGraph

* Changed sets arguments to lists and remove is None check

Author: pmantica1

graphql_compiler/schema_generation/orientdb/schema_graph_builder.py

@@ -111,10 +111,8 @@ class is an edge class, and
     link_schema_elements(elements, inheritance_structure)
     for element in six.itervalues(elements):
         element.freeze()
-    if index_data is None:
-        all_indexes = set()
-    else:
-        all_indexes = _get_indexes(index_data, elements)
+
+    all_indexes = _get_indexes(index_data, elements)
     return SchemaGraph(elements, inheritance_structure, all_indexes)
 
 

graphql_compiler/schema_generation/orientdb/utils.py

@@ -6,3 +6,10 @@
     'WHERE name NOT IN [\'ORole\', \'ORestricted\', \'OTriggered\', '
     '\'ORIDs\', \'OUser\', \'OIdentity\', \'OSchedule\', \'OFunction\']'
 )
+
+ORIENTDB_INDEX_RECORDS_QUERY = (
+    'SELECT name, type, indexDefinition, metadata FROM ('
+    'SELECT expand(indexes) FROM metadata:indexmanager'
+    ') WHERE type IN '
+    '[\'UNIQUE\', \'NOTUNIQUE\', \'UNIQUE_HASH_INDEX\', \'NOTUNIQUE_HASH_INDEX\']'
+)

graphql_compiler/tests/test_helpers.py

@@ -10,7 +10,9 @@
 from .. import get_graphql_schema_from_orientdb_schema_data
 from ..debugging_utils import pretty_print_gremlin, pretty_print_match
 from ..schema_generation.orientdb.schema_graph_builder import get_orientdb_schema_graph
-from ..schema_generation.orientdb.utils import ORIENTDB_SCHEMA_RECORDS_QUERY
+from ..schema_generation.orientdb.utils import (
+    ORIENTDB_INDEX_RECORDS_QUERY, ORIENTDB_SCHEMA_RECORDS_QUERY
+)
 
 
 # The strings which we will be comparing have newlines and spaces we'd like to get rid of,
@@ -274,7 +276,9 @@ def generate_schema_graph(graph_client):
     """Generate SchemaGraph from a pyorient client"""
     schema_records = graph_client.command(ORIENTDB_SCHEMA_RECORDS_QUERY)
     schema_data = [x.oRecordData for x in schema_records]
-    return get_orientdb_schema_graph(schema_data, set())
+    index_records = graph_client.command(ORIENTDB_INDEX_RECORDS_QUERY)
+    index_query_data = [x.oRecordData for x in index_records]
+    return get_orientdb_schema_graph(schema_data, index_query_data)
 
 
 def generate_schema(graph_client, class_to_field_type_overrides=None, hidden_classes=None):