วิธีเทสโค้ดที่ใช้ Time.Now

package customer

type Customer struct {
        UpdatedAt time.Time
}

func (c *Customer) Save() {
        c.UpdatedAt = time.Now()
}

package customer

func TestSaveCustomer(t *testing.T) {
        var c Customer
        c.Save()

        if c.UpdatedAt.Equal(__??__) {
                t.Error("UpdatedAt is not correct")
        }
}

package customer

var timeNow = time.Now // ไม่ใส่ () นะเพราะเราจะเก็บตัวฟังก์ชัน ไม่ได้เรียกฟังก์ชันตรงนี้

type Customer struct {
        UpdatedAt time.Time
}

func (c *Customer) Save() {
        c.UpdatedAt = timeNow() // เรียกฟังก์ชันผ่านตัวแปรที่เราเก็บฟังก์ชันเอาไว้แทน
}

package customer

func TestSaveCustomer(t *testing.T) {
        now := time.Now() // เก็บค่า time.Now() ตอนเทสเอาไว้ไปเช็คได้อีกที
        timeNow = func() time.Time { // เปลี่ยน timeNow เป็นฟังก์ชันใหม่ที่ตอบกลับเป็นค่า now ที่เราเก็บเอาไว้ก่อน
                return now
        }
        defer func() { timeNow = time.Now } // เปลี่ยนกลับเมื่อเทสฟังก์ชันทำงานจบ

        var c Customer
        c.Save()

        if !c.UpdatedAt.Equal(now) { // เอามาไว้เช็คตรงนี้ได้แล้ว
                t.Error("UpdatedAt is not correct")
        }
}