def setUp(self): super(ImpimAPIAsyncHTTPTestCase, self).setUp() self._http_client = AsyncHTTPClient(self) self._elastic_search_for_test = ElasticSearchForTest( self._app.config, self._http_client) self._elastic_search_for_test.cleanup() file_system_for_test.cleanup()
def setUp(self): super(ElasticSearchTestCase, self).setUp() config = MockConfig() self._http_client = AsyncHTTPClient(self.io_loop) self._test_http_client = TestAsyncHTTPClient(self) self._elastic_search_urls = Urls(config) self._elastic_search = ElasticSearch(config=config, http_client=self._http_client) self._elastic_search_for_test = ElasticSearchForTest(config=config, http_client=self._test_http_client) self._elastic_search_for_test.cleanup()
class ImpimAPIAsyncHTTPTestCase(AsyncHTTPTestCase): def setUp(self): super(ImpimAPIAsyncHTTPTestCase, self).setUp() self._http_client = AsyncHTTPClient(self) self._elastic_search_for_test = ElasticSearchForTest(self._app.config, self._http_client) self._elastic_search_for_test.cleanup() file_system_for_test.cleanup() def get_app(self): return ImagesApplication(conf_file=abspath(join(dirname(__file__), '..', 'impim_api.test.conf')), base_url='') def get_new_ioloop(self): return IOLoop.instance()
def setUp(self): super(ImpimAPIAsyncHTTPTestCase, self).setUp() self._http_client = AsyncHTTPClient(self) self._elastic_search_for_test = ElasticSearchForTest(self._app.config, self._http_client) self._elastic_search_for_test.cleanup() file_system_for_test.cleanup()
class ImpimAPIAsyncHTTPTestCase(AsyncHTTPTestCase): def setUp(self): super(ImpimAPIAsyncHTTPTestCase, self).setUp() self._http_client = AsyncHTTPClient(self) self._elastic_search_for_test = ElasticSearchForTest( self._app.config, self._http_client) self._elastic_search_for_test.cleanup() file_system_for_test.cleanup() def get_app(self): return ImagesApplication(conf_file=abspath( join(dirname(__file__), '..', 'impim_api.test.conf')), base_url='') def get_new_ioloop(self): return IOLoop.instance()
class ElasticSearchTestCase(AsyncTestCase): def setUp(self): super(ElasticSearchTestCase, self).setUp() config = MockConfig() self._http_client = AsyncHTTPClient(self.io_loop) self._test_http_client = TestAsyncHTTPClient(self) self._elastic_search_urls = Urls(config) self._elastic_search = ElasticSearch(config=config, http_client=self._http_client) self._elastic_search_for_test = ElasticSearchForTest(config=config, http_client=self._test_http_client) self._elastic_search_for_test.cleanup() def test_search(self): self._elastic_search.store_meta_data(self._noop_callback, 'id1', title=u'Title') self.wait() self._elastic_search_for_test.refresh() self._elastic_search.search(self.assert_search_callback, page=1, page_size=10) self.wait() def assert_search_callback(self, response): assert response['total'] == 1 assert len(response['items']) == 1 assert response['items'][0]['title'] == u'Title' self.stop() def test_search_query(self): self._elastic_search.store_meta_data(self._noop_callback, 'id1', title=u'One') self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id2', title=u'Two') self.wait() self._elastic_search_for_test.refresh() self._elastic_search.search(self.assert_search_query_callback, q='One', page=1, page_size=10) self.wait() def assert_search_query_callback(self, response): assert response['total'] == 1 assert response['items'][0]['title'] == u'One' self.stop() def test_search_created_date_filter(self): self._elastic_search.store_meta_data(self._noop_callback, 'id1', title=u'First', created_date=datetime(2012, 10, 4, 13, 0, 3)) self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id2', title=u'Second', created_date=datetime(2012, 10, 4, 13, 0, 2)) self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id3', title=u'Third', created_date=datetime(2012, 10, 4, 13, 0, 1)) self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id4', title=u'Fourth', created_date=datetime(2012, 10, 4, 13, 0, 0)) self.wait() self._elastic_search_for_test.refresh() self._elastic_search.search( self.assert_search_created_date_filter_callback, created_date_from=datetime(2012, 10, 4, 13, 0, 1), created_date_to=datetime(2012, 10, 4, 13, 0, 2), page=1, page_size=10 ) self.wait() self._elastic_search.search( self.assert_search_created_date_from_filter_callback, created_date_from=datetime(2012, 10, 4, 13, 0, 1), page=1, page_size=10 ) self.wait() self._elastic_search.search( self.assert_search_created_date_to_filter_callback, created_date_to=datetime(2012, 10, 4, 13, 0, 2), page=1, page_size=10 ) self.wait() def assert_search_created_date_filter_callback(self, response): assert response['total'] == 2 assert response['items'][0]['title'] == u'Second' assert response['items'][1]['title'] == u'Third' self.stop() def assert_search_created_date_from_filter_callback(self, response): assert response['total'] == 3 assert response['items'][0]['title'] == u'First' assert response['items'][1]['title'] == u'Second' assert response['items'][2]['title'] == u'Third' self.stop() def assert_search_created_date_to_filter_callback(self, response): assert response['total'] == 3 assert response['items'][0]['title'] == u'Second' assert response['items'][1]['title'] == u'Third' assert response['items'][2]['title'] == u'Fourth' self.stop() def test_search_event_date_filter(self): self._elastic_search.store_meta_data(self._noop_callback, 'id1', title=u'First', event_date=datetime(2012, 10, 4, 13, 0, 3), created_date=datetime(2012, 10, 4, 13, 0, 3)) self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id2', title=u'Second', event_date=datetime(2012, 10, 4, 13, 0, 2), created_date=datetime(2012, 10, 4, 13, 0, 2)) self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id3', title=u'Third', event_date=datetime(2012, 10, 4, 13, 0, 1), created_date=datetime(2012, 10, 4, 13, 0, 1)) self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id4', title=u'Fourth', event_date=datetime(2012, 10, 4, 13, 0, 0), created_date=datetime(2012, 10, 4, 13, 0, 0)) self.wait() self._elastic_search_for_test.refresh() self._elastic_search.search( self.assert_search_event_date_filter_callback, event_date_from=datetime(2012, 10, 4, 13, 0, 1), event_date_to=datetime(2012, 10, 4, 13, 0, 2), page=1, page_size=10 ) self.wait() self._elastic_search.search( self.assert_search_event_date_from_filter_callback, event_date_from=datetime(2012, 10, 4, 13, 0, 1), page=1, page_size=10 ) self.wait() self._elastic_search.search( self.assert_search_event_date_to_filter_callback, event_date_to=datetime(2012, 10, 4, 13, 0, 2), page=1, page_size=10 ) self.wait() def assert_search_event_date_filter_callback(self, response): assert response['total'] == 2 assert response['items'][0]['title'] == u'Second' assert response['items'][1]['title'] == u'Third' self.stop() def assert_search_event_date_from_filter_callback(self, response): assert response['total'] == 3 assert response['items'][0]['title'] == u'First' assert response['items'][1]['title'] == u'Second' assert response['items'][2]['title'] == u'Third' self.stop() def assert_search_event_date_to_filter_callback(self, response): assert response['total'] == 3 assert response['items'][0]['title'] == u'Second' assert response['items'][1]['title'] == u'Third' assert response['items'][2]['title'] == u'Fourth' self.stop() def test_search_order_by_relevance_with_query(self): self._elastic_search.store_meta_data(self._noop_callback, 'id1', title=u'Exact title') self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id2', title=u'Title') self.wait() self._elastic_search_for_test.refresh() self._elastic_search.search(self.assert_search_order_by_relevance_with_query, q='Exact title', page=1, page_size=10) self.wait() def assert_search_order_by_relevance_with_query(self, response): assert response['total'] == 2 assert response['items'][0]['title'] == u'Exact title' assert response['items'][1]['title'] == u'Title' self.stop() def test_search_order_by_newest_first_with_no_query(self): self._elastic_search.store_meta_data(self._noop_callback, 'id1', title=u'Exact title', created_date=datetime(2012, 10, 4, 13, 0, 0)) self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id2', title=u'Title', created_date=datetime(2012, 10, 5, 13, 0, 0)) self.wait() self._elastic_search_for_test.refresh() self._elastic_search.search(self.assert_search_order_by_newest_first_with_no_query, page=1, page_size=10) self.wait() def assert_search_order_by_newest_first_with_no_query(self, response): assert response['total'] == 2 assert response['items'][0]['title'] == u'Title' assert response['items'][1]['title'] == u'Exact title' self.stop() def test_search_pagination(self): self._elastic_search.store_meta_data(self._noop_callback, 'id1', title=u'Title') self.wait() self._elastic_search.store_meta_data(self._noop_callback, 'id2', title=u'Title') self.wait() self._elastic_search_for_test.refresh() self._elastic_search.search(self.assert_search_pagination_callback_page_1, page=1, page_size=1) self.wait() self._elastic_search.search(self.assert_search_pagination_callback_page_2, page=2, page_size=1) self.wait() def assert_search_pagination_callback_page_1(self, response): assert response['total'] == 2 assert len(response['items']) == 1 self.stop() def assert_search_pagination_callback_page_2(self, response): assert response['total'] == 2 assert len(response['items']) == 1 self.stop() def test_fetch_meta_data(self): image_id = 'id' self._elastic_search.store_meta_data(self._noop_callback, image_id) self.wait() self._elastic_search.fetch_meta_data(self.assert_fetch_meta_data, image_id) self.wait() def assert_fetch_meta_data(self, response): assert response['id'] == u'id' self.stop() def _noop_callback(self): self.stop()