Intellij: Change JUnit Test Class template

The link provided by CrazyCoder suggests that this is not directly possible but that I you need to add if statements to the Class File Type template. I will probably split the template in two using a big if-else.

This is now solved in the latest IntelliJ IDEA / Android Studio. Go to:

Settings
-Editor
-File and Code Templates
-Code (tab)
-JUnit4 Test Class 

edit the code in the right window.