TestNG Automation Framework : Customize SoftAssert

In this post, we will see how to customize SoftAssert in TestNG Framework. Sometimes, we might write more than one assert in a @Test method and want to make all the asserts run even if one or more fails. This can be achieved using SoftAssert in TestNG ( SoftAssert in TestNG Framework ).

However, we will try to customize SoftAssert. Lets us see how to run all asserts and capture screenshots for all failed asserts in a test method. To modify the default behavior sub classes  should override various hooks provided by Assertion class.

Taking screenshot when an Assert fails in TestNG tests.

Basic things:

SoftAssert just records the failure when an assertion fails. Calling .assertAll() will cause an exception to be thrown if at least one assertion has failed in our test. It just like collecting errors while assertion.

protected void doAssert(IAssert<?> a) {

.....

 

IAssert is an interface in TestNG for the assert command.It has specification for getting assert message, doAssert , getActual and getExpected.

CustomSoftAssert

public class CustomSoftAssert extends Assertion {

.....

.....

 

Our CustomSoftAssert would retain doAssert() and .assertAll() but hooks custom implementation to capture screenshot by overriding below method:

 @Override
 public void onAssertFailure(IAssert<?> assertCommand, AssertionError ex)
{
// taking screenshot code goes here .....
}

 

TestNG allows you to extend Assertion and do some custom implementation as opposed to default framework behavior. By this approach we can hook some piece of code ( custom implementations ). We can hook or override methods for alternative implementation. For example, the capture screenshot code can be hooked to CustomSoftAssert().

Instantiate Class:

CustomSoftAssert customsoftassert  = new CustomSoftAssert();

Sample test method:

@Test
      public void sampleTilteFailureTests()
      {
         System.setProperty("webdriver.edge.driver", "MicrosoftWebDriver.exe");
          driver=new EdgeDriver();
         CustomSoftAssert csa = new CustomSoftAssert();
         driver.get("https://www.bing.com/");
         driver.manage().window().maximize();
         WebDriverWait wait = new WebDriverWait(driver, 120);
         wait.until(ExpectedConditions.titleContains("Bing"));
         csa.assertEquals("BogusTitle", driver.getTitle(),"Assert1,this will fail");
          
          driver.get("https://start.duckduckgo.com/");
          driver.manage().window().maximize();
          WebDriverWait wait1 = new WebDriverWait(driver, 120);
          wait1.until(ExpectedConditions.titleContains("Duck"));
          csa.assertEquals("BogusTitle", driver.getTitle(),"Assert2,this too will fail");
          csa.assertAll();
               
      }

 

Note that : I am capturing the screenshot with assert message in the filenames.

Related Posts