def test_create_financial_req(self): c = Controller(self.test_db) req_id = c.create_financial_req(event_id='ev1234', req_amount=321, reason='reason1', req_dpt='service') req_data = c.get_financial_req('id', req_id)[0] self.assertEqual('ev1234', req_data['event_id']) self.assertEqual(321, req_data['req_amount']) self.assertEqual('reason1', req_data['reason']) self.assertEqual('service', req_data['req_dpt'])
def test_update_financial_req(self): c = Controller(self.test_db) req_id = c.create_financial_req(event_id='ev1234', req_amount=321, reason='reason1', req_dpt='service') updated_fin_req = {'id': req_id, 'event_id': 'ev321', 'req_amount': 3213, 'reason': 'reason2', 'req_dpt': 'production'} c.update_financial_req(**updated_fin_req) req_data = c.get_financial_req('id', req_id)[0] self.assertEqual('ev321', req_data['event_id']) self.assertEqual(3213, req_data['req_amount']) self.assertEqual('reason2', req_data['reason']) self.assertEqual('production', req_data['req_dpt'])