import unittest
from eu.dnetlib.es_connector import DLIESConnector
import time

DLI_INDEX = 'dli'
ES_HOST = 'ip-90-147-167-27.ct1.garrservices.it'


class TestDLIESConnector(unittest.TestCase):
    def setUp(self):
        self.connector = DLIESConnector(ES_HOST, DLI_INDEX)

    def testQuery(self):
        start_time = time.time()
        result = self.connector.simple_query('protein')
        print("---Request time: %s milliseconds ---" % ((time.time() - start_time) * 1000))
        self.assertTrue(result.total > 0)

    def tearDown(self):
        self.connector = None

if __name__ == '__main__':
    unittest.main()