only compile ApplyMode::Temporary in test mode

Author: jyn514

src/db/migrate.rs

@@ -10,6 +10,7 @@ use std::borrow::Cow;
 #[derive(Copy, Clone)]
 enum ApplyMode {
     Permanent,
+    #[cfg(test)]
     Temporary,
 }
 
@@ -22,6 +23,7 @@ impl MigrationContext {
     fn format_query<'a>(&self, query: &'a str) -> Cow<'a, str> {
         match self.apply_mode {
             ApplyMode::Permanent => Cow::Borrowed(query),
+            #[cfg(test)]
             ApplyMode::Temporary => {
                 Cow::Owned(query.replace("CREATE TABLE", "CREATE TEMPORARY TABLE"))
             }
@@ -73,6 +75,7 @@ pub fn migrate(version: Option<Version>, conn: &Connection) -> CratesfyiResult<(
     migrate_inner(version, conn, ApplyMode::Permanent)
 }
 
+#[cfg(test)]
 pub fn migrate_temporary(version: Option<Version>, conn: &Connection) -> CratesfyiResult<()> {
     migrate_inner(version, conn, ApplyMode::Temporary)
 }

src/db/mod.rs

@@ -5,6 +5,8 @@ pub(crate) use self::add_package::add_build_into_database;
 pub(crate) use self::add_package::CratesIoData;
 pub use self::file::{add_path_into_database, move_to_s3};
 pub use self::migrate::migrate;
+#[cfg(test)]
+pub(crate) use self::migrate::migrate_temporary;
 pub use self::delete_crate::delete_crate;
 
 use postgres::{Connection, TlsMode};