def test_delete_student_when_student_num_not_on_file(
self, mock_input, mock_stdout):
with mock.patch('builtins.open',
return_value=io.StringIO(original_file)):
expected_output = "\nThe student number you entered is not on file.\n\n"
delete_student()
self.assertEqual(mock_stdout.getvalue(), expected_output)
def test_delete_student_unsuccessful(self, mock_input, mock_stdout):
with mock.patch('builtins.open',
side_effect=[
io.StringIO(original_file),
io.StringIO(original_file),
io.StringIO(original_file),
io.StringIO(original_file)
]):
expected_output = "\nThe student could not be deleted.\n\n"
delete_student()
self.assertEqual(mock_stdout.getvalue(), expected_output)
def test_delete_student(self, mock_input):
crud.file_write([])
crud.add_student()
crud.delete_student()
self.assertEqual(0, len(crud.file_read()))
def test_delete_student_invalid(self, mock_stdout, mock_input):
crud.delete_student()
self.assertTrue("does not exist" in mock_stdout.getvalue())
def delete_student(student_id: int, db: Session = Depends(get_db)):
return crud.delete_student(db=db, student_id=student_id)