Esempio n. 1
0
 def test_insert_data(self):
     with session_scope(self.dal) as session:
         insert_data(session, self.data)
     result = []
     with session_scope(self.dal) as session:
         notices = session.query(Notice).all()
         for n in notices:
             notice = object_as_dict(n)
             #pop the date and createdAt attributes since they're constructed programmatically
             notice.pop('date')
             notice.pop('createdAt')
             #pop this as it'll vary
             notice.pop('notice_type_id')
             result.append(notice)
     expected = [{
         'id': 1,
         'solicitation_number': 'test',
         'agency': 'agency',
         'notice_data': {
             'url': 'url',
             'naics': 'test',
             'office': 'office',
             'subject': 'test',
             'classcod': 'test',
             'setaside': 'test',
             'emails': ['*****@*****.**']
         },
         'compliant': 0,
         'feedback': None,
         'history': None,
         'action': None,
         'updatedAt': None,
         'na_flag': False
     }]
     self.assertCountEqual(result, expected)
Esempio n. 2
0
 def test_insert_data_with_new_notice_type(self):
     opp = self.data[0].copy()
     nnt = "new notice type"
     opp['notice type'] = nnt
     with session_scope(self.dal) as session:
         insert_data(session, [opp])
     result = []
     with session_scope(self.dal) as session:
         notices = session.query(Notice).all()
         for n in notices:
             notice = object_as_dict(n)
             notice_type_id = int(notice['notice_type_id'])
             notice_type = fetch_notice_type_by_id(notice_type_id, session)
             self.assertCountEqual(notice_type.notice_type, nnt)
Esempio n. 3
0
 def test_insert_model(self):
     results = {'c': 'd'}
     params = {'a': 'b'}
     score = .99
     with session_scope(self.dal) as session:
         insert_model(session, results=results, params=params, score=score)
     result = []
     with session_scope(self.dal) as session:
         models = session.query(Model).all()
         for m in models:
             model = object_as_dict(m)
             model.pop('create_date')
             result.append(model)
     expected = [{
         'id': 1,
         'results': results,
         'params': params,
         'score': score
     }]
     self.assertCountEqual(result, expected)
Esempio n. 4
0
 def test_insert_updated_nightly_file_day_two(self):
     '''
     Simulate a second batch entry with a repeating solnbr that now has attachments
     '''
     with session_scope(self.dal) as session:
         insert_updated_nightly_file(session, self.predicted_nightly_data)
     with session_scope(self.dal) as session:
         insert_updated_nightly_file(session,
                                     self.predicted_nightly_data_day_two)
     result = []
     with session_scope(self.dal) as session:
         notices = session.query(Notice).all()
         for n in notices:
             notice = object_as_dict(n)
             #pop the date and createdAt attributes since they're constructed programmatically
             notice.pop('date')
             notice.pop('createdAt')
             if notice['history']:
                 notice['history'][0]['date'] = "test date"
             result.append(notice)
     expected = [{
         'id': 1,
         'notice_type_id': 6,
         'solicitation_number': 'rfp-e-bpm-djf-18-0800-pr-0000828',
         'agency': 'department of justice',
         'notice_data': {
             'url': 'url',
             'zip': '20535',
             'date': '0506',
             'desc': '  link to document',
             'year': '18',
             'naics': '511210',
             'ntype': 'combine',
             'offadd': '935 pennsylvania avenue, n.w. washington dc 20535',
             'office': 'federal bureau of investigation',
             'popzip': '20535',
             'contact':
             'clark kent, contracting officer, phone 5555555555, email [email protected]',
             'subject':
             'enterprise business process management software tool',
             'classcod': '70',
             'location': 'procurement section',
             'setaside': 'n/a',
             'popaddress': '935 pennsylvania ave. n.w. washington, dc  ',
             'popcountry': 'us'
         },
         'compliant': 0,
         'feedback': None,
         'history': None,
         'action': None,
         'updatedAt': None
     }, {
         'id': 2,
         'notice_type_id': 5,
         'solicitation_number': 'spe4a618t934n',
         'agency': 'defense logistics agency',
         'notice_data': {
             'url': 'test_url',
             'zip': '23297',
             'date': '0506',
             'desc': 'test123',
             'year': '18',
             'naics': '334511',
             'offadd': '334511',
             'office': 'dla acquisition locations',
             'contact': '*****@*****.**',
             'subject': 'subject',
             'archdate': '06132018',
             'classcod': '66',
             'location': 'dla aviation - bsm',
             'respdate': '051418',
             'setaside': 'n/a  '
         },
         'compliant': 0,
         'feedback': None,
         'history': None,
         'action': None,
         'updatedAt': None
     }, {
         'id':
         3,
         'notice_type_id':
         6,
         'solicitation_number':
         'spe4a618t934n',
         'agency':
         'defense logistics agency',
         'notice_data': {
             'url': 'test_url',
             'zip': '23297',
             'date': '0506',
             'desc': 'test123',
             'year': '17',
             'naics': '334511',
             'offadd': '334511',
             'office': 'dla acquisition locations',
             'contact': '*****@*****.**',
             'subject': 'subject',
             'archdate': '06132018',
             'classcod': '66',
             'location': 'dla aviation - bsm',
             'respdate': '051418',
             'setaside': 'n/a  '
         },
         'compliant':
         0,
         'feedback':
         None,
         'history': [{
             "date": "test date",
             "user": "",
             "action": "Solicitation Updated on FBO.gov",
             "status": ""
         }],
         'action':
         None,
         'updatedAt':
         None
     }]
     self.assertEqual(result, expected)