javaClass.addImport("org.jboss.arquillian.container.test.api.Deployment");
javaClass.addImport("org.jboss.arquillian.junit.Arquillian");
javaClass.addImport("org.junit.runner.RunWith");
javaClass.addImport("org.jboss.forge.arquillian.AddonDependency");
javaClass.addImport("org.jboss.forge.arquillian.Dependencies");
javaClass.addImport("org.jboss.forge.arquillian.archive.ForgeArchive");
javaClass.addImport("org.jboss.forge.furnace.repositories.AddonDependencyEntry");
javaClass.addImport("org.jboss.shrinkwrap.api.ShrinkWrap");
// Add Arquillian annotation
javaClass.addAnnotation("RunWith").setLiteralValue("Arquillian.class");