def test_update_local_db_translates_opt_fields_correctly(self): i_dict = { 'doc_num': '2', 'attachment_num': '3', 'pacer_doc_id': '789', 'pacer_de_seq_num': '20', 'pacer_dm_id': '12', 'upload_date': '2007-12-25', 'free_import': 1, 'sha1': 'hash' } self.doc_xml.remove_document('1', '2') self.doc_xml.add_document('2', '3', i_dict) DocumentManager.update_local_db(self.doc_xml) self.assertEquals(1, Document.objects.count()) created_doc = Document.objects.all()[0] self.assertEquals(int(i_dict['doc_num']), created_doc.docnum) self.assertEquals(int(i_dict['attachment_num']), created_doc.subdocnum) self.assertEquals(i_dict['pacer_doc_id'], created_doc.docid) self.assertEquals(int(i_dict['pacer_de_seq_num']), created_doc.de_seq_num) self.assertEquals(int(i_dict['pacer_dm_id']), created_doc.dm_id) self.assertEquals(i_dict['sha1'], created_doc.sha1) expected_upload_date = datetime.strptime(i_dict['upload_date'], "%Y-%m-%d") self.assertEquals(expected_upload_date, created_doc.lastdate) self.assertEquals(int(i_dict['free_import']), created_doc.free_import)
def test_update_local_db_basic(self): DocumentManager.update_local_db(self.doc_xml) created_doc = Document.objects.all()[0] self.assertEquals(1, Document.objects.count()) self.assertEquals('nysd', created_doc.court) self.assertEquals(1234, created_doc.casenum) self.assertEquals(1, created_doc.docnum) self.assertEquals(2, created_doc.subdocnum)
def test_update_local_db_ignore_available(self): doc_meta = self.doc_xml.get_document_metadict('1', '2') doc_meta['available'] = '1' DocumentManager.update_local_db(self.doc_xml) created_doc = Document.objects.all()[0] self.assertEquals(0, created_doc.available) DocumentManager.update_local_db(self.doc_xml, ignore_available=0) created_doc = Document.objects.all()[0] self.assertEquals(1, created_doc.available)
def test_update_local_db_updates_existing(self): d1 = Document(court='nysd', casenum='1234', docnum='1', subdocnum='2') d1.save() self.assertEquals(1, Document.objects.count()) doc_meta = self.doc_xml.get_document_metadict('1', '2') doc_meta['pacer_doc_id'] = '12' DocumentManager.update_local_db(self.doc_xml) created_doc = Document.objects.all()[0] self.assertEquals(1, Document.objects.count()) self.assertEquals('12', created_doc.docid)
def test_update_local_db_doesnt_overwrite_local(self): d1 = Document(court='nysd', casenum='1234', docnum='1', subdocnum='2', docid='120') d1.save() self.assertEquals(1, Document.objects.count()) # This document doesn't have docid, but we shouldn't overwrite DocumentManager.update_local_db(self.doc_xml) created_doc = Document.objects.all()[0] self.assertEquals(1, Document.objects.count()) self.assertEquals('120', created_doc.docid)
def test_update_local_db_translates_opt_fields_correctly(self): i_dict = {'doc_num': '2', 'attachment_num': '3', 'pacer_doc_id': '789', 'pacer_de_seq_num': '20', 'pacer_dm_id': '12', 'upload_date': '2007-12-25', 'free_import': 1, 'sha1': 'hash'} self.doc_xml.remove_document('1', '2') self.doc_xml.add_document('2', '3', i_dict) DocumentManager.update_local_db(self.doc_xml) self.assertEquals(1, Document.objects.count()) created_doc = Document.objects.all()[0] self.assertEquals(int(i_dict['doc_num']), created_doc.docnum) self.assertEquals(int(i_dict['attachment_num']), created_doc.subdocnum) self.assertEquals(i_dict['pacer_doc_id'], created_doc.docid) self.assertEquals(int(i_dict['pacer_de_seq_num']), created_doc.de_seq_num) self.assertEquals(int(i_dict['pacer_dm_id']), created_doc.dm_id) self.assertEquals(i_dict['sha1'], created_doc.sha1) expected_upload_date = datetime.strptime(i_dict['upload_date'], "%Y-%m-%d") self.assertEquals(expected_upload_date, created_doc.lastdate) self.assertEquals(int(i_dict['free_import']), created_doc.free_import)