Revert "removed test class"

This reverts commit 390f084cc00f4864d3f350f4c994f06aa195b8d7.