Add missing @Test annotations