Testing with proper mock