Skip to content

Commit 8b3f39a

Browse files
committed
Create test-aware db env vars.
1 parent 2079fc9 commit 8b3f39a

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

src/config.rs

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -59,13 +59,19 @@ pub(crate) fn cache_path() -> String {
5959

6060
pub(crate) fn db_connection_config() -> Config {
6161
let mut config = Config::new();
62-
let host = env::var("RAPID_GOSSIP_SYNC_SERVER_DB_HOST").unwrap_or("localhost".to_string());
63-
let user = env::var("RAPID_GOSSIP_SYNC_SERVER_DB_USER").unwrap_or("alice".to_string());
64-
let db = env::var("RAPID_GOSSIP_SYNC_SERVER_DB_NAME").unwrap_or("ln_graph_sync".to_string());
62+
let env_name_prefix = if cfg!(test) {
63+
"RAPID_GOSSIP_TEST_DB"
64+
} else {
65+
"RAPID_GOSSIP_SYNC_SERVER_DB"
66+
};
67+
68+
let host = env::var(format!("{}{}", env_name_prefix, "_HOST")).unwrap_or("localhost".to_string());
69+
let user = env::var(format!("{}{}", env_name_prefix, "_USER")).unwrap_or("alice".to_string());
70+
let db = env::var(format!("{}{}", env_name_prefix, "_NAME")).unwrap_or("ln_graph_sync".to_string());
6571
config.host(&host);
6672
config.user(&user);
6773
config.dbname(&db);
68-
if let Ok(password) = env::var("RAPID_GOSSIP_SYNC_SERVER_DB_PASSWORD") {
74+
if let Ok(password) = env::var(format!("{}{}", env_name_prefix, "_PASSWORD")) {
6975
config.password(&password);
7076
}
7177
config

0 commit comments

Comments
 (0)