rm-hull / dot-block / 27101107513

61%

push

github

feat: add integrarion test (+support graceful shutdown and optional IP2Location) (#144)

This PR introduces several improvements to server lifecycle management, configuration flexibility, and testing.

### Changes

- **Graceful Shutdown**:
    - Updated `RunServer` to accept `context.Context`.
    - Implemented `monitorShutdown` to ensure HTTP, UDP DNS, TCP DNS, and DoT servers shut down cleanly.
    - Integrated `signal.NotifyContext` in `main.go` to trigger shutdown on `SIGINT`/`SIGTERM`.
    - Switched to `errgroup.WithContext` for better group lifecycle management.

- **Optional IP2Location**:
    - Added `DisableIp2Location` configuration flag.
    - Refactored IP2Location initialization into a dedicated method.
    - Added safety checks in metrics recording to handle disabled geolocation.

- **Local Blocklist Support**:
    - Added support for `file://` URIs in blocklists, allowing local files to be used as sources.

- **Integration Testing**:
    - Added `internal/integration_test.go` which performs end-to-end tests of the server's DNS functionality across UDP, TCP, and DoT (plain) protocols.

- **General Cleanup**:
    - Various formatting and alignment fixes.

63 of 95 new or added lines in 5 files covered. (66.32%)

1215 of 1991 relevant lines covered (61.02%)

883.64 hits per line