docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-27 18:56:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
Lev Kokotov
2023-05-03 15:42:16 -07:00
parent ff80bb75cc
commit d738ba28b6
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/plugins/intercept.rs
View File

@@ -150,6 +150,7 @@ impl Plugin for Intercept {
/// Make IntelliJ SQL plugin believe it's talking to an actual database /// Make IntelliJ SQL plugin believe it's talking to an actual database
/// instead of PgCat. /// instead of PgCat.
#[allow(dead_code)]
fn fool_datagrip(database: &str, user: &str) -> Value { fn fool_datagrip(database: &str, user: &str) -> Value {
json!([ json!([
{ {

16
src/query_router.rs
View File

@@ -1161,7 +1161,6 @@ mod test {
auth_query_password: None, auth_query_password: None,
auth_query_user: None, auth_query_user: None,
db: "test".to_string(), db: "test".to_string(),
plugins: None,
}; };
let mut qr = QueryRouter::new(); let mut qr = QueryRouter::new();
assert_eq!(qr.active_role, None); assert_eq!(qr.active_role, None);
@@ -1236,7 +1235,6 @@ mod test {
auth_query_password: None, auth_query_password: None,
auth_query_user: None, auth_query_user: None,
db: "test".to_string(), db: "test".to_string(),
plugins: None,
}; };
let mut qr = QueryRouter::new(); let mut qr = QueryRouter::new();
qr.update_pool_settings(pool_settings.clone()); qr.update_pool_settings(pool_settings.clone());
@@ -1381,13 +1379,17 @@ mod test {
#[tokio::test] #[tokio::test]
async fn test_table_access_plugin() { async fn test_table_access_plugin() {
use crate::config::TableAccess;
let ta = TableAccess {
enabled: true,
tables: vec![String::from("pg_database")],
};
crate::plugins::table_access::setup(&ta);
QueryRouter::setup(); QueryRouter::setup();
let mut qr = QueryRouter::new(); let qr = QueryRouter::new();
let mut pool_settings = PoolSettings::default();
pool_settings.plugins = Some(vec![String::from("pg_table_access")]);
qr.update_pool_settings(pool_settings);
let query = simple_query("SELECT * FROM pg_database"); let query = simple_query("SELECT * FROM pg_database");
let ast = QueryRouter::parse(&query).unwrap(); let ast = QueryRouter::parse(&query).unwrap();