Inject category toml string

Author: carols10cents

src/bin/server.rs

@@ -117,7 +117,8 @@ fn main() {
     let app = cargo_registry::App::new(&config);
     let app = cargo_registry::middleware(Arc::new(app));
 
-    cargo_registry::categories::sync().unwrap();
+    let categories_toml = include_str!("../categories.toml");
+    cargo_registry::categories::sync(&categories_toml).unwrap();
 
     let port = if heroku {
         8888

src/categories.rs

@@ -98,15 +98,14 @@ struct NewCategory {
     description: String,
 }
 
-pub fn sync() -> CargoResult<()> {
+pub fn sync(toml_str: &str) -> CargoResult<()> {
     use diesel::pg::upsert::*;
     use diesel::expression::dsl::all;
 
     let conn = db::connect_now().unwrap();
 
-    let categories = include_str!("./categories.toml");
     let toml: toml::value::Table =
-        toml::from_str(categories).expect("Could not parse categories.toml");
+        toml::from_str(toml_str).expect("Could not parse categories toml");
 
     let categories = categories_from_toml(&toml, None)
         .expect("Could not convert categories from TOML")