JUnit类别-如何避免代码重复

Question

Let's say I have two test classes annotated with proper categories:

@Category({CategoryA.class})
public class A {
   ...
}

and

@Category({CategoryB.class})
public class B {
   ...
}

which share common codebase and differ only in 1 line of test method. What would be the best approach to make this code better? For example, if I use only one class:

@Category({CategoryA.class, CategoryB.class})
public class AB {
   ...
}

Is there any way to determine category in runtime and add simple if statement?

Answer 1

Any reason you can't have an abstract class which does most of the work, then two extending classes and have different @Category annotations? Something like:

public abstract class ABParent {
    protected abstract void thingThatsDifferent();
}

@Category(CategoryA.class)
public class A extends ABParent {
    @Override
    protected void thingThatsDifferent() {
        System.out.println("Woof");
    }
}

@Category(CategoryB.class)
public class B extends ABParent {
    @Override
    protected void thingThatsDifferent() {
        System.out.println("Meow");
    }
}