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")