Add more tests for method swizzling