def main(collection_key): v = CouchDBCollectionFilter(couchdb_obj=get_couchdb(), collection_key=collection_key) solr_db = Solr(URL_SOLR) results = [] for r in v: dt_start = dt_end = datetime.datetime.now() try: doc = fill_in_title(r.doc) has_required_fields(r.doc) except KeyError, e: print(e.message) continue solr_doc = map_couch_to_solr_doc(r.doc) results.append(solr_doc) solr_doc = push_doc_to_solr(solr_doc, solr_db=solr_db) dt_end = datetime.datetime.now()
def main(collection_key): v = CouchDBCollectionFilter( couchdb_obj=get_couchdb(), collection_key=collection_key) solr_db = Solr(URL_SOLR) results = [] for r in v: dt_start = dt_end = datetime.datetime.now() try: doc = fill_in_title(r.doc) has_required_fields(r.doc) except KeyError, e: print(e.message) continue solr_doc = map_couch_to_solr_doc(r.doc) results.append(solr_doc) solr_doc = push_doc_to_solr(solr_doc, solr_db=solr_db) dt_end = datetime.datetime.now()
def test_push_doc_to_solr(self, mock_solr): '''Unit test calls to solr''' doc = json.load(open(DIR_FIXTURES + '/couchdb_doc.json')) sdoc = map_couch_to_solr_doc(doc) push_doc_to_solr(sdoc, mock_solr)
def test_push_doc_to_solr(self, mock_solr): """Unit test calls to solr""" doc = json.load(open(DIR_FIXTURES + "/couchdb_doc.json")) sdoc = map_couch_to_solr_doc(doc) push_doc_to_solr(sdoc, mock_solr)