Skip to content

Refactor env var guard#16

Merged
leynos merged 4 commits intomainfrom
codex/create-envvarguard-and-refactor-tests
Jul 31, 2025
Merged

Refactor env var guard#16
leynos merged 4 commits intomainfrom
codex/create-envvarguard-and-refactor-tests

Conversation

@leynos
Copy link
Copy Markdown
Owner

@leynos leynos commented Jul 29, 2025

Summary

  • centralize environment variable guard for tests
  • use the shared guard in configuration steps and tests

Testing

  • make fmt
  • make lint
  • make test
  • make markdownlint

https://chatgpt.com/codex/tasks/task_e_6888f7ea4f2083228cdd174537e3132b

Summary by Sourcery

Centralize environment variable guard logic for tests into a shared support module and update existing tests to use the common implementation

Enhancements:

  • Extract EnvVarGuard and its helper functions into tests/support/env_guard.rs
  • Re-export shared guard utilities through support modules in config and test directories
  • Update config and step tests to import and use the centralized EnvVarGuard and remove_env_var

@sourcery-ai
Copy link
Copy Markdown

sourcery-ai Bot commented Jul 29, 2025

Reviewer's Guide

Centralize environment variable handling by extracting the EnvVarGuard and its helper functions into a shared test support module and updating both the crate’s config module and test suites to use this unified implementation.

Class diagram for centralized EnvVarGuard and helpers

classDiagram
    class EnvVarGuard {
        - key: String
        - original: Option<String>
        + set(key: &str, val: &str) -> Self
        + drop()
    }
    class env_guard {
        + set_env_var(key: &str, val: &str)
        + remove_env_var(key: &str)
    }
    EnvVarGuard <.. env_guard : uses

    %% Show re-export in support module
    class support {
        + EnvVarGuard
        + set_env_var(key: &str, val: &str)
        + remove_env_var(key: &str)
    }
    support <.. env_guard : re-exports
    support <.. EnvVarGuard : re-exports
Loading

File-Level Changes

Change Details Files
Introduce shared EnvVarGuard utilities
  • Add tests/support/env_guard.rs implementing EnvVarGuard, set_env_var, remove_env_var
  • Create tests/support/mod.rs to expose the env_guard module
tests/support/env_guard.rs
tests/support/mod.rs
Refactor config.rs to consume the shared guard
  • Remove local EnvVarGuard, set_env_var, remove_env_var definitions
  • Include the shared implementation via include! macro in an env_guard mod
  • Expose the shared guard in a pub support module and update imports
crates/comenqd/src/config.rs
Update test-step definitions to use the centralized guard
  • Remove local guard and safe set/remove functions from config_steps.rs
  • Import EnvVarGuard and remove_env_var from crate support
  • Replace remove_env calls with remove_env_var
tests/steps/config_steps.rs

Tips and commands

Interacting with Sourcery

  • Trigger a new review: Comment @sourcery-ai review on the pull request.
  • Continue discussions: Reply directly to Sourcery's review comments.
  • Generate a GitHub issue from a review comment: Ask Sourcery to create an
    issue from a review comment by replying to it. You can also reply to a
    review comment with @sourcery-ai issue to create an issue from it.
  • Generate a pull request title: Write @sourcery-ai anywhere in the pull
    request title to generate a title at any time. You can also comment
    @sourcery-ai title on the pull request to (re-)generate the title at any time.
  • Generate a pull request summary: Write @sourcery-ai summary anywhere in
    the pull request body to generate a PR summary at any time exactly where you
    want it. You can also comment @sourcery-ai summary on the pull request to
    (re-)generate the summary at any time.
  • Generate reviewer's guide: Comment @sourcery-ai guide on the pull
    request to (re-)generate the reviewer's guide at any time.
  • Resolve all Sourcery comments: Comment @sourcery-ai resolve on the
    pull request to resolve all Sourcery comments. Useful if you've already
    addressed all the comments and don't want to see them anymore.
  • Dismiss all Sourcery reviews: Comment @sourcery-ai dismiss on the pull
    request to dismiss all existing Sourcery reviews. Especially useful if you
    want to start fresh with a new review - don't forget to comment
    @sourcery-ai review to trigger a new review!

Customizing Your Experience

Access your dashboard to:

  • Enable or disable review features such as the Sourcery-generated pull request
    summary, the reviewer's guide, and others.
  • Change the review language.
  • Add, remove or edit custom review instructions.
  • Adjust other review settings.

Getting Help

@coderabbitai
Copy link
Copy Markdown
Contributor

coderabbitai Bot commented Jul 29, 2025

Warning

Rate limit exceeded

@leynos has exceeded the limit for the number of commits or files that can be reviewed per hour. Please wait 6 minutes and 46 seconds before requesting another review.

⌛ How to resolve this issue?

After the wait time has elapsed, a review can be triggered using the @coderabbitai review command as a PR comment. Alternatively, push new commits to this PR.

We recommend that you space out your commits to avoid hitting the rate limit.

🚦 How do rate limits work?

CodeRabbit enforces hourly rate limits for each developer per organization.

Our paid plans have higher rate limits than the trial, open-source and free plans. In all cases, we re-allow further reviews after a brief timeout.

Please see our FAQ for further information.

📥 Commits

Reviewing files that changed from the base of the PR and between a490cab and 80c8d5a.

📒 Files selected for processing (1)
  • tests/support/env_guard.rs (1 hunks)

Summary by CodeRabbit

  • Refactor

    • Centralised environment variable management for tests into a shared utility module, replacing local implementations across multiple test files.
  • New Features

    • Introduced a reusable utility for safely setting and restoring environment variables during tests, improving test reliability and maintainability.

Walkthrough

Refactor environment variable management in tests by extracting the EnvVarGuard utility and related helper functions into a shared module under tests/support/env_guard.rs. Update test files to import and use the new shared utilities, removing redundant local implementations and streamlining environment variable handling across the test suite.

Changes

Cohort / File(s) Change Summary
Refactor env var guard in config tests
crates/comenqd/src/config.rs
Remove local EnvVarGuard and helpers; include external env_guard module; re-export utilities via a new support module.
Introduce shared env var guard utility
tests/support/env_guard.rs
Add new module with EnvVarGuard struct, set_env_var, and remove_env_var functions for test environment management.
Declare shared support module
tests/support/mod.rs
Add new module to group shared test utilities; publicly declare env_guard submodule.
Update config step tests to use shared guard
tests/steps/config_steps.rs
Remove local guard implementation; import and use EnvVarGuard and remove_env_var from shared support module.
Import test support in cucumber tests
tests/cucumber.rs
Add mod support; to enable access to shared test utilities; no other logic changed.

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~8 minutes

Poem

In the land where tests must roam,
A guard now keeps the env at home.
No more repeats, no more dread—
Shared support stands in their stead!
With modules neat and helpers bright,
The test suite sleeps well every night.
🦀✨

✨ Finishing Touches
  • 📝 Generate Docstrings
🧪 Generate unit tests
  • Create PR with unit tests
  • Post copyable unit tests in a comment
  • Commit unit tests in branch codex/create-envvarguard-and-refactor-tests

🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

  • Review comments: Directly reply to a review comment made by CodeRabbit. Example:
    • I pushed a fix in commit <commit_id>, please review it.
    • Explain this complex logic.
    • Open a follow-up GitHub issue for this discussion.
  • Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query. Examples:
    • @coderabbitai explain this code block.
    • @coderabbitai modularize this function.
  • PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
    • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
    • @coderabbitai read src/utils.ts and explain its main purpose.
    • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.
    • @coderabbitai help me debug CodeRabbit configuration file.

Support

Need help? Create a ticket on our support page for assistance with any issues or questions.

Note: Be mindful of the bot's finite context window. It's strongly recommended to break down tasks such as reading entire modules into smaller chunks. For a focused discussion, use review comments to chat about specific files and their changes, instead of using the PR comments.

CodeRabbit Commands (Invoked using PR comments)

  • @coderabbitai pause to pause the reviews on a PR.
  • @coderabbitai resume to resume the paused reviews.
  • @coderabbitai review to trigger an incremental review. This is useful when automatic reviews are disabled for the repository.
  • @coderabbitai full review to do a full review from scratch and review all the files again.
  • @coderabbitai summary to regenerate the summary of the PR.
  • @coderabbitai generate docstrings to generate docstrings for this PR.
  • @coderabbitai generate sequence diagram to generate a sequence diagram of the changes in this PR.
  • @coderabbitai generate unit tests to generate unit tests for this PR.
  • @coderabbitai resolve resolve all the CodeRabbit review comments.
  • @coderabbitai configuration to show the current CodeRabbit configuration for the repository.
  • @coderabbitai help to get help.

Other keywords and placeholders

  • Add @coderabbitai ignore anywhere in the PR description to prevent this PR from being reviewed.
  • Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
  • Add @coderabbitai anywhere in the PR title to generate the title automatically.

CodeRabbit Configuration File (.coderabbit.yaml)

  • You can programmatically configure CodeRabbit by adding a .coderabbit.yaml file to the root of your repository.
  • Please see the configuration documentation for more information.
  • If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: # yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json

Documentation and Community

  • Visit our Documentation for detailed information on how to use CodeRabbit.
  • Join our Discord Community to get help, request features, and share feedback.
  • Follow us on X/Twitter for updates and announcements.

Copy link
Copy Markdown

@sourcery-ai sourcery-ai Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hey @leynos - I've reviewed your changes and they look great!

Prompt for AI Agents
Please address the comments from this code review:
## Individual Comments

### Comment 1
<location> `tests/support/env_guard.rs:12` </location>
<code_context>
+        ));
     }

-    impl EnvVarGuard {
-        fn set(key: &str, val: &str) -> Self {
-            let original = std::env::var(key).ok();
</code_context>

<issue_to_address>
Consider adding a test for nested EnvVarGuard usage.

Testing nested EnvVarGuards for the same key will verify correct restoration of the original value when guards overlap.

Suggested implementation:

```rust
impl Drop for EnvVarGuard {
    fn drop(&mut self) {

```

```rust
#[cfg(test)]
mod tests {
    use super::EnvVarGuard;
    use std::env;

    #[test]
    fn test_nested_env_var_guard() {
        let key = "ENV_GUARD_TEST_NESTED";
        // Ensure the variable is unset before starting
        env::remove_var(key);

        // Set an initial value
        env::set_var(key, "initial");
        assert_eq!(env::var(key).unwrap(), "initial");

        // First guard sets to "first"
        let guard1 = EnvVarGuard::set(key, "first");
        assert_eq!(env::var(key).unwrap(), "first");

        {
            // Nested guard sets to "second"
            let guard2 = EnvVarGuard::set(key, "second");
            assert_eq!(env::var(key).unwrap(), "second");
            // guard2 drops here
        }
        // After inner guard drops, should be restored to "first"
        assert_eq!(env::var(key).unwrap(), "first");

        // guard1 drops here
        drop(guard1);

        // Should be restored to "initial"
        assert_eq!(env::var(key).unwrap(), "initial");

        // Clean up
        env::remove_var(key);
    }
}

```
</issue_to_address>

### Comment 2
<location> `tests/support/env_guard.rs:34` </location>
<code_context>
+}
+
+/// Safely set an environment variable for tests.
+pub fn set_env_var(key: &str, value: &str) {
+    // Safety: tests execute serially so no concurrent access occurs.
+    unsafe { std::env::set_var(key, value) };
+}
+
</code_context>

<issue_to_address>
Missing tests for set_env_var and remove_env_var helpers.

Direct unit tests are recommended for these helpers, given their use of unsafe blocks.
</issue_to_address>

<suggested_fix>
<<<<<<< SEARCH
}
=======
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_set_env_var_sets_variable() {
        let key = "TEST_ENV_SET_VAR";
        let value = "test_value";
        // Ensure the variable is not set before
        std::env::remove_var(key);
        assert!(std::env::var(key).is_err());

        set_env_var(key, value);
        assert_eq!(std::env::var(key).unwrap(), value);

        // Clean up
        std::env::remove_var(key);
    }

    #[test]
    fn test_remove_env_var_removes_variable() {
        let key = "TEST_ENV_REMOVE_VAR";
        let value = "to_remove";
        std::env::set_var(key, value);
        assert_eq!(std::env::var(key).unwrap(), value);

        remove_env_var(key);
        assert!(std::env::var(key).is_err());
    }
}
>>>>>>> REPLACE

</suggested_fix>

### Comment 3
<location> `tests/steps/config_steps.rs:8` </location>
<code_context>
 use std::path::PathBuf;
 use tempfile::TempDir;

+use crate::support::env_guard::{EnvVarGuard, remove_env_var};
 use comenqd::config::Config;

</code_context>

<issue_to_address>
Consider adding a test for environment variable removal when not set.

Add a test to verify that calling remove_env_var on an unset key does not cause errors or unexpected behavior.
</issue_to_address>

Sourcery is free for open source - if you like our reviews please consider sharing them ✨
Help me be more useful! Please click 👍 or 👎 on each comment and I'll use the feedback to improve your reviews.

Comment thread tests/support/env_guard.rs
Comment thread tests/support/env_guard.rs
Comment thread tests/steps/config_steps.rs
@github-actions
Copy link
Copy Markdown

@coderabbitai review

@coderabbitai
Copy link
Copy Markdown
Contributor

coderabbitai Bot commented Jul 30, 2025

✅ Actions performed

Review triggered.

Note: CodeRabbit is an incremental review system and does not re-review already reviewed commits. This command is applicable only when automatic reviews are paused.

Copy link
Copy Markdown
Contributor

@coderabbitai coderabbitai Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

♻️ Duplicate comments (1)
tests/support/env_guard.rs (1)

45-103: Comprehensive test coverage addresses previous review feedback.

The test module provides excellent coverage of all scenarios including edge cases and nested guard behaviour. The use of #[serial_test::serial] properly handles test isolation for environment variable manipulation.

📜 Review details

Configuration used: CodeRabbit UI
Review profile: ASSERTIVE
Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between af818e4 and 572153e.

📒 Files selected for processing (1)
  • tests/support/env_guard.rs (1 hunks)
🧰 Additional context used
📓 Path-based instructions (1)
**/*.rs

⚙️ CodeRabbit Configuration File

**/*.rs: * Seek to keep the cyclomatic complexity of functions no more than 12.

  • Adhere to single responsibility and CQRS

  • Place function attributes after doc comments.

  • Do not use return in single-line functions.

  • Move conditionals with >2 branches into a predicate function.

  • Avoid unsafe unless absolutely necessary.

  • Every module must begin with a //! doc comment that explains the module's purpose and utility.

  • Comments and docs must follow en-GB-oxendict (-ize / -our) spelling and grammar

  • Lints must not be silenced except as a last resort.

    • #[allow] is forbidden.
    • Only narrowly scoped #[expect(lint, reason = "...")] is allowed.
    • No lint groups, no blanket or file-wide suppression.
    • Include FIXME: with link if a fix is expected.
  • Use rstest fixtures for shared setup and to avoid repetition between tests.

  • Replace duplicated tests with #[rstest(...)] parameterised cases.

  • Prefer mockall for mocks/stubs.

  • Prefer .expect() over .unwrap()

  • Ensure that any API or behavioural changes are reflected in the documentation in docs/

  • Ensure that any completed roadmap steps are recorded in the appropriate roadmap in docs/

  • Files must not exceed 400 lines in length

    • Large modules must be decomposed
    • Long match statements or dispatch tables should be decomposed by domain and collocated with targets
    • Large blocks of inline data (e.g., test fixtures, constants or templates) must be moved to external files and inlined at compile-time or loaded at run-time.

Files:

  • tests/support/env_guard.rs
🧬 Code Graph Analysis (1)
tests/support/env_guard.rs (1)
tests/steps/config_steps.rs (2)
  • set_env_var (83-85)
  • drop (123-127)
🔇 Additional comments (4)
tests/support/env_guard.rs (4)

1-4: Module documentation meets requirements.

The module documentation correctly uses //! comments and clearly explains the module's purpose and utility as required by the coding guidelines.


6-10: Well-designed RAII struct.

The EnvVarGuard struct correctly implements the RAII pattern with appropriate field types to handle both set and unset environment variables.


12-22: Correct RAII implementation.

The set method properly captures the original environment variable state before modification and constructs the guard correctly for restoration on drop.


24-31: Proper cleanup implementation.

The Drop implementation correctly restores the environment variable to its original state, handling both previously set and unset variables appropriately.

Comment thread tests/support/env_guard.rs Outdated
Copy link
Copy Markdown
Contributor

@coderabbitai coderabbitai Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

♻️ Duplicate comments (1)
tests/support/env_guard.rs (1)

33-46: Remove unnecessary unsafe blocks.

std::env::set_var and std::env::remove_var are safe functions that don't require unsafe blocks. The concurrency concerns are addressed by the #[serial_test::serial] attribute on tests, not by unsafe code.

Apply this diff to fix the unnecessary unsafe usage:

 /// Set an environment variable for tests.
-///
-/// The nightly compiler marks `std::env::set_var` as `unsafe`.
-/// Tests run serially so using it is acceptable here.
 pub fn set_env_var(key: &str, value: &str) {
-    unsafe { std::env::set_var(key, value) };
+    std::env::set_var(key, value);
 }

 /// Remove an environment variable for tests.
-///
-/// `std::env::remove_var` is also `unsafe` on nightly.
 pub fn remove_env_var(key: &str) {
-    unsafe { std::env::remove_var(key) };
+    std::env::remove_var(key);
 }
📜 Review details

Configuration used: CodeRabbit UI
Review profile: ASSERTIVE
Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between 572153e and a490cab.

📒 Files selected for processing (1)
  • tests/support/env_guard.rs (1 hunks)
🧰 Additional context used
📓 Path-based instructions (1)
**/*.rs

📄 CodeRabbit Inference Engine (AGENTS.md)

**/*.rs: Clippy warnings MUST be disallowed.
Fix any warnings emitted during tests in the code itself rather than silencing them.
Where a function is too long, extract meaningfully named helper functions adhering to separation of concerns and CQRS.
Where a function has too many parameters, group related parameters in meaningfully named structs.
Where a function is returning a large error consider using Arc to reduce the amount of data returned.
Write unit and behavioural tests for new functionality. Run both before and after making any change.
Every module must begin with a module level (//!) comment explaining the module's purpose and utility.
Document public APIs using Rustdoc comments (///) so documentation can be generated with cargo doc.
Prefer immutable data and avoid unnecessary mut bindings.
Handle errors with the Result type instead of panicking where feasible.
Avoid unsafe code unless absolutely necessary and document any usage clearly.
Place function attributes after doc comments.
Do not use return in single-line functions.
Use predicate functions for conditional criteria with more than two branches.
Lints must not be silenced except as a last resort.
Lint rule suppressions must be tightly scoped and include a clear reason.
Prefer expect over allow.
Prefer .expect() over .unwrap().
Use concat!() to combine long string literals rather than escaping newlines with a backslash.
Prefer semantic error enums. Derive std::error::Error (via the thiserror crate) for any condition the caller might inspect, retry, or map to an HTTP status.
Use an opaque error only at the app boundary. Use eyre::Report for human-readable logs; these should not be exposed in public APIs.
Never export the opaque type from a library. Convert to domain enums at API boundaries, and to eyre only in the main main() entrypoint or top-level async task.

Files:

  • tests/support/env_guard.rs

⚙️ CodeRabbit Configuration File

**/*.rs: * Seek to keep the cyclomatic complexity of functions no more than 12.

  • Adhere to single responsibility and CQRS

  • Place function attributes after doc comments.

  • Do not use return in single-line functions.

  • Move conditionals with >2 branches into a predicate function.

  • Avoid unsafe unless absolutely necessary.

  • Every module must begin with a //! doc comment that explains the module's purpose and utility.

  • Comments and docs must follow en-GB-oxendict (-ize / -our) spelling and grammar

  • Lints must not be silenced except as a last resort.

    • #[allow] is forbidden.
    • Only narrowly scoped #[expect(lint, reason = "...")] is allowed.
    • No lint groups, no blanket or file-wide suppression.
    • Include FIXME: with link if a fix is expected.
  • Use rstest fixtures for shared setup and to avoid repetition between tests.

  • Replace duplicated tests with #[rstest(...)] parameterised cases.

  • Prefer mockall for mocks/stubs.

  • Prefer .expect() over .unwrap()

  • Ensure that any API or behavioural changes are reflected in the documentation in docs/

  • Ensure that any completed roadmap steps are recorded in the appropriate roadmap in docs/

  • Files must not exceed 400 lines in length

    • Large modules must be decomposed
    • Long match statements or dispatch tables should be decomposed by domain and collocated with targets
    • Large blocks of inline data (e.g., test fixtures, constants or templates) must be moved to external files and inlined at compile-time or loaded at run-time.

Files:

  • tests/support/env_guard.rs
🧬 Code Graph Analysis (1)
tests/support/env_guard.rs (1)
tests/steps/config_steps.rs (2)
  • set_env_var (83-85)
  • drop (123-127)
🔇 Additional comments (4)
tests/support/env_guard.rs (4)

1-10: Excellent module documentation and struct design.

The module-level documentation clearly explains the purpose and utility, and the EnvVarGuard struct is well-designed with appropriate fields for tracking the key and original value.


12-22: Solid RAII implementation.

The set method correctly captures the original environment variable value before setting the new one, ensuring proper restoration on drop.


24-31: Correct restoration logic in Drop.

The Drop implementation properly restores the environment variable to its original state, handling both cases where the variable previously existed or was unset.


48-49: Proper test module structure.

The test module is correctly configured with #[cfg(test)] following Rust conventions.

Comment thread tests/support/env_guard.rs
@leynos leynos merged commit 69de73d into main Jul 31, 2025
1 check passed
@leynos leynos deleted the codex/create-envvarguard-and-refactor-tests branch July 31, 2025 19:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant