def results(): if request.method == 'POST': results_connection = sqlite3.connect('results.sqlite3') service_connection = sqlite3.connect('service.sqlite3') results_cursor = results_connection.cursor() service_cursor = service_connection.cursor() student_data = dict( first_name = request.form['first_name'], second_name = request.form['second_name'], last_name = request.form['last_name'], school_id = request.form['login_statgrad'].strip('sch') ) logging.debug(student_data) results_response = get_results(student_data, results_cursor, service_cursor) service_connection.close() results_connection.close() if results_response['status'] == 'OK': return render_template('results.html', results_table=results_response['table']) else: return render_template('results.html') else: return render_template('results.html')
def test_get_results_not_found(self): real_results = get_results(self.not_existing_student, self.cursor) right_results = {'status': 'NOT FOUND'} self.assertEqual(real_results, right_results)
def test_get_results(self): real_results = get_results(self.existing_student, self.cursor) right_results = {'status': 'OK' , 'table': [['test', '100', 'Победитель', 2007]]} self.assertEqual(real_results, right_results)