1
|
import unittest
|
2
|
from eu.dnetlib.es_connector import DLIESConnector
|
3
|
import time
|
4
|
|
5
|
DLI_INDEX = 'dli'
|
6
|
ES_HOST = 'ip-90-147-167-27.ct1.garrservices.it'
|
7
|
|
8
|
|
9
|
class TestDLIESConnector(unittest.TestCase):
|
10
|
def setUp(self):
|
11
|
self.connector = DLIESConnector(ES_HOST, DLI_INDEX)
|
12
|
|
13
|
def testQuery(self):
|
14
|
start_time = time.time()
|
15
|
result = self.connector.simple_query('protein')
|
16
|
print("---Request time: %s milliseconds ---" % ((time.time() - start_time) * 1000))
|
17
|
self.assertTrue(result.total > 0)
|
18
|
|
19
|
def tearDown(self):
|
20
|
self.connector = None
|
21
|
|
22
|
if __name__ == '__main__':
|
23
|
unittest.main()
|