Python Admin.add_supervisor Examples

Example #1

File: test_admin.py Project: riyaz489/AIMS

 def test_add_supervisor_second_exception(self, mock_db):
     mock_db().commit_data.side_effect = ['', '', sqlite3.IntegrityError]
     admin = Admin()
     admin.add_supervisor(1, 1)
     self.assertEqual(3, mock_db().commit_data.call_count)

Example #2

File: test_admin.py Project: riyaz489/AIMS

 def test_add_supervisor_third_exception(self, mock_db):
     mock_db().change_accident_to_wip.side_effect = Exception
     admin = Admin()
     admin.add_supervisor(1, 1)
     mock_db().rollback_data.assert_called_once()

Example #3

File: test_admin.py Project: riyaz489/AIMS

 def test_add_supervisor(self, mock_db):
     admin = Admin()
     admin.add_supervisor(1, 1)
     self.assertEqual(3, mock_db().commit_data.call_count)