Add ClassTrait
Add basic test for MockTrait