import gzip
import bson
from bson import json_util
from bson import Binary
import pymongo
from pymongo import MongoClient
import time
import timeit

from ahl.testing.mongo_server import MongoTestServer

server = MongoTestServer()
server.start()
client = server.api

coll = client.some_db.create_collection('some_coll', options={'capped': True})
coll.insert([{}, {}])

# while True:
cursor = coll.find()
assert cursor.next()
while cursor.alive:
    print "cursor.alive"
    server.kill()  # MongoDB will die at this point
    try:
        do_something_with_data(cursor.next())
    except StopIteration:
        time.sleep(1)
    except Exception:
        # Catch any client exceptions...
        pass

server.kill()
while True:
    cursor = coll.find()
    assert cursor.alive