use chrono instead of time in tests

1 files changed, 62 insertions, 20 deletions

diff --git a/tosin-macros/src/lib.rs b/tosin-macros/src/lib.rs
index 131888c..15a0266 100644
--- a/tosin-macros/src/lib.rs
+++ b/tosin-macros/src/lib.rs
@@ -31,7 +31,7 @@ struct FieldsInfo<T: quote::ToTokens> {
 }
 
 impl<T: quote::ToTokens> std::iter::FromIterator<FieldInfo<T>> for FieldsInfo<T> {
-    fn from_iter<I: IntoIterator<Item=FieldInfo<T>>>(iter: I) -> Self {
+    fn from_iter<I: IntoIterator<Item = FieldInfo<T>>>(iter: I) -> Self {
         let mut result = Self {
             tosin_fields: vec![],
             diesel_columns: vec![],
@@ -52,9 +52,17 @@ fn to_field_spec(field: &syn::Field) -> FieldInfo<impl quote::ToTokens> {
     }
     let field_name = &field.ident;
     let field_type = &field.ty;
-    let model_options: HashMap<syn::Path, syn::Lit> = field.attrs.iter()
+    let model_options: HashMap<syn::Path, syn::Lit> = field
+        .attrs
+        .iter()
         .filter_map(|attr| attr.parse_meta().ok())
-        .filter_map(|meta| if let syn::Meta::List(meta) = meta { Some(meta) } else { None })
+        .filter_map(|meta| {
+            if let syn::Meta::List(meta) = meta {
+                Some(meta)
+            } else {
+                None
+            }
+        })
         .filter(|meta| meta.path.get_ident().map_or(false, |path| path == "model"))
         .flat_map(|model| {
             model.nested.into_iter().filter_map(|item| {
@@ -87,27 +95,28 @@ fn to_field_spec(field: &syn::Field) -> FieldInfo<impl quote::ToTokens> {
             diesel_type: quote! { ::tosin::db::sql_types::Integer },
         }
     } else if field_type == &parse_type("String") {
-        let max_length = model_options.iter()
+        let max_length = model_options
+            .iter()
             .find(|(name, _value)| name.get_ident().map_or(false, |path| path == "max_length"))
             .map(|(_name, value)| value);
         if let Some(max_length) = max_length {
             FieldInfo {
                 tosin_field: quote! { ::tosin::db::models::Field::CharField { name: stringify!(#field_name), max_length: Some(#max_length) } },
                 diesel_column: quote! { #field_name -> Text },
-                diesel_type: quote! { ::tosin::db::sql_types::Text }
+                diesel_type: quote! { ::tosin::db::sql_types::Text },
             }
         } else {
             FieldInfo {
                 tosin_field: quote! { ::tosin::db::models::Field::CharField { name: stringify!(#field_name), max_length: None } },
                 diesel_column: quote! { #field_name -> Text },
-                diesel_type: quote! { ::tosin::db::sql_types::Text }
+                diesel_type: quote! { ::tosin::db::sql_types::Text },
             }
         }
-    } else if field_type == &parse_type("time::PrimitiveDateTime") {
+    } else if field_type == &parse_type("chrono::NaiveDateTime") {
         FieldInfo {
             tosin_field: quote! { ::tosin::db::models::Field::DateTimeField { name: stringify!(#field_name) } },
             diesel_column: quote! { #field_name -> Timestamp },
-            diesel_type: quote! { ::tosin::db::sql_types::Timestamp }
+            diesel_type: quote! { ::tosin::db::sql_types::Timestamp },
         }
     } else {
         use quote::ToTokens;
@@ -124,7 +133,11 @@ fn impl_model(ast: &syn::DeriveInput) -> TokenStream {
         panic!("not on a struct");
     };
     let real_types: Vec<_> = ast_data.fields.iter().map(|field| &field.ty).collect();
-    let FieldsInfo { tosin_fields, diesel_columns, diesel_types } = ast_data.fields.iter().map(to_field_spec).collect();
+    let FieldsInfo {
+        tosin_fields,
+        diesel_columns,
+        diesel_types,
+    } = ast_data.fields.iter().map(to_field_spec).collect();
     let gen = quote! {
         impl #name {
             pub const META: ::tosin::db::models::ModelMeta = ::tosin::db::models::ModelMeta {
@@ -163,7 +176,8 @@ pub fn gather_migrations(_input: TokenStream) -> TokenStream {
     }
 
     let migrations_dir = call_site_path.parent().unwrap();
-    let migrations: Vec<syn::Ident> = migrations_dir.read_dir()
+    let migrations: Vec<syn::Ident> = migrations_dir
+        .read_dir()
         .unwrap()
         .map(Result::unwrap)
         .map(|x| x.path().file_stem().unwrap().to_string_lossy().into_owned())
@@ -192,18 +206,46 @@ pub fn gather_models(_input: TokenStream) -> TokenStream {
     }
 
     let call_site_ast = syn::parse_file(&fs::read_to_string(call_site_path).unwrap()).unwrap();
-    let models = call_site_ast.items.iter()
-        .filter_map(|item| if let syn::Item::Struct(item) = item { Some(item) } else { None })
-        .filter(|item| item.attrs.iter().any(|attr| {
-            let attr = if let Ok(syn::Meta::List(attr)) = attr.parse_meta() { attr } else { return false; };
-            if attr.path.get_ident().map_or(false, |hopefully_derive| hopefully_derive == "derive") {
-                let mut derived = attr.nested.iter()
-                    .filter_map(|derived| if let syn::NestedMeta::Meta(derived) = derived { Some(derived) } else { None });
-                derived.any(|derived| derived.path().get_ident().map_or(false, |hopefully_model| hopefully_model == "Model"))
+    let models = call_site_ast
+        .items
+        .iter()
+        .filter_map(|item| {
+            if let syn::Item::Struct(item) = item {
+                Some(item)
             } else {
-                false
+                None
             }
-        }))
+        })
+        .filter(|item| {
+            item.attrs.iter().any(|attr| {
+                let attr = if let Ok(syn::Meta::List(attr)) = attr.parse_meta() {
+                    attr
+                } else {
+                    return false;
+                };
+                if attr
+                    .path
+                    .get_ident()
+                    .map_or(false, |hopefully_derive| hopefully_derive == "derive")
+                {
+                    let mut derived = attr.nested.iter().filter_map(|derived| {
+                        if let syn::NestedMeta::Meta(derived) = derived {
+                            Some(derived)
+                        } else {
+                            None
+                        }
+                    });
+                    derived.any(|derived| {
+                        derived
+                            .path()
+                            .get_ident()
+                            .map_or(false, |hopefully_model| hopefully_model == "Model")
+                    })
+                } else {
+                    false
+                }
+            })
+        })
         .map(|item| &item.ident);
 
     let gen = quote! {