Testng-为所有数据提供者运行测试方法后如何运行清除代码？

Question

I have a testng test (below) where I use one value of "totalAmount" for each array given by the data provider "junkDP". I want to reset "totalAmount", ONLY after "test" method has run for each array in "junkDP". Is this possible in testng ? How to do it ?

Please note that @AfterMethod & @AfterTest do not do what I want. @AfterMethod resets "totalAmount" before "test" method is run for each array after the first array in "junkDP". @AfterTest is run after the @AfterClass.

The code:

import org.testng.annotations.DataProvider;
import org.testng.annotations.*;

public class JUNKdP {

    @DataProvider( name = "junkDP")
    public static Object[][] junkDP() {
        Object [] [] dataSet = new Object[][] {
                new Object[] {1, 2},
                new Object[] {3, 4},
                new Object[] {5, 6}};
        return dataSet;
    }

}


public class JUNK {

    private int totalAmount = 0;

    @BeforeClass
    public void beforeClass(){System.out.println("BeforeClass\n");}

    @BeforeMethod
    public void beforeMethod(){
        System.out.println("BeforeMethod\n");
    }

    @AfterMethod
    public void afterMethod(){
        System.out.println("AfterMethod\n");
    }

    @AfterTest
    public void afterTest(){
        System.out.println("AfterTest\n");
        this.totalAmount = 0;
        System.out.println("Reset total amount to 0");
    }

    @AfterClass
    public void afterClass(){System.out.println("AfterClass\n");}

    @Test(dataProvider = "junkDP", dataProviderClass = JUNKdP.class, enabled = true)
    public void test(int a, int b){
        System.out.println("Test method");
        int sum = a + b;
        System.out.println("Sum: " + sum);
        this.totalAmount = this.totalAmount + sum;
        System.out.println("totalAmount: " + this.totalAmount + "\n");
    }
}

Output:

BeforeClass

BeforeMethod

Test method
Sum: 3
totalAmount: 3

AfterMethod

BeforeMethod

Test method
Sum: 7
totalAmount: 10

AfterMethod

BeforeMethod

Test method
Sum: 11
totalAmount: 21

AfterMethod

AfterClass

AfterTest

Reset total amount to 0

Answer 1

If i understand the question, you are searching for executor after the test method ?

...if yes, depending on the test runner

here you go, how testng is doing it:

http://testng.org/doc/documentation-main.html#annotations

@AfterTest or @AfterMethod