crates/db_schema/src/impls/language.rs

   1 use crate::{newtypes::LanguageId, source::language::Language};
   2 use diesel::{result::Error, PgConnection, RunQueryDsl, *};
   3
   4 impl Language {
   5   pub fn read_all(conn: &mut PgConnection) -> Result<Vec<Language>, Error> {
   6     use crate::schema::language::dsl::*;
   7     language.load::<Self>(conn)
   8   }
   9
  10   pub fn read_from_id(conn: &mut PgConnection, id_: LanguageId) -> Result<Language, Error> {
  11     use crate::schema::language::dsl::*;
  12     language.filter(id.eq(id_)).first::<Self>(conn)
  13   }
  14
  15   pub fn read_id_from_code(conn: &mut PgConnection, code_: &str) -> Result<LanguageId, Error> {
  16     use crate::schema::language::dsl::*;
  17     Ok(language.filter(code.eq(code_)).first::<Self>(conn)?.id)
  18   }
  19
  20   pub fn read_id_from_code_opt(
  21     conn: &mut PgConnection,
  22     code_: Option<&str>,
  23   ) -> Result<Option<LanguageId>, Error> {
  24     if let Some(code_) = code_ {
  25       Ok(Some(Language::read_id_from_code(conn, code_)?))
  26     } else {
  27       Ok(None)
  28     }
  29   }
  30
  31   pub fn read_undetermined(conn: &mut PgConnection) -> Result<LanguageId, Error> {
  32     use crate::schema::language::dsl::*;
  33     Ok(language.filter(code.eq("und")).first::<Self>(conn)?.id)
  34   }
  35 }
  36
  37 #[cfg(test)]
  38 mod tests {
  39   use crate::{source::language::Language, utils::establish_unpooled_connection};
  40   use serial_test::serial;
  41
  42   #[test]
  43   #[serial]
  44   fn test_languages() {
  45     let conn = &mut establish_unpooled_connection();
  46
  47     let all = Language::read_all(conn).unwrap();
  48
  49     assert_eq!(184, all.len());
  50     assert_eq!("ak", all[5].code);
  51     assert_eq!("lv", all[99].code);
  52     assert_eq!("yi", all[179].code);
  53   }
  54 }