def test_db_script(self, mock_cursor): db = ConnectDb() db.db_script() mock_cursor().cursor().executescript.assert_called_once() mock_cursor().commit.assert_called_once() mock_cursor().cursor().close.assert_called_once() mock_cursor().close.assert_called_once()
def test_db_script_fail(self, mock_cursor, mock_sys, mock_connect): db = ConnectDb() db.db_script() mock_sys.exit.assert_called_once() mock_cursor().cursor().close.assert_called_once() mock_cursor().rollback.assert_called_once() mock_cursor().close.assert_called_once()
member = Member() member.member_id = result[0] member.member_features() if result is None or flag: print(Color.F_Red + 'wrong credentials' + Base.END) input() main() except Exception as e: print(e) sys.exit() finally: conn.close_conn() # only executes when login file run as main script or run directly from terminal if __name__ == '__main__': """ creating db schema and calling main function """ con = ConnectDb() con.db_script() main()