chore: formatted and pub(crate) visibility

Author: Harshdev098

rust/impls/src/migrations.rs

@@ -5,7 +5,7 @@ pub(crate) const MIGRATION_LOG_COLUMN: &str = "upgrade_from";
 pub(crate) const CHECK_DB_STMT: &str = "SELECT 1 FROM pg_database WHERE datname = $1";
 pub(crate) const INIT_DB_CMD: &str = "CREATE DATABASE";
 #[cfg(test)]
-const DROP_DB_CMD: &str = "DROP DATABASE";
+pub(crate) const DROP_DB_CMD: &str = "DROP DATABASE";
 pub(crate) const GET_VERSION_STMT: &str = "SELECT db_version FROM vss_db_version;";
 pub(crate) const UPDATE_VERSION_STMT: &str = "UPDATE vss_db_version SET db_version=$1;";
 pub(crate) const LOG_MIGRATION_STMT: &str = "INSERT INTO vss_db_upgrades VALUES($1);";
@@ -36,4 +36,4 @@ pub(crate) const MIGRATIONS: &[&str] = &[
 	);",
 ];
 #[cfg(test)]
-const DUMMY_MIGRATION: &str = "SELECT 1 WHERE FALSE;";
+pub(crate) const DUMMY_MIGRATION: &str = "SELECT 1 WHERE FALSE;";

rust/impls/src/postgres_store.rs

@@ -93,10 +93,7 @@ async fn drop_database(postgres_endpoint: &str, db_name: &str) -> Result<(), Err
 
 	let drop_database_statement = format!("{} {};", DROP_DB_CMD, db_name);
 	let num_rows = client.execute(&drop_database_statement, &[]).await.map_err(|e| {
-		Error::new(
-			ErrorKind::Other,
-			format!("Failed to drop database {}: {}", db_name, e),
-		)
+		Error::new(ErrorKind::Other, format!("Failed to drop database {}: {}", db_name, e))
 	})?;
 	assert_eq!(num_rows, 0);
 
@@ -134,10 +131,7 @@ impl PostgresBackendImpl {
 
 	async fn migrate_vss_database(&self, migrations: &[&str]) -> Result<(usize, usize), Error> {
 		let mut conn = self.pool.get().await.map_err(|e| {
-			Error::new(
-				ErrorKind::Other,
-				format!("Failed to fetch a connection from Pool: {}", e),
-			)
+			Error::new(ErrorKind::Other, format!("Failed to fetch a connection from Pool: {}", e))
 		})?;
 
 		// Get the next migration to be applied.
@@ -230,7 +224,9 @@ impl PostgresBackendImpl {
 	async fn get_upgrades_list(&self) -> Vec<usize> {
 		let conn = self.pool.get().await.unwrap();
 		let rows = conn.query(GET_MIGRATION_LOG_STMT, &[]).await.unwrap();
-		rows.iter().map(|row| usize::try_from(row.get::<&str, i32>(MIGRATION_LOG_COLUMN)).unwrap()).collect()
+		rows.iter()
+			.map(|row| usize::try_from(row.get::<&str, i32>(MIGRATION_LOG_COLUMN)).unwrap())
+			.collect()
 	}
 
 	fn build_vss_record(&self, user_token: String, store_id: String, kv: KeyValue) -> VssDbRecord {
@@ -581,10 +577,10 @@ impl KvStore for PostgresBackendImpl {
 
 #[cfg(test)]
 mod tests {
+	use super::{drop_database, DUMMY_MIGRATION, MIGRATIONS};
 	use crate::postgres_store::PostgresBackendImpl;
 	use api::define_kv_store_tests;
 	use tokio::sync::OnceCell;
-	use super::{MIGRATIONS, DUMMY_MIGRATION, drop_database};
 
 	const POSTGRES_ENDPOINT: &str = "postgresql://postgres:postgres@localhost:5432";
 	const MIGRATIONS_START: usize = 0;
@@ -670,7 +666,10 @@ mod tests {
 			let (start, end) = store.migrate_vss_database(&migrations).await.unwrap();
 			assert_eq!(start, MIGRATIONS_END + 1);
 			assert_eq!(end, MIGRATIONS_END + 3);
-			assert_eq!(store.get_upgrades_list().await, [MIGRATIONS_START, MIGRATIONS_END, MIGRATIONS_END + 1]);
+			assert_eq!(
+				store.get_upgrades_list().await,
+				[MIGRATIONS_START, MIGRATIONS_END, MIGRATIONS_END + 1]
+			);
 			assert_eq!(store.get_schema_version().await, MIGRATIONS_END + 3);
 		};