client/common_lib/cros/graphite/elasticsearch_mock_unittest.py

#!/usr/bin/python

import unittest

import elasticsearch_mock as elasticsearch

class statsd_mock_test(unittest.TestCase):
    """Test statsd_mock"""
    def setUp(self):
        self.es = elasticsearch.Elasticsearch(host='host',
                                              port=1,
                                              timeout=10)


    def test_index_call_mock(self):
        """Test mock Elasticsearch.index method"""
        self.es.index(index='blah', doc_type='blah blah', body='random')


    def test_index_exists_mock(self):
        """Test mock Elasticsearch.indices.exists method"""
        self.es.indices.exists(index='random index')


    def test_index_delete_mock(self):
        """Test mock Elasticsearch.indices.delete method"""
        self.es.indices.delete(index='random index')


    def test_search_mock(self):
        """Test mock Elasticsearch.search method"""
        self.es.search(index='index', body='query')


    def test_exception_mock(self):
        """Test mock elasticsearch.ElasticsearchException method"""
        try:
            raise elasticsearch.ElasticsearchException('error message')
        except elasticsearch.ElasticsearchException as e:
            pass


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