Nick Lang
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
from collections import defaultdict
|
|
|
|
try:
|
|
# python 2.6
|
|
from unittest2 import TestCase, SkipTest
|
|
except ImportError:
|
|
from unittest import TestCase, SkipTest
|
|
|
|
from elasticsearch import Elasticsearch
|
|
|
|
|
|
class DummyTransport(object):
|
|
def __init__(self, hosts, responses=None, **kwargs):
|
|
self.hosts = hosts
|
|
self.responses = responses
|
|
self.call_count = 0
|
|
self.calls = defaultdict(list)
|
|
|
|
def perform_request(self, method, url, params=None, body=None):
|
|
resp = 200, {}
|
|
if self.responses:
|
|
resp = self.responses[self.call_count]
|
|
self.call_count += 1
|
|
self.calls[(method, url)].append((params, body))
|
|
return resp
|
|
|
|
|
|
class ElasticsearchTestCase(TestCase):
|
|
def setUp(self):
|
|
super(ElasticsearchTestCase, self).setUp()
|
|
self.client = Elasticsearch(transport_class=DummyTransport)
|
|
|
|
def assert_call_count_equals(self, count):
|
|
self.assertEquals(count, self.client.transport.call_count)
|
|
|
|
def assert_url_called(self, method, url, count=1):
|
|
self.assertIn((method, url), self.client.transport.calls)
|
|
calls = self.client.transport.calls[(method, url)]
|
|
self.assertEquals(count, len(calls))
|
|
return calls
|
|
|
|
|
|
class TestElasticsearchTestCase(ElasticsearchTestCase):
|
|
def test_our_transport_used(self):
|
|
self.assertIsInstance(self.client.transport, DummyTransport)
|
|
|
|
def test_start_with_0_call(self):
|
|
self.assert_call_count_equals(0)
|
|
|
|
def test_each_call_is_recorded(self):
|
|
self.client.transport.perform_request("GET", "/")
|
|
self.client.transport.perform_request("DELETE", "/42", params={}, body="body")
|
|
self.assert_call_count_equals(2)
|
|
self.assertEquals([({}, "body")], self.assert_url_called("DELETE", "/42", 1))
|