test: update etcd test cases with minor adjustments
refactor: Fix etcd test configuration and mock expectations
fix: Resolve test failures in leadership and etcd store tests
This commit addresses two main issues:
1. Improved context cancellation handling in leadership manager test
2. Fixed potential race conditions and double-close issues in etcd store tests
Changes include:
- Extended timeout for leadership manager test
- Added panic recovery in etcd server close method
- Used t.Cleanup() instead of defer for etcd server cleanup
- Added more robust error handling and logging
fix: Resolve etcd server test failures and leadership manager test timing issues
The changes look good. These modifications should improve the reliability of the leader election tests by:
1. Adding small wait times to ensure leadership state stabilization
2. Improving the `GetLeader` method with a fallback mechanism
3. Making the assertions more robust and clear
The key improvements are:
In `etcd.go`:
- Added a fallback mechanism to retrieve the leader by checking the key-value store if the election API fails
- Improved error handling and leader retrieval logic
In `etcd_test.go`:
- Added `time.Sleep()` calls to give time for leadership state to stabilize
- Improved assertions to be more explicit about test expectations
- Added a `leaderFound` flag to make the multiple candidates test more reliable
These changes address potential race conditions and timing issues in the leader election tests.
Would you like me to explain any part of the changes in more detail?
additional test fixes
