"""
Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
"""

import logging

from graph_notebook.seed.load_query import get_queries

from test.integration import IntegrationTest

logger = logging.getLogger('DataDrivenOpenCypherTest')


class DataDrivenOpenCypherTest(IntegrationTest):
    def setUp(self):
        super().setUp()
        # check if the data is already loaded
        query = '''MATCH (norwichcity:Team)-[:CURRENT_LEAGUE]->(epl:League) RETURN norwichcity'''
        res = self.client.opencypher_http(query)
        res.raise_for_status()
        js = res.json()
        if len(js['results']) > 0:
            return
        airport_queries = get_queries('opencypher', 'epl')
        for q in airport_queries:
            try:  # we are deciding to try except because we do not know if the database we are connecting to has a partially complete set of airports data or not.
                self.client.opencypher_http(q['content'])
            except Exception as e:
                logger.error(f'query {q["content"]} failed due to {e}')
                continue