fix: use the same timestamps in new/0 function

Author: zookzook

lib/mongo/collection.ex

@@ -545,8 +545,16 @@ defmodule Mongo.Collection do
           |> Enum.map(fn {name, opts} -> {name, opts[:default]} end)
           |> Enum.filter(fn {_name, fun} -> is_function(fun) end)
 
-        def new() do
-          %__MODULE__{unquote_splicing(Collection.struct_args(args))}
+        case @timestamps != [] do
+          true ->
+            def new() do
+              new_timestamps(%__MODULE__{unquote_splicing(Collection.struct_args(args))})
+            end
+
+          false ->
+            def new() do
+              %__MODULE__{unquote_splicing(Collection.struct_args(args))}
+            end
         end
       end
 
@@ -669,6 +677,17 @@ defmodule Mongo.Collection do
           updated_at = @timestamps[:updated_at]
           Collection.timestamps(struct, updated_at, @attributes[updated_at])
         end
+
+        def new_timestamps(struct) do
+          inserted_at = @timestamps[:inserted_at]
+          opts = @attributes[inserted_at]
+          ts = opts[:default].()
+          updated_at = @timestamps[:updated_at]
+
+          struct
+          |> Map.put(inserted_at, ts)
+          |> Map.put(updated_at, ts)
+        end
       end
 
     quote do

mix.exs

@@ -2,11 +2,12 @@ defmodule Mongodb.Mixfile do
   use Mix.Project
 
   @source_url "https://github.com/zookzook/elixir-mongodb-driver"
+  @version "0.9.2"
 
   def project() do
     [
       app: :mongodb_driver,
-      version: "0.9.2",
+      version: @version,
       elixirc_paths: elixirc_paths(Mix.env()),
       elixir: "~> 1.8",
       name: "mongodb-driver",
@@ -53,7 +54,7 @@ defmodule Mongodb.Mixfile do
       ],
       main: "readme",
       source_url: @source_url,
-      source_ref: "#{project()[:version]}",
+      source_ref: @version,
       formatters: ["html"]
     ]
   end

test/collections/simple_test.exs

@@ -38,7 +38,7 @@ defmodule Collections.SimpleTest do
     collection "cards" do
       attribute :title, String.t(), default: "new title"
       embeds_one(:label, Label, default: &Label.new/0)
-      timestamps(inserted_at: :created, updated_at: :modified)
+      timestamps(inserted_at: :created, updated_at: :modified, default: &Card.ts/0)
     end
 
     def insert_one(%Card{} = card, top) do
@@ -53,6 +53,22 @@ defmodule Collections.SimpleTest do
       |> Mongo.find_one(@collection, %{@id => id})
       |> load()
     end
+
+    def ts() do
+      Process.sleep(100)
+      DateTime.utc_now()
+    end
+  end
+
+  test "timestamps", _c do
+    alias Collections.SimpleTest.Card
+    alias Collections.SimpleTest.Label
+
+    new_card = Card.new()
+    map_card = Card.dump(new_card)
+
+    ts = Map.get(new_card, :created)
+    assert %{created: ^ts, modified: ^ts} = map_card
   end
 
   test "load and dump", _c do