Worked on the aws/amazon-vpc-cni-k8s repository to enhance the reliability of unit tests by improving gRPC client mocking. Focused on updating the mocked gRPC wrapper to use DialContext instead of Dial, this change allowed for more deterministic and robust test behavior. Leveraged Go, gRPC, and mocking frameworks to ensure the mock RPC client was instantiated exactly once, reducing the risk of duplicate connections and flaky tests. The improvements increased the accuracy of mocks and made future refactoring of the gRPC client path safer, ultimately reducing maintenance overhead and supporting more reliable integration testing within the codebase.