Add tests to the ClassDefinitionCode