add back JNITest class

This reverts commit f6b4241bafb676db8cc9620be46c8b38f9f2b27c.