Best Testng code snippet using org.testng.Interface IInvokedMethod.getTestMethod
Source:SeleniumGridListener.java  
...133        }134        logger.exiting();135    }136    private boolean isSeLionAnnotatedTestClass(IInvokedMethod method) {137        Class<?> cls = method.getTestMethod().getInstance().getClass();138        final boolean isWebTestClass = cls.getAnnotation(WebTest.class) != null;139        final boolean isMobileTestClass = cls.getAnnotation(MobileTest.class) != null;140        return isMobileTestClass || isWebTestClass;141    }142    private boolean isValidBeforeCondition(IInvokedMethod method) {143        if (method.isTestMethod()) {144            return true;145        }146        return method.getTestMethod().isBeforeClassConfiguration();147    }148    private void testSessionSharingRules(IInvokedMethod method) {149        Test t = method.getTestMethod().getInstance().getClass().getAnnotation(Test.class);150        if (t != null && t.singleThreaded()) {151            if (!isPriorityUnique(method)) {152                throw new IllegalStateException(153                        "All the session sharing test methods within the same class should have a unique priority.");154            } else {155                return;156            }157        }158        throw new IllegalStateException(159                "Session sharing test should have a class level @Test annotation with the property singleThreaded = true defined.");160    }161    private boolean isLowPriority(IInvokedMethod method) {162        int low = method.getTestMethod().getPriority();163        for (ITestNGMethod test : method.getTestMethod().getTestClass().getTestMethods()) {164            // ensures all test methods have the @Test annotation. Throw exception if that's not the case165            if (!isAnnotatedWithTest(test.getConstructorOrMethod().getMethod())) {166                throw new IllegalStateException(167                        "Session sharing requires all test methods to define a priority via the @Test annotation.");168            }169            if (test.getEnabled() && test.getPriority() < low) {170                return false;171            }172        }173        Test t = method.getTestMethod().getConstructorOrMethod().getMethod().getAnnotation(Test.class);174        // If there is an existing session and the test method has a DP then don't create a session175        // For a data driven test method with the first data the session must be created176        // Hence return true if currentInvocationCount is 1 otherwise utilize the same session177        // by returning false178        int currentInvocationCount = method.getTestMethod().getCurrentInvocationCount();179        if (!t.dataProvider().isEmpty()) {180            return currentInvocationCount == 0;181        }182        return true;183    }184    private boolean isHighPriority(IInvokedMethod method) {185        if (!isAnnotatedWithTest(method)) {186            // Abort. There will already be an exception thrown in isLowPriority for this case.187            return true;188        }189        int high = method.getTestMethod().getPriority();190        for (ITestNGMethod test : method.getTestMethod().getTestClass().getTestMethods()) {191            if (test.getEnabled() && test.getPriority() > high) {192                return false;193            }194        }195        Test t = method.getTestMethod().getConstructorOrMethod().getMethod().getAnnotation(Test.class);196        // For a test method with a data provider197        if (!(t.dataProvider().isEmpty())) {198            int currentInvocationCount = method.getTestMethod().getCurrentInvocationCount();199            int parameterInvocationCount = method.getTestMethod().getParameterInvocationCount();200            // If the data set from the data provider is exhausted201            // It means its the last method with the data provider- this is the exit condition202            return (currentInvocationCount == parameterInvocationCount);203        }204        return true;205    }206    private boolean isAnnotatedWithTest(IInvokedMethod method) {207        return isAnnotatedWithTest(method.getTestMethod().getConstructorOrMethod().getMethod());208    }209    private boolean isAnnotatedWithTest(Method method) {210        Test t = method.getAnnotation(Test.class);211        return t != null;212   }213    private boolean isPriorityUnique(IInvokedMethod method) {214        // Logic to check priorities of all test methods are unique. This is used in Session Sharing.215        Set<Integer> check = new HashSet<Integer>();216        int length = method.getTestMethod().getTestClass().getTestMethods().length;217        int expectedSize = 0;218        for (int i = 0; i < length; i++) {219            //ignore @Test(enabled = false) methods220            if (!method.getTestMethod().getTestClass().getTestMethods()[i].getEnabled()) {221                continue;222            }223            check.add(method.getTestMethod().getTestClass().getTestMethods()[i].getPriority());224            expectedSize += 1;225            if (check.size() != expectedSize) {226                return false;227            }228        }229        return true;230    }231    /**232     * Executes when test case is finished<br>233     * 234     * Identify if webtest wants to have session open, otherwise close session<br>235     * <b>sample</b><br>236     * @webtest(browser="*firefox", <b>keepSessionOpen = true</b>)<br>237     * Analyzes failure if any238     * 239     * @see org.testng.IInvokedMethodListener#afterInvocation(org.testng.IInvokedMethod, org.testng.ITestResult)240     * 241     */242    @Override243    public void afterInvocation(IInvokedMethod method, ITestResult testResult) {244        logger.entering(new Object[] { method, testResult });245        try {246            if (ListenerManager.isCurrentMethodSkipped(this)) {247                logger.exiting(ListenerManager.THREAD_EXCLUSION_MSG);248                return;249            }250            // Abort at this point, if there is no AbstractTestSession instance.251            if (Grid.getTestSession() == null) {252                return;253            }254            // For non-session sharing, we only allow our annotation(s) on @Test methods.255            // When this condition is true, we allow the session to be closed.256            if (!method.isTestMethod() && !isSeLionAnnotatedTestClass(method)) {257                return;258            }259            // For session sharing, we only allow our annotation on the class.260            // In this case the session can only be closed in the @Test with the lowest priority (last test, biggest261            // number) or in an @AfterClass262            if (isSeLionAnnotatedTestClass(method)) {263                if (!isValidAfterCondition(method)) {264                    return;265                }266                if (method.isTestMethod() && hasValidAfterCondition(method)) {267                    return;268                }269                if (method.isTestMethod() && !isHighPriority(method)) {270                    // For session sharing tests, Need to close session only for Test (Web or Mobile) with highest271                    // priority (last test) in the class.272                    return;273                }274            }275            // let's attempt to capture a screenshot in case of failure from Selenium or SeLion PageObject276            // or when there was an assertion failure.277            // That way a user can see the how the page looked like when a test failed.278            if (testResult.getStatus() == ITestResult.FAILURE279                    && (testResult.getThrowable() instanceof WebDriverException ||280                    testResult.getThrowable() instanceof AssertionError)) {281                warnUserOfTestFailures(testResult);282            }283            AbstractTestSession testSession = Grid.getTestSession();284            testSession.closeSession();285            testResult.setAttribute(JsonRuntimeReporterHelper.IS_COMPLETED, true);286        } catch (Exception e) { // NOSONAR287            logger.log(Level.WARNING, "An error occurred while processing afterInvocation: " + e.getMessage(), e);288        }289        logger.exiting();290    }291    private boolean isValidAfterCondition(IInvokedMethod method) {292        return method.isTestMethod() || method.getTestMethod().isAfterClassConfiguration();293    }294    private boolean hasValidAfterCondition(IInvokedMethod method) {295        return method.getTestMethod().getTestClass().getAfterClassMethods().length > 0;296    }297    private void warnUserOfTestFailures(ITestResult testResult) {298        // don't bother if we don't have a session299        if (!Grid.getTestSession().isStarted()) {300            return;301        }302        // don't bother if we don't have a web or mobile test session303        if (!(Grid.getTestSession() instanceof MobileTestSession) &&304                !(Grid.getTestSession() instanceof WebTestSession)) {305            return;306        }307        String errMsg = "";308        if (testResult.getThrowable() != null) {309            errMsg = testResult.getThrowable().getMessage();...Source:AbstractChainedListener.java  
...77    }78    79    @Override80    public void beforeInvocation(IInvokedMethod method, ITestResult testResult) {81        if (method.getTestMethod().isBeforeMethodConfiguration()) {82            beforeMethodBefore.add(testResult.getName());83        } else if (method.isTestMethod()) {84            testMethodBefore.add(testResult.getName());85        } else if (method.getTestMethod().isAfterMethodConfiguration()) {86            afterMethodBefore.add(testResult.getName());87        }88    }89    @Override90    public void afterInvocation(IInvokedMethod method, ITestResult testResult) {91        if (method.getTestMethod().isBeforeMethodConfiguration()) {92            beforeMethodAfter.add(testResult.getName());93        } else if (method.isTestMethod()) {94            testMethodAfter.add(testResult.getName());95        } else if (method.getTestMethod().isAfterMethodConfiguration()) {96            afterMethodAfter.add(testResult.getName());97        }98    }99    @Override100    public void onBeforeClass(ITestClass testClass) {101        beforeClass.add(testClass.getRealClass().getSimpleName());102    }103    @Override104    public void onAfterClass(ITestClass testClass) {105        afterClass.add(testClass.getRealClass().getSimpleName());106    }107    @Override108    public void onTestStart(ITestResult result) {109        testStarted.add(result.getName());...Source:MyAnnotationListener.java  
...45//    }46    @SneakyThrows47    public void beforeInvocation(IInvokedMethod method, ITestResult testResult){48        if (method.isTestMethod()){49            final Method method1 = method.getTestMethod().getConstructorOrMethod().getMethod();50            if ((method1.isAnnotationPresent(TestApi.class)) || method.getTestMethod().getInstance().getClass().isAnnotationPresent(TestApi.class)){51                //intefaceNameï¼æ¥å£åç§°52                String intefaceName = method1.isAnnotationPresent(TestApi.class)?53                    method1.getAnnotation(TestApi.class).name():54                    method.getTestMethod().getInstance().getClass().getAnnotation(TestApi.class).name();55                if (StringUtils.isBlank(intefaceName)){56                    logger.error("current test method has not unspecified iterface name");57                    throw new MissingRequestPartException("TestApi.name()");58                }59                BaseRequest tmpReq = getRequestExtend(intefaceName);60                //è·åæ¥å£æµè¯æ¹æ³ç»§æ¿çbasttestä¸çbaseRequest屿§61                Field request = method.getTestMethod().getInstance().getClass().getSuperclass().getDeclaredField("baseRequest");62                //å°ä»æ¥å£æä»¶ä¸æ¿å°çextend屿§ï¼æåè£
å
¥è¯·æ±æ¹æ³ä¸ï¼ä¸ºåç»è¯·æ±åå¤63                request.set(method.getTestMethod().getInstance(), tmpReq);64            }65        }66    }67    /*68    æ ¹æ®æ¥å£æè¿°æä»¶ï¼è·årequestçç¸å
³å±æ§ï¼å¹¶ä»¥baserequestå½¢å¼è¿å69     */70    private BaseRequest getRequestExtend(String intefaceName){71        //todo æ ¹æ®configä¸çè§£æå¼ææ¥å³å®ç±é£ä¸ªserverè§£æ72        InterfaceParseService service = new DefaultParseImpl();73        String basePath = service.getBasePath(intefaceName);74        String serviceMethod = service.getMethod(intefaceName);75        String mine = service.getMINE(intefaceName);76        BaseRequest request = new BaseRequest();77        request.setUrl(host+basePath);...Source:TestExecution.java  
...25	@Override26	public void beforeInvocation(IInvokedMethod method, ITestResult testResult) {27		AndroidDriver<MobileElement> driver;28		extent = ExtentReportUtility.GetExtent();29		test = ExtentReportUtility.createTest(method.getTestMethod().getMethodName(),method.getTestMethod().getMethodName());30		test.log(Status.INFO, "Started executing "+ method.getTestMethod().getMethodName());31		System.out.println("Started executing "+ method.getTestMethod().getMethodName());32		if(TestManager.getReportLogger()== null){33			TestManager.setReportLogger(test);34		}35		ConstructorOrMethod consOrMethod =36				method.getTestMethod().getConstructorOrMethod();37	        DisableListener disable = consOrMethod.getMethod().getDeclaringClass().getAnnotation(DisableListener.class);38	        if (disable != null) {39	            return;40	        }41		String deviceId = method.getTestMethod().getXmlTest().getLocalParameters().get("deviceId");42		String deviceName = method.getTestMethod().getXmlTest().getLocalParameters().get("deviceName");43		String portNumber = method.getTestMethod().getXmlTest().getLocalParameters().get("portNumber");44		try {45			driver = DriverFactory.getDriver(deviceId,deviceName,portNumber);46			if (driver!=null) {47				TestManager.setAndroidDriver(driver);48			}49			test.assignCategory(deviceName);50			System.out.println("Method Name is  "+ method.getTestMethod().getMethodName());51		} catch (MalformedURLException e) {52			e.printStackTrace();53		}54	}55	@Override56	public void afterInvocation(IInvokedMethod method, ITestResult testResult) {57		try {58			test.log(Status.INFO, "Sucessfully executed "+ method.getTestMethod().getMethodName());59			ConstructorOrMethod consOrMethod =60					method.getTestMethod().getConstructorOrMethod();61		        DisableListener disable = consOrMethod.getMethod().getDeclaringClass().getAnnotation(DisableListener.class);62		        if (disable != null) {63		            return;64		        }65			TestManager.tearDown();66			DriverFactory.closeApp();67			ExtentReportUtility.endReport();68			System.out.println("Sucessfully executed-  "+ method.getTestMethod().getMethodName());69		} catch (Exception e) {70			e.printStackTrace();71		}72	}73}...Source:XrayListener.java  
...16     * @see org.testng.IInvokedMethodListener#beforeInvocation(org.testng.IInvokedMethod, org.testng.ITestResult)17     */18    public void beforeInvocation(IInvokedMethod method, ITestResult testResult) {19        if(method.isTestMethod() && annotationPresent(method, Xray.class) ) {20            testResult.setAttribute("requirement", method.getTestMethod().getConstructorOrMethod().getMethod().getAnnotation(Xray.class).requirement());21            testResult.setAttribute("test", method.getTestMethod().getConstructorOrMethod().getMethod().getAnnotation(Xray.class).test());22            testResult.setAttribute("labels", method.getTestMethod().getConstructorOrMethod().getMethod().getAnnotation(Xray.class).labels());23        }24    }25    private boolean annotationPresent(IInvokedMethod method, Class clazz) {26        boolean retVal = method.getTestMethod().getConstructorOrMethod().getMethod().isAnnotationPresent(clazz) ? true : false;27        return retVal;28    }29    /* (non-Javadoc)30     * @see org.testng.IInvokedMethodListener#afterInvocation(org.testng.IInvokedMethod, org.testng.ITestResult)31     */32    public void afterInvocation(IInvokedMethod method, ITestResult testResult) {33        if(method.isTestMethod()) {34            if( !testSuccess ) {35                testResult.setStatus(ITestResult.FAILURE);36                testResult.setAttribute("env",System.getProperty("testedEnv"));37                testResult.setAttribute("version",System.getProperty("ActualVersion"));38            }39        }40    }...Source:SauceLabsIntegration.java  
...21				LOGGER.debug("After in SL Integration driver Session ID: "22						+ driver.getSessionId());23				WebInterface slWebInterface = new WebInterface();24				slWebInterface.updateSauceLabsJob(driver.getSessionId()25						.toString(), method.getTestMethod().getMethodName(),26						method.getTestResult().isSuccess());27				Reporter.setCurrentTestResult(testResult);28				Reporter.log("SauceLabs Report :: ");29				Reporter.log(slWebInterface.generateLinkForJob(driver30						.getSessionId().toString()));31				// Reporter.log(slWebInterface.generateLinkForEmbedScript(driver32				// .getSessionId().toString(), true));33				// Reporter.log("<br>");34				// Reporter.log("SauceLabs Video :: "35				// + slWebInterface.generateLinkForEmbedScript(driver36				// .getSessionId().toString(), false));37				Reporter.log("<br>");38			}39		} catch (Exception e) {...Source:CustomListener1.java  
...9public class CustomListener1 implements IInvokedMethodListener {10	@Override11	public void beforeInvocation(IInvokedMethod method, ITestResult testResult) {    //it will run before every method (all method present in the class which every class going to implement it)12		System.out.println("beforeInvocation: " + testResult.getTestClass().getName() + 13				" => " + method.getTestMethod().getMethodName() );14	}15	@Override16	public void afterInvocation(IInvokedMethod method, ITestResult testResult) {17		System.out.println("AfterInvocation: " + testResult.getTestClass().getName() + 18				" => " + method.getTestMethod().getMethodName() );19	}20}...Source:InvokedMethodListener.java  
...6public class InvokedMethodListener implements IInvokedMethodListener{7	@Override8	public void beforeInvocation(IInvokedMethod method, ITestResult testResult) {9		//dice cuanto test se ha ejecutado10		System.out.println("Antes de invocar" + method.getTestMethod().getMethodName());11		12	}13	@Override14	public void afterInvocation(IInvokedMethod method, ITestResult testResult) {15		System.out.println("Después de invocar" + method.getTestMethod().getMethodName());16		17	}18}...getTestMethod
Using AI Code Generation
1public void testMethod1() {2    Assert.assertTrue(true);3}4public void testMethod2() {5    Assert.assertTrue(true);6}7public void testMethod3() {8    Assert.assertTrue(true);9}10public void testMethod4() {11    Assert.assertTrue(true);12}13public void testMethod5() {14    Assert.assertTrue(true);15}16public void testMethod6() {17    Assert.assertTrue(true);18}19public void testMethod7() {20    Assert.assertTrue(true);21}22public void testMethod8() {23    Assert.assertTrue(true);24}25public void testMethod9() {26    Assert.assertTrue(true);27}28public void testMethod10() {29    Assert.assertTrue(true);30}31public void testMethod11() {32    Assert.assertTrue(true);33}34public void testMethod12() {35    Assert.assertTrue(true);36}37public void testMethod13() {38    Assert.assertTrue(true);39}40public void testMethod14() {41    Assert.assertTrue(true);42}43public void testMethod15() {44    Assert.assertTrue(true);45}46public void testMethod16() {47    Assert.assertTrue(true);48}49public void testMethod17() {50    Assert.assertTrue(true);51}52public void testMethod18() {53    Assert.assertTrue(true);54}55public void testMethod19() {56    Assert.assertTrue(true);57}58public void testMethod20() {59    Assert.assertTrue(true);60}61public void testMethod21() {62    Assert.assertTrue(true);63}64public void testMethod22() {65    Assert.assertTrue(true);66}67public void testMethod23() {68    Assert.assertTrue(true);69}70public void testMethod24() {71    Assert.assertTrue(true);72}73public void testMethod25() {74    Assert.assertTrue(true);75}76public void testMethod26() {77    Assert.assertTrue(true);78}79public void testMethod27() {80    Assert.assertTrue(true);81}82public void testMethod28() {83    Assert.assertTrue(true);84}85public void testMethod29() {86    Assert.assertTrue(true);87}88public void testMethod30() {89    Assert.assertTrue(true);90}91public void testMethod31() {92    Assert.assertTrue(true);93}getTestMethod
Using AI Code Generation
1public class TestNgTestName {2    public void test1() {3        System.out.println("Test 1");4    }5    public void test2() {6        System.out.println("Test 2");7    }8    public void test3() {9        System.out.println("Test 3");10    }11}TestNG is a Java-based open-source framework for test automation that includes various test types, such as unit testing, functional testing, E2E testing, etc. TestNG is in many ways similar to JUnit and NUnit. But in contrast to its competitors, its extensive features make it a lot more reliable framework. One of the major reasons for its popularity is its ability to structure tests and improve the scripts' readability and maintainability. Another reason can be the important characteristics like the convenience of using multiple annotations, reliance, and priority that make this framework popular among developers and testers for test design. You can refer to the TestNG tutorial to learn why you should choose the TestNG framework.
You can push your abilities to do automated testing using TestNG and advance your career by earning a TestNG certification. Check out our TestNG certification.
Watch this complete tutorial to learn how you can leverage the capabilities of the TestNG framework for Selenium automation testing.
Get 100 minutes of automation test minutes FREE!!
