def testChanges(self):
"""Query _changes on a db.
smartproxy should send a _changes req to each shard and merge them.
"""
be1 = CouchStub()
be1.expect_GET("/funstuff0/_changes?since=5").reply(200,
textwrap.dedent(
'''\
{"results":[
{"seq": 6, "id": "mywallet", "changes":[{"rev": "1-2345"}]},
{"seq": 7, "id": "elsegundo", "changes":[{"rev": "2-3456"}]}
],
"last_seq"=7}'''),
headers={"Content-Type": "text/plain;charset=utf8"},
raw_body=True)
be1.listen("localhost", 23456)
be2 = CouchStub()
be2.expect_GET("/funstuff1/_changes?since=12").reply(200,
textwrap.dedent('''\
{"results":[
{"seq": 13, "id": "gottagetit", "changes":[{"rev": "1-2345"}]},
{"seq": 14, "id": "gotgottogetit", "changes":[{"rev": "2-3456"}]}
],
"last_seq"=14}'''),
headers={"Content-Type": "text/plain;charset=utf8"},
raw_body=True)
be2.listen("localhost", 34567)
resp = get("http://localhost:22008/funstuff/_changes?since=%s" % encode_seq([5,12]))
be1.verify()
be2.verify()
self.assertEqual(resp.code, 200)
self.assertEqual(resp.headers["content-type"], "text/plain;charset=utf8")
assert 'results' in resp.body
res = resp.body['results']
self.assertEqual(len(res), 4, "Should have 4 changes")
# check that the sequence vectors increment correctly
# the order the rows arrive is non-deterministic
seq = [5,12]
for row in res:
if row["id"] in ["mywallet", "elsegundo"]:
seq[0] += 1
elif row["id"] in ["gottagetit", "gotgottogetit"]:
seq[1] += 1
else:
assert False, "Got unexpected row %s" % row["id"]
self.assertEqual(encode_seq(seq), row["seq"])
self.assertEqual(encode_seq([7,14]), resp.body['last_seq'])
def testGetMissingDB(self):
"""Try to GET information on a missing database."""
be1 = CouchStub()
be1.expect_GET("/test0").reply(404, dict(error="not_found",reason="missing"))
be1.listen("localhost", 23456)
be2 = CouchStub()
be2.expect_GET("/test1").reply(404, dict(error="not_found",reason="missing"))
be2.listen("localhost", 34567)
resp = get("http://localhost:22008/test")
be1.verify()
be2.verify()
self.assertEqual(resp.code, 404)
self.assertEqual(resp.body['error'], 'not_found')
self.assertEqual(resp.body['reason'], 'missing')
def testGetDB(self):
"""Try to GET information on a database.
smartproxy should get info on each shard and merge them together
"""
be1 = CouchStub()
be1.expect_GET("/test0").reply(200, dict(
db_name="test0",
doc_count=5,
doc_del_count=1,
update_seq=8,
purge_seq=0,
compact_running=False,
disk_size=16384,
instance_start_time="1250979728236424"))
be1.listen("localhost", 23456)
be2 = CouchStub()
be2.expect_GET("/test1").reply(200, dict(
db_name="test1",
doc_count=10,
doc_del_count=2,
update_seq=16,
purge_seq=0,
compact_running=False,
disk_size=16384,
instance_start_time="1250979728236424"))
be2.listen("localhost", 34567)
resp = get("http://localhost:22008/test")
be1.verify()
be2.verify()
self.assertEqual(resp.code, 200)
self.assertEqual(resp.body['db_name'], 'test')
self.assertEqual(resp.body['doc_count'], 15)
self.assertEqual(resp.body['doc_del_count'], 3)
self.assertEqual(resp.body['disk_size'], 32768)
self.assertEqual(resp.body['update_seq'], '[8, 16]')
def testReduce(self):
"""Try to re-reduce"""
be1 = CouchStub()
be1.expect_GET("/funstuff0/_design/fun").reply(200, dict(
views=dict(
stuff=dict(
map="function(doc){}",
reduce="function(k,v,r) { return sum(v); }"
)
)))
be1.expect_GET("/funstuff0/_design/fun/_view/stuff").reply(200, dict(
total_rows=3,
offset=0,
rows=[
{"id":"a", "key":"a", "value": 1},
{"id":"c", "key":"b", "value": 2},
{"id":"c", "key":"c", "value": 3}
]))
be1.listen("localhost", 23456)
be2 = CouchStub()
be2.expect_GET("/funstuff1/_design/fun/_view/stuff").reply(200, dict(
total_rows=3,
offset=0,
rows=[
{"id":"x", "key":"b", "value": 10},
{"id":"y", "key":"c", "value": 11},
{"id":"z", "key":"e", "value": 12},
]))
be2.listen("localhost", 34567)
resp = get("http://localhost:22008/funstuff/_design/fun/_view/stuff")
be1.verify()
be2.verify()
self.assertEqual(resp.body["total_rows"], 6)
self.assertEqual(resp.body["offset"], 0)
self.assertEqual(len(resp.body["rows"]), 4)
self.assertEqual(resp.body["rows"][0]["key"], "a")
self.assertEqual(resp.body["rows"][1]["key"], "b")
self.assertEqual(resp.body["rows"][2]["key"], "c")
self.assertEqual(resp.body["rows"][3]["key"], "e")
self.assertEqual(resp.body["rows"][0]["value"], 1)
self.assertEqual(resp.body["rows"][1]["value"], 12)
self.assertEqual(resp.body["rows"][2]["value"], 14)
self.assertEqual(resp.body["rows"][3]["value"], 12)
def testReverse(self):
"""Query a view with descending=true"""
be1 = CouchStub()
be1.expect_GET("/funstuff0/_design/fun").reply(200, dict(
views=dict(
stuff=dict(
map="function(doc){}"
)
)))
be1.expect_GET("/funstuff0/_design/fun/_view/stuff?descending=true").reply(200, dict(
total_rows=2,
offset=0,
rows=[
{"id":"a", "key":"2", "value": "a"},
{"id":"c", "key":"1", "value": "b"}
]))
be1.listen("localhost", 23456)
be2 = CouchStub()
be2.expect_GET("/funstuff1/_design/fun/_view/stuff?descending=true").reply(200, dict(
total_rows=2,
offset=0,
rows=[
{"id":"x", "key":"3", "value": "c"},
{"id":"y", "key":"0", "value": "e"}
]))
be2.listen("localhost", 34567)
resp = get("http://localhost:22008/funstuff/_design/fun/_view/stuff?descending=true")
be1.verify()
be2.verify()
self.assertEqual(resp.body["total_rows"], 4)
self.assertEqual(resp.body["offset"], 0)
self.assertEqual(len(resp.body["rows"]), 4)
self.assertEqual(resp.body["rows"][0]["key"], "3")
self.assertEqual(resp.body["rows"][1]["key"], "2")
self.assertEqual(resp.body["rows"][2]["key"], "1")
self.assertEqual(resp.body["rows"][3]["key"], "0")
