def testTempView(self): """Make a temp view.""" be1 = CouchStub() be1.expect_POST("/funstuff0/_temp_view").reply(200, dict( total_rows=2, offset=0, rows=[ {"id":"a", "key":"a", "value": "b"}, {"id":"c", "key":"c", "value": "d"} ])) be1.listen("localhost", 23456) be2 = CouchStub() be2.expect_POST("/funstuff1/_temp_view").reply(200, dict( total_rows=2, offset=0, rows=[ {"id":"x", "key":"b", "value": "c"}, {"id":"y", "key":"d", "value": "e"} ])) be2.listen("localhost", 34567) resp = post("http://localhost:22008/funstuff/_temp_view", body={"language":"javascript", "map": "function(doc) { emit(doc.x, doc.y); }"}) 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"], "a") self.assertEqual(resp.body["rows"][1]["key"], "b") self.assertEqual(resp.body["rows"][2]["key"], "c") self.assertEqual(resp.body["rows"][3]["key"], "d")
def testMultikeyGet(self): """Make a temp view.""" be1 = CouchStub() be1_request = be1.expect_POST("/funstuff0/_all_docs?include_docs=true") be1_request.reply(200, dict( total_rows=2, offset=0, rows=[ {"id":"a", "key":"a", "value": {"rev":"2"},"doc":"b"}, {"id":"c", "key":"c", "value": {"rev":"3"},"doc":"d"} ])) be1.listen("localhost", 23456) be2 = CouchStub() be2_request = be2.expect_POST("/funstuff1/_all_docs?include_docs=true") be2_request.reply(200, dict( total_rows=2, offset=0, rows=[ {"id":"b", "key":"b", "value": {"rev":"7"},"doc":"z"}, {"id":"y", "key":"y", "value": {"rev":"9"},"doc":"w"} ])) be2.listen("localhost", 34567) resp = post("http://localhost:22008/funstuff/_all_docs?include_docs=true", body={"keys":["a","c","x","y"]}) be1.verify() be2.verify() be1_post = cjson.decode(be1_request.input_body) be2_post = cjson.decode(be2_request.input_body) def lounge_hash(x): crc = zlib.crc32(x,0) return (crc >> 16)&0x7fff keys1 = be1_post['keys'] keys2 = be2_post['keys'] keys = {0:keys1, 1:keys2} num_shards = 2 for v, k in keys.items(): for key in k: self.assertEqual(lounge_hash(key) % num_shards, int(v)) self.assertEqual(resp.body["total_rows"], 4) self.assertEqual(resp.body["offset"], 0) self.assertEqual(len(resp.body["rows"]), 4) rows = [x["key"] for x in resp.body["rows"]] rows.sort() self.assertEqual(rows, ["a","b","c","y"])
def testBulkDocsWrongContentType(self): """Make a bulk_docs req with the wrong content-type. We should handle this the same way a single node does (treat the body as JSON) """ be1 = CouchStub() be1_request = be1.expect_POST("/funstuff0/_bulk_docs") be1_request.reply(201, [ {"id":"b","rev":"1-23456"}, {"id":"e","error":"conflict","reason":"Document update conflict."} ]) be1.listen("localhost", 23456) be2 = CouchStub() be2_request = be2.expect_POST("/funstuff1/_bulk_docs") be2_request.reply(201, [ {"id":"a","rev":"1-23456"}, {"id":"c","rev":"2-34567"} ]) be2.listen("localhost", 34567) resp = post("http://localhost:22008/funstuff/_bulk_docs", {"docs":[ {"_id":"a","how":"low"}, {"_id":"b","can":"a"}, {"_id":"c","punk":"get"}, {"_id":"e","bannedin":"dc"} ]}, {"Content-Type": "application/x-www-form-urlencoded"}) be1.verify() be2.verify() be1_post = cjson.decode(be1_request.input_body) be2_post = cjson.decode(be2_request.input_body) self.assertEqual(len(resp.body), 4) for row in resp.body: if row['id']=='e': self.assertEqual(row['error'],'conflict') else: assert "rev" in row be1_post = cjson.decode(be1_request.input_body) self.assertEqual(len(be1_post['docs']), 2) self.assertEqual(sorted([row['_id'] for row in be1_post['docs']]), ['b', 'e']) be2_post = cjson.decode(be2_request.input_body) self.assertEqual(len(be2_post['docs']), 2) self.assertEqual(sorted([row['_id'] for row in be2_post['docs']]), ['a', 'c'])
def testBulkDocs(self): be1 = CouchStub() be1_request = be1.expect_POST("/funstuff0/_bulk_docs") be1_request.reply(201, [ {"id":"b","rev":"1-23456"}, {"id":"e","error":"conflict","reason":"Document update conflict."} ]) be1.listen("localhost", 23456) be2 = CouchStub() be2_request = be2.expect_POST("/funstuff1/_bulk_docs") be2_request.reply(201, [ {"id":"a","rev":"1-23456"}, {"id":"c","rev":"2-34567"} ]) be2.listen("localhost", 34567) resp = post("http://localhost:22008/funstuff/_bulk_docs", {"docs":[ {"_id":"a","how":"low"}, {"_id":"b","can":"a"}, {"_id":"c","punk":"get"}, {"_id":"e","bannedin":"dc"} ]}) be1.verify() be2.verify() be1_post = cjson.decode(be1_request.input_body) be2_post = cjson.decode(be2_request.input_body) self.assertEqual(len(resp.body), 4) for row in resp.body: if row['id']=='e': self.assertEqual(row['error'],'conflict') else: assert "rev" in row be1_post = cjson.decode(be1_request.input_body) self.assertEqual(len(be1_post['docs']), 2) self.assertEqual(sorted([row['_id'] for row in be1_post['docs']]), ['b', 'e']) be2_post = cjson.decode(be2_request.input_body) self.assertEqual(len(be2_post['docs']), 2) self.assertEqual(sorted([row['_id'] for row in be2_post['docs']]), ['a', 'c'])