From 7eed1703c50e58ad64f2eee2b361712987af8cec Mon Sep 17 00:00:00 2001 From: Jannis Mattheis Date: Sat, 3 Feb 2018 16:48:14 +0100 Subject: [PATCH] Add unpatching the monkey (: --- api/message_test.go | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/api/message_test.go b/api/message_test.go index f6967c5..c06565d 100644 --- a/api/message_test.go +++ b/api/message_test.go @@ -153,7 +153,8 @@ func (s *MessageSuite) Test_CreateMessage_onJson_allParams() { auth.RegisterAuthentication(s.ctx, nil, 4, "app-token") t, _ := time.Parse("2006/01/02", "2017/01/02") - monkey.Patch(time.Now, func() time.Time { return t }) + patch := monkey.Patch(time.Now, func() time.Time { return t }) + defer patch.Unpatch() expected := &model.Message{ID: 0, ApplicationID: "app-token", Title: "mytitle", Message: "mymessage", Priority: 1, Date: t} s.ctx.Request = httptest.NewRequest("POST", "/token", strings.NewReader(`{"title": "mytitle", "message": "mymessage", "priority": 1}`)) @@ -170,7 +171,8 @@ func (s *MessageSuite) Test_CreateMessage_onlyRequired() { auth.RegisterAuthentication(s.ctx, nil, 4, "app-token") t, _ := time.Parse("2006/01/02", "2017/01/02") - monkey.Patch(time.Now, func() time.Time { return t }) + patch := monkey.Patch(time.Now, func() time.Time { return t }) + defer patch.Unpatch() expected := &model.Message{ID: 0, ApplicationID: "app-token", Title: "mytitle", Message: "mymessage", Date: t} s.ctx.Request = httptest.NewRequest("POST", "/token", strings.NewReader(`{"title": "mytitle", "message": "mymessage"}`)) @@ -223,7 +225,8 @@ func (s *MessageSuite) Test_CreateMessage_onQueryData() { auth.RegisterAuthentication(s.ctx, nil, 4, "app-token") t, _ := time.Parse("2006/01/02", "2017/01/02") - monkey.Patch(time.Now, func() time.Time { return t }) + patch := monkey.Patch(time.Now, func() time.Time { return t }) + defer patch.Unpatch() expected := &model.Message{ID: 0, ApplicationID: "app-token", Title: "mytitle", Message: "mymessage", Priority: 1, Date: t} s.ctx.Request = httptest.NewRequest("POST", "/token?title=mytitle&message=mymessage&priority=1", nil) @@ -240,7 +243,8 @@ func (s *MessageSuite) Test_CreateMessage_onFormData() { auth.RegisterAuthentication(s.ctx, nil, 4, "app-token") t, _ := time.Parse("2006/01/02", "2017/01/02") - monkey.Patch(time.Now, func() time.Time { return t }) + patch := monkey.Patch(time.Now, func() time.Time { return t }) + defer patch.Unpatch() expected := &model.Message{ID: 0, ApplicationID: "app-token", Title: "mytitle", Message: "mymessage", Priority: 1, Date: t} s.ctx.Request = httptest.NewRequest("POST", "/token", strings.NewReader("title=mytitle&message=mymessage&priority=1"))