Convert tests in "app/plugin_hooks_test.go" to use assert/require (#12061)

* change t.Ftal to require

* remove t.errorf

* fix build error

* adjust for custom error

* fix TestUserWillLogIn_Blocked

* address pr comments
Этот коммит содержится в:
Arshdeep Singh Chimni
2019-09-06 16:54:38 +05:30
коммит произвёл Miguel de la Cruz
родитель 1a9a2b1430
Коммит adb1a98760

Просмотреть файл

@@ -11,7 +11,6 @@ import (
"net/http/httptest" "net/http/httptest"
"os" "os"
"path/filepath" "path/filepath"
"strings"
"testing" "testing"
"time" "time"
@@ -172,9 +171,8 @@ func TestHookMessageWillBePosted(t *testing.T) {
CreateAt: model.GetMillis() - 10000, CreateAt: model.GetMillis() - 10000,
} }
post, err := th.App.CreatePost(post, th.BasicChannel, false) post, err := th.App.CreatePost(post, th.BasicChannel, false)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
assert.Equal(t, "message", post.Message) assert.Equal(t, "message", post.Message)
retrievedPost, errSingle := th.App.Srv.Store.Post().GetSingle(post.Id) retrievedPost, errSingle := th.App.Srv.Store.Post().GetSingle(post.Id)
require.Nil(t, errSingle) require.Nil(t, errSingle)
@@ -217,15 +215,12 @@ func TestHookMessageWillBePosted(t *testing.T) {
CreateAt: model.GetMillis() - 10000, CreateAt: model.GetMillis() - 10000,
} }
post, err := th.App.CreatePost(post, th.BasicChannel, false) post, err := th.App.CreatePost(post, th.BasicChannel, false)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
assert.Equal(t, "message_fromplugin", post.Message) assert.Equal(t, "message_fromplugin", post.Message)
if retrievedPost, errSingle := th.App.Srv.Store.Post().GetSingle(post.Id); err != nil { retrievedPost, errSingle := th.App.Srv.Store.Post().GetSingle(post.Id)
t.Fatal(errSingle) require.Nil(t, errSingle)
} else { assert.Equal(t, "message_fromplugin", retrievedPost.Message)
assert.Equal(t, "message_fromplugin", retrievedPost.Message)
}
}) })
t.Run("multiple updated", func(t *testing.T) { t.Run("multiple updated", func(t *testing.T) {
@@ -286,9 +281,7 @@ func TestHookMessageWillBePosted(t *testing.T) {
CreateAt: model.GetMillis() - 10000, CreateAt: model.GetMillis() - 10000,
} }
post, err := th.App.CreatePost(post, th.BasicChannel, false) post, err := th.App.CreatePost(post, th.BasicChannel, false)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
assert.Equal(t, "prefix_message_suffix", post.Message) assert.Equal(t, "prefix_message_suffix", post.Message)
}) })
} }
@@ -332,9 +325,7 @@ func TestHookMessageHasBeenPosted(t *testing.T) {
CreateAt: model.GetMillis() - 10000, CreateAt: model.GetMillis() - 10000,
} }
_, err := th.App.CreatePost(post, th.BasicChannel, false) _, err := th.App.CreatePost(post, th.BasicChannel, false)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
} }
func TestHookMessageWillBeUpdated(t *testing.T) { func TestHookMessageWillBeUpdated(t *testing.T) {
@@ -373,15 +364,11 @@ func TestHookMessageWillBeUpdated(t *testing.T) {
CreateAt: model.GetMillis() - 10000, CreateAt: model.GetMillis() - 10000,
} }
post, err := th.App.CreatePost(post, th.BasicChannel, false) post, err := th.App.CreatePost(post, th.BasicChannel, false)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
assert.Equal(t, "message_", post.Message) assert.Equal(t, "message_", post.Message)
post.Message = post.Message + "edited_" post.Message = post.Message + "edited_"
post, err = th.App.UpdatePost(post, true) post, err = th.App.UpdatePost(post, true)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
assert.Equal(t, "message_edited_fromplugin", post.Message) assert.Equal(t, "message_edited_fromplugin", post.Message)
} }
@@ -425,15 +412,11 @@ func TestHookMessageHasBeenUpdated(t *testing.T) {
CreateAt: model.GetMillis() - 10000, CreateAt: model.GetMillis() - 10000,
} }
post, err := th.App.CreatePost(post, th.BasicChannel, false) post, err := th.App.CreatePost(post, th.BasicChannel, false)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
assert.Equal(t, "message_", post.Message) assert.Equal(t, "message_", post.Message)
post.Message = post.Message + "edited" post.Message = post.Message + "edited"
_, err = th.App.UpdatePost(post, true) _, err = th.App.UpdatePost(post, true)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
} }
func TestHookFileWillBeUploaded(t *testing.T) { func TestHookFileWillBeUploaded(t *testing.T) {
@@ -582,8 +565,8 @@ func TestHookFileWillBeUploaded(t *testing.T) {
assert.Nil(t, err) assert.Nil(t, err)
assert.NotNil(t, response) assert.NotNil(t, response)
assert.Equal(t, 1, len(response.FileInfos)) assert.Equal(t, 1, len(response.FileInfos))
fileId := response.FileInfos[0].Id fileId := response.FileInfos[0].Id
fileInfo, err := th.App.GetFileInfo(fileId) fileInfo, err := th.App.GetFileInfo(fileId)
assert.Nil(t, err) assert.Nil(t, err)
assert.NotNil(t, fileInfo) assert.NotNil(t, fileInfo)
@@ -678,11 +661,7 @@ func TestUserWillLogIn_Blocked(t *testing.T) {
defer th.TearDown() defer th.TearDown()
err := th.App.UpdatePassword(th.BasicUser, "hunter2") err := th.App.UpdatePassword(th.BasicUser, "hunter2")
assert.Nil(t, err, "Error updating user password: %s", err)
if err != nil {
t.Errorf("Error updating user password: %s", err)
}
tearDown, _, _ := SetAppEnvironmentWithPlugins(t, tearDown, _, _ := SetAppEnvironmentWithPlugins(t,
[]string{ []string{
` `
@@ -711,9 +690,7 @@ func TestUserWillLogIn_Blocked(t *testing.T) {
w := httptest.NewRecorder() w := httptest.NewRecorder()
_, err = th.App.DoLogin(w, r, th.BasicUser, "") _, err = th.App.DoLogin(w, r, th.BasicUser, "")
if !strings.HasPrefix(err.Id, "Login rejected by plugin") { assert.Contains(t, err.Id, "Login rejected by plugin", "Expected Login rejected by plugin, got %s", err.Id)
t.Errorf("Expected Login rejected by plugin, got %s", err.Id)
}
} }
func TestUserWillLogInIn_Passed(t *testing.T) { func TestUserWillLogInIn_Passed(t *testing.T) {
@@ -722,9 +699,7 @@ func TestUserWillLogInIn_Passed(t *testing.T) {
err := th.App.UpdatePassword(th.BasicUser, "hunter2") err := th.App.UpdatePassword(th.BasicUser, "hunter2")
if err != nil { assert.Nil(t, err, "Error updating user password: %s", err)
t.Errorf("Error updating user password: %s", err)
}
tearDown, _, _ := SetAppEnvironmentWithPlugins(t, tearDown, _, _ := SetAppEnvironmentWithPlugins(t,
[]string{ []string{
@@ -754,13 +729,8 @@ func TestUserWillLogInIn_Passed(t *testing.T) {
w := httptest.NewRecorder() w := httptest.NewRecorder()
session, err := th.App.DoLogin(w, r, th.BasicUser, "") session, err := th.App.DoLogin(w, r, th.BasicUser, "")
if err != nil { assert.Nil(t, err, "Expected nil, got %s", err)
t.Errorf("Expected nil, got %s", err) assert.Equal(t, session.UserId, th.BasicUser.Id)
}
if session.UserId != th.BasicUser.Id {
t.Errorf("Expected %s, got %s", th.BasicUser.Id, session.UserId)
}
} }
func TestUserHasLoggedIn(t *testing.T) { func TestUserHasLoggedIn(t *testing.T) {
@@ -769,9 +739,7 @@ func TestUserHasLoggedIn(t *testing.T) {
err := th.App.UpdatePassword(th.BasicUser, "hunter2") err := th.App.UpdatePassword(th.BasicUser, "hunter2")
if err != nil { assert.Nil(t, err, "Error updating user password: %s", err)
t.Errorf("Error updating user password: %s", err)
}
tearDown, _, _ := SetAppEnvironmentWithPlugins(t, tearDown, _, _ := SetAppEnvironmentWithPlugins(t,
[]string{ []string{
@@ -802,17 +770,13 @@ func TestUserHasLoggedIn(t *testing.T) {
w := httptest.NewRecorder() w := httptest.NewRecorder()
_, err = th.App.DoLogin(w, r, th.BasicUser, "") _, err = th.App.DoLogin(w, r, th.BasicUser, "")
if err != nil { assert.Nil(t, err, "Expected nil, got %s", err)
t.Errorf("Expected nil, got %s", err)
}
time.Sleep(2 * time.Second) time.Sleep(2 * time.Second)
user, _ := th.App.GetUser(th.BasicUser.Id) user, _ := th.App.GetUser(th.BasicUser.Id)
if user.FirstName != "plugin-callback-success" { assert.Equal(t, user.FirstName, "plugin-callback-success", "Expected firstname overwrite, got default")
t.Errorf("Expected firstname overwrite, got default")
}
} }
func TestUserHasBeenCreated(t *testing.T) { func TestUserHasBeenCreated(t *testing.T) {
@@ -858,7 +822,6 @@ func TestUserHasBeenCreated(t *testing.T) {
user, err = th.App.GetUser(user.Id) user, err = th.App.GetUser(user.Id)
require.Nil(t, err) require.Nil(t, err)
require.Equal(t, "plugin-callback-success", user.Nickname) require.Equal(t, "plugin-callback-success", user.Nickname)
} }
@@ -988,7 +951,5 @@ func TestHookContext(t *testing.T) {
CreateAt: model.GetMillis() - 10000, CreateAt: model.GetMillis() - 10000,
} }
_, err := th.App.CreatePost(post, th.BasicChannel, false) _, err := th.App.CreatePost(post, th.BasicChannel, false)
if err != nil { require.Nil(t, err)
t.Fatal(err)
}
} }