diff options
author | Harrison Healey <harrisonmhealey@gmail.com> | 2016-06-14 09:38:19 -0400 |
---|---|---|
committer | Christopher Speller <crspeller@gmail.com> | 2016-06-14 07:38:19 -0600 |
commit | a0cc913b85dea5023b705697afa5cd8749a6e5de (patch) | |
tree | debe3365ea1e66e94bd0a4738bf4faa0f10eac05 /api | |
parent | 661f221727109f2298812fea89347bfeaf984109 (diff) | |
download | chat-a0cc913b85dea5023b705697afa5cd8749a6e5de.tar.gz chat-a0cc913b85dea5023b705697afa5cd8749a6e5de.tar.bz2 chat-a0cc913b85dea5023b705697afa5cd8749a6e5de.zip |
PLT-3143 Added serverside code for custom Emoji (#3311)
* Added model objects for emoji
* Added database tables for emoji
* Added settings for custom emoji
* Added serverside APIs and unit tests for custom emoji
* Added additional validation to catch duplicate emoji names earlier on
* Added additional validation to prevent users from adding emoji as another user
Diffstat (limited to 'api')
-rw-r--r-- | api/api.go | 4 | ||||
-rw-r--r-- | api/emoji.go | 252 | ||||
-rw-r--r-- | api/emoji_test.go | 445 |
3 files changed, 701 insertions, 0 deletions
diff --git a/api/api.go b/api/api.go index 3404e0c0b..37172260b 100644 --- a/api/api.go +++ b/api/api.go @@ -46,6 +46,8 @@ type Routes struct { License *mux.Router // 'api/v3/license' Public *mux.Router // 'api/v3/public' + + Emoji *mux.Router // 'api/v3/emoji' } var BaseRoutes *Routes @@ -72,6 +74,7 @@ func InitApi() { BaseRoutes.Preferences = BaseRoutes.ApiRoot.PathPrefix("/preferences").Subrouter() BaseRoutes.License = BaseRoutes.ApiRoot.PathPrefix("/license").Subrouter() BaseRoutes.Public = BaseRoutes.ApiRoot.PathPrefix("/public").Subrouter() + BaseRoutes.Emoji = BaseRoutes.ApiRoot.PathPrefix("/emoji").Subrouter() InitUser() InitTeam() @@ -86,6 +89,7 @@ func InitApi() { InitWebhook() InitPreference() InitLicense() + InitEmoji() // 404 on any api route before web.go has a chance to serve it Srv.Router.Handle("/api/{anything:.*}", http.HandlerFunc(Handle404)) diff --git a/api/emoji.go b/api/emoji.go new file mode 100644 index 000000000..24989924a --- /dev/null +++ b/api/emoji.go @@ -0,0 +1,252 @@ +// Copyright (c) 2016 Mattermost, Inc. All Rights Reserved. +// See License.txt for license information. + +package api + +import ( + "bytes" + "image" + _ "image/gif" + _ "image/jpeg" + _ "image/png" + "io" + "mime/multipart" + "net/http" + "strings" + + l4g "github.com/alecthomas/log4go" + "github.com/gorilla/mux" + "github.com/mattermost/platform/model" + "github.com/mattermost/platform/utils" +) + +const ( + MaxEmojiFileSize = 64 * 1024 // 64 KB + MaxEmojiWidth = 128 + MaxEmojiHeight = 128 +) + +func InitEmoji() { + l4g.Debug(utils.T("api.emoji.init.debug")) + + BaseRoutes.Emoji.Handle("/list", ApiUserRequired(getEmoji)).Methods("GET") + BaseRoutes.Emoji.Handle("/create", ApiUserRequired(createEmoji)).Methods("POST") + BaseRoutes.Emoji.Handle("/delete", ApiUserRequired(deleteEmoji)).Methods("POST") + BaseRoutes.Emoji.Handle("/{id:[A-Za-z0-9_]+}", ApiUserRequired(getEmojiImage)).Methods("GET") +} + +func getEmoji(c *Context, w http.ResponseWriter, r *http.Request) { + if !*utils.Cfg.ServiceSettings.EnableCustomEmoji { + c.Err = model.NewLocAppError("getEmoji", "api.emoji.disabled.app_error", nil, "") + c.Err.StatusCode = http.StatusNotImplemented + return + } + + if result := <-Srv.Store.Emoji().GetAll(); result.Err != nil { + c.Err = result.Err + return + } else { + emoji := result.Data.([]*model.Emoji) + w.Write([]byte(model.EmojiListToJson(emoji))) + } +} + +func createEmoji(c *Context, w http.ResponseWriter, r *http.Request) { + if !*utils.Cfg.ServiceSettings.EnableCustomEmoji { + c.Err = model.NewLocAppError("createEmoji", "api.emoji.disabled.app_error", nil, "") + c.Err.StatusCode = http.StatusNotImplemented + return + } + + if !(*utils.Cfg.ServiceSettings.RestrictCustomEmojiCreation == model.RESTRICT_EMOJI_CREATION_ALL || c.IsSystemAdmin()) { + c.Err = model.NewLocAppError("createEmoji", "api.emoji.create.permissions.app_error", nil, "user_id="+c.Session.UserId) + c.Err.StatusCode = http.StatusUnauthorized + return + } + + if len(utils.Cfg.FileSettings.DriverName) == 0 { + c.Err = model.NewLocAppError("createEmoji", "api.emoji.storage.app_error", nil, "") + c.Err.StatusCode = http.StatusNotImplemented + return + } + + if r.ContentLength > MaxEmojiFileSize { + c.Err = model.NewLocAppError("createEmoji", "api.emoji.create.too_large.app_error", nil, "") + c.Err.StatusCode = http.StatusRequestEntityTooLarge + return + } + + if err := r.ParseMultipartForm(MaxEmojiFileSize); err != nil { + c.Err = model.NewLocAppError("createEmoji", "api.emoji.create.parse.app_error", nil, err.Error()) + c.Err.StatusCode = http.StatusBadRequest + return + } + + m := r.MultipartForm + props := m.Value + + emoji := model.EmojiFromJson(strings.NewReader(props["emoji"][0])) + if emoji == nil { + c.SetInvalidParam("createEmoji", "emoji") + return + } + + // wipe the emoji id so that existing emojis can't get overwritten + emoji.Id = "" + + // do our best to validate the emoji before committing anything to the DB so that we don't have to clean up + // orphaned files left over when validation fails later on + emoji.PreSave() + if err := emoji.IsValid(); err != nil { + c.Err = err + c.Err.StatusCode = http.StatusBadRequest + return + } + + if emoji.CreatorId != c.Session.UserId { + c.Err = model.NewLocAppError("createEmoji", "api.emoji.create.other_user.app_error", nil, "") + c.Err.StatusCode = http.StatusUnauthorized + return + } + + if result := <-Srv.Store.Emoji().GetByName(emoji.Name); result.Err == nil && result.Data != nil { + c.Err = model.NewLocAppError("createEmoji", "api.emoji.create.duplicate.app_error", nil, "") + c.Err.StatusCode = http.StatusBadRequest + return + } + + if imageData := m.File["image"]; len(imageData) == 0 { + c.SetInvalidParam("createEmoji", "image") + return + } else if err := uploadEmojiImage(emoji.Id, imageData[0]); err != nil { + c.Err = err + return + } + + if result := <-Srv.Store.Emoji().Save(emoji); result.Err != nil { + c.Err = result.Err + return + } else { + w.Write([]byte(result.Data.(*model.Emoji).ToJson())) + } +} + +func uploadEmojiImage(id string, imageData *multipart.FileHeader) *model.AppError { + file, err := imageData.Open() + if err != nil { + return model.NewLocAppError("uploadEmojiImage", "api.emoji.upload.open.app_error", nil, "") + } + defer file.Close() + + buf := bytes.NewBuffer(nil) + io.Copy(buf, file) + + // make sure the file is an image and is within the required dimensions + if config, _, err := image.DecodeConfig(bytes.NewReader(buf.Bytes())); err != nil { + return model.NewLocAppError("uploadEmojiImage", "api.emoji.upload.image.app_error", nil, err.Error()) + } else if config.Width > MaxEmojiWidth || config.Height > MaxEmojiHeight { + return model.NewLocAppError("uploadEmojiImage", "api.emoji.upload.large_image.app_error", nil, "") + } + + if err := WriteFile(buf.Bytes(), getEmojiImagePath(id)); err != nil { + return err + } + + return nil +} + +func deleteEmoji(c *Context, w http.ResponseWriter, r *http.Request) { + if !*utils.Cfg.ServiceSettings.EnableCustomEmoji { + c.Err = model.NewLocAppError("deleteEmoji", "api.emoji.disabled.app_error", nil, "") + c.Err.StatusCode = http.StatusNotImplemented + return + } + + if len(utils.Cfg.FileSettings.DriverName) == 0 { + c.Err = model.NewLocAppError("deleteImage", "api.emoji.storage.app_error", nil, "") + c.Err.StatusCode = http.StatusNotImplemented + return + } + + props := model.MapFromJson(r.Body) + + id := props["id"] + if len(id) == 0 { + c.SetInvalidParam("deleteEmoji", "id") + return + } + + if result := <-Srv.Store.Emoji().Get(id); result.Err != nil { + c.Err = result.Err + return + } else { + if c.Session.UserId != result.Data.(*model.Emoji).CreatorId && !c.IsSystemAdmin() { + c.Err = model.NewLocAppError("deleteEmoji", "api.emoji.delete.permissions.app_error", nil, "user_id="+c.Session.UserId) + c.Err.StatusCode = http.StatusUnauthorized + return + } + } + + if err := (<-Srv.Store.Emoji().Delete(id, model.GetMillis())).Err; err != nil { + c.Err = err + return + } + + go deleteEmojiImage(id) + + ReturnStatusOK(w) +} + +func deleteEmojiImage(id string) { + if err := MoveFile(getEmojiImagePath(id), "emoji/"+id+"/image_deleted"); err != nil { + l4g.Error("Failed to rename image when deleting emoji %v", id) + } +} + +func getEmojiImage(c *Context, w http.ResponseWriter, r *http.Request) { + if !*utils.Cfg.ServiceSettings.EnableCustomEmoji { + c.Err = model.NewLocAppError("getEmojiImage", "api.emoji.disabled.app_error", nil, "") + c.Err.StatusCode = http.StatusNotImplemented + return + } + + if len(utils.Cfg.FileSettings.DriverName) == 0 { + c.Err = model.NewLocAppError("getEmojiImage", "api.emoji.storage.app_error", nil, "") + c.Err.StatusCode = http.StatusNotImplemented + return + } + + params := mux.Vars(r) + + id := params["id"] + if len(id) == 0 { + c.SetInvalidParam("getEmojiImage", "id") + return + } + + if result := <-Srv.Store.Emoji().Get(id); result.Err != nil { + c.Err = result.Err + return + } else { + var img []byte + + if data, err := ReadFile(getEmojiImagePath(id)); err != nil { + c.Err = model.NewLocAppError("getEmojiImage", "api.emoji.get_image.read.app_error", nil, err.Error()) + return + } else { + img = data + } + + if _, imageType, err := image.DecodeConfig(bytes.NewReader(img)); err != nil { + model.NewLocAppError("getEmojiImage", "api.emoji.get_image.decode.app_error", nil, err.Error()) + } else { + w.Header().Set("Content-Type", "image/"+imageType) + } + + w.Write(img) + } +} + +func getEmojiImagePath(id string) string { + return "emoji/" + id + "/image" +} diff --git a/api/emoji_test.go b/api/emoji_test.go new file mode 100644 index 000000000..26dbe9323 --- /dev/null +++ b/api/emoji_test.go @@ -0,0 +1,445 @@ +// Copyright (c) 2016 Mattermost, Inc. All Rights Reserved. +// See License.txt for license information. + +package api + +import ( + "bytes" + "image" + "image/color" + "image/gif" + "image/jpeg" + "image/png" + "testing" + "time" + + "github.com/mattermost/platform/model" + "github.com/mattermost/platform/store" + "github.com/mattermost/platform/utils" +) + +func TestGetEmoji(t *testing.T) { + th := Setup().InitBasic() + Client := th.BasicClient + + emojis := []*model.Emoji{ + { + CreatorId: model.NewId(), + Name: model.NewId(), + }, + { + CreatorId: model.NewId(), + Name: model.NewId(), + }, + { + CreatorId: model.NewId(), + Name: model.NewId(), + }, + } + + for i, emoji := range emojis { + emojis[i] = store.Must(Srv.Store.Emoji().Save(emoji)).(*model.Emoji) + } + defer func() { + for _, emoji := range emojis { + store.Must(Srv.Store.Emoji().Delete(emoji.Id, time.Now().Unix())) + } + }() + + if returnedEmojis, err := Client.ListEmoji(); err != nil { + t.Fatal(err) + } else { + for _, emoji := range emojis { + found := false + + for _, savedEmoji := range returnedEmojis { + if emoji.Id == savedEmoji.Id { + found = true + break + } + } + + if !found { + t.Fatalf("failed to get emoji with id %v", emoji.Id) + } + } + } + + deleted := &model.Emoji{ + CreatorId: model.NewId(), + Name: model.NewId(), + DeleteAt: 1, + } + deleted = store.Must(Srv.Store.Emoji().Save(deleted)).(*model.Emoji) + + if returnedEmojis, err := Client.ListEmoji(); err != nil { + t.Fatal(err) + } else { + found := false + + for _, savedEmoji := range returnedEmojis { + if deleted.Id == savedEmoji.Id { + found = true + break + } + } + + if found { + t.Fatalf("souldn't have gotten deleted emoji %v", deleted.Id) + } + } +} + +func TestCreateEmoji(t *testing.T) { + th := Setup().InitBasic().InitSystemAdmin() + Client := th.BasicClient + + EnableCustomEmoji := *utils.Cfg.ServiceSettings.EnableCustomEmoji + RestrictCustomEmojiCreation := *utils.Cfg.ServiceSettings.RestrictCustomEmojiCreation + defer func() { + *utils.Cfg.ServiceSettings.EnableCustomEmoji = EnableCustomEmoji + *utils.Cfg.ServiceSettings.RestrictCustomEmojiCreation = RestrictCustomEmojiCreation + }() + *utils.Cfg.ServiceSettings.EnableCustomEmoji = false + *utils.Cfg.ServiceSettings.RestrictCustomEmojiCreation = model.RESTRICT_EMOJI_CREATION_ALL + + emoji := &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + + // try to create an emoji when they're disabled + if _, err := Client.CreateEmoji(emoji, createTestGif(t, 10, 10), "image.gif"); err == nil { + t.Fatal("shouldn't be able to create an emoji when they're disabled") + } + + *utils.Cfg.ServiceSettings.EnableCustomEmoji = true + + // try to create a valid gif emoji when they're enabled + if emojiResult, err := Client.CreateEmoji(emoji, createTestGif(t, 10, 10), "image.gif"); err != nil { + t.Fatal(err) + } else { + emoji = emojiResult + } + + // try to create an emoji with a duplicate name + emoji2 := &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: emoji.Name, + } + if _, err := Client.CreateEmoji(emoji2, createTestGif(t, 10, 10), "image.gif"); err == nil { + t.Fatal("shouldn't be able to create an emoji with a duplicate name") + } + + Client.MustGeneric(Client.DeleteEmoji(emoji.Id)) + + // try to create a valid animated gif emoji + emoji = &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + if emojiResult, err := Client.CreateEmoji(emoji, createTestAnimatedGif(t, 10, 10, 10), "image.gif"); err != nil { + t.Fatal(err) + } else { + emoji = emojiResult + } + Client.MustGeneric(Client.DeleteEmoji(emoji.Id)) + + // try to create a valid jpeg emoji + emoji = &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + if emojiResult, err := Client.CreateEmoji(emoji, createTestJpeg(t, 10, 10), "image.jpeg"); err != nil { + t.Fatal(err) + } else { + emoji = emojiResult + } + Client.MustGeneric(Client.DeleteEmoji(emoji.Id)) + + // try to create a valid png emoji + emoji = &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + if emojiResult, err := Client.CreateEmoji(emoji, createTestPng(t, 10, 10), "image.png"); err != nil { + t.Fatal(err) + } else { + emoji = emojiResult + } + Client.MustGeneric(Client.DeleteEmoji(emoji.Id)) + + // try to create an emoji that's too wide + emoji = &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + if _, err := Client.CreateEmoji(emoji, createTestGif(t, 1000, 10), "image.gif"); err == nil { + t.Fatal("shouldn't be able to create an emoji that's too wide") + } + + // try to create an emoji that's too tall + emoji = &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + if _, err := Client.CreateEmoji(emoji, createTestGif(t, 10, 1000), "image.gif"); err == nil { + t.Fatal("shouldn't be able to create an emoji that's too tall") + } + + // try to create an emoji that's too large + emoji = &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + if _, err := Client.CreateEmoji(emoji, createTestAnimatedGif(t, 100, 100, 4000), "image.gif"); err == nil { + t.Fatal("shouldn't be able to create an emoji that's too large") + } + + // try to create an emoji with data that isn't an image + emoji = &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + if _, err := Client.CreateEmoji(emoji, make([]byte, 100, 100), "image.gif"); err == nil { + t.Fatal("shouldn't be able to create an emoji with non-image data") + } + + // try to create an emoji as another user + emoji = &model.Emoji{ + CreatorId: th.BasicUser2.Id, + Name: model.NewId(), + } + if _, err := Client.CreateEmoji(emoji, createTestGif(t, 10, 10), "image.gif"); err == nil { + t.Fatal("shouldn't be able to create an emoji as another user") + } + + *utils.Cfg.ServiceSettings.RestrictCustomEmojiCreation = model.RESTRICT_EMOJI_CREATION_ADMIN + + // try to create an emoji when only system admins are allowed to create them + emoji = &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + if _, err := Client.CreateEmoji(emoji, createTestGif(t, 10, 10), "image.gif"); err == nil { + t.Fatal("shouldn't be able to create an emoji when not a system admin") + } + + emoji = &model.Emoji{ + CreatorId: th.SystemAdminUser.Id, + Name: model.NewId(), + } + if emojiResult, err := th.SystemAdminClient.CreateEmoji(emoji, createTestPng(t, 10, 10), "image.png"); err != nil { + t.Fatal(err) + } else { + emoji = emojiResult + } + th.SystemAdminClient.MustGeneric(th.SystemAdminClient.DeleteEmoji(emoji.Id)) +} + +func TestDeleteEmoji(t *testing.T) { + th := Setup().InitBasic().InitSystemAdmin() + Client := th.BasicClient + + EnableCustomEmoji := *utils.Cfg.ServiceSettings.EnableCustomEmoji + defer func() { + *utils.Cfg.ServiceSettings.EnableCustomEmoji = EnableCustomEmoji + }() + *utils.Cfg.ServiceSettings.EnableCustomEmoji = false + + emoji1 := createTestEmoji(t, &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + }, createTestGif(t, 10, 10)) + + if _, err := Client.DeleteEmoji(emoji1.Id); err == nil { + t.Fatal("shouldn't have been able to delete an emoji when they're disabled") + } + + *utils.Cfg.ServiceSettings.EnableCustomEmoji = true + + if deleted, err := Client.DeleteEmoji(emoji1.Id); err != nil { + t.Fatal(err) + } else if !deleted { + t.Fatalf("should be able to delete your own emoji %v", emoji1.Id) + } + + if _, err := Client.DeleteEmoji(emoji1.Id); err == nil { + t.Fatal("shouldn't be able to delete an already-deleted emoji") + } + + emoji2 := createTestEmoji(t, &model.Emoji{ + CreatorId: th.BasicUser2.Id, + Name: model.NewId(), + }, createTestGif(t, 10, 10)) + + if _, err := Client.DeleteEmoji(emoji2.Id); err == nil { + t.Fatal("shouldn't be able to delete another user's emoji") + } + + if deleted, err := th.SystemAdminClient.DeleteEmoji(emoji2.Id); err != nil { + t.Fatal(err) + } else if !deleted { + t.Fatalf("system admin should be able to delete anyone's emoji %v", emoji2.Id) + } +} + +func createTestGif(t *testing.T, width int, height int) []byte { + var buffer bytes.Buffer + + if err := gif.Encode(&buffer, image.NewRGBA(image.Rect(0, 0, width, height)), nil); err != nil { + t.Fatalf("failed to create gif: %v", err.Error()) + } + + return buffer.Bytes() +} + +func createTestAnimatedGif(t *testing.T, width int, height int, frames int) []byte { + var buffer bytes.Buffer + + img := gif.GIF{ + Image: make([]*image.Paletted, frames, frames), + Delay: make([]int, frames, frames), + } + for i := 0; i < frames; i++ { + img.Image[i] = image.NewPaletted(image.Rect(0, 0, width, height), color.Palette{color.Black}) + img.Delay[i] = 0 + } + if err := gif.EncodeAll(&buffer, &img); err != nil { + t.Fatalf("failed to create animated gif: %v", err.Error()) + } + + return buffer.Bytes() +} + +func createTestJpeg(t *testing.T, width int, height int) []byte { + var buffer bytes.Buffer + + if err := jpeg.Encode(&buffer, image.NewRGBA(image.Rect(0, 0, width, height)), nil); err != nil { + t.Fatalf("failed to create jpeg: %v", err.Error()) + } + + return buffer.Bytes() +} + +func createTestPng(t *testing.T, width int, height int) []byte { + var buffer bytes.Buffer + + if err := png.Encode(&buffer, image.NewRGBA(image.Rect(0, 0, width, height))); err != nil { + t.Fatalf("failed to create png: %v", err.Error()) + } + + return buffer.Bytes() +} + +func createTestEmoji(t *testing.T, emoji *model.Emoji, imageData []byte) *model.Emoji { + emoji = store.Must(Srv.Store.Emoji().Save(emoji)).(*model.Emoji) + + if err := WriteFile(imageData, "emoji/"+emoji.Id+"/image"); err != nil { + store.Must(Srv.Store.Emoji().Delete(emoji.Id, time.Now().Unix())) + t.Fatalf("failed to write image: %v", err.Error()) + } + + return emoji +} + +func TestGetEmojiImage(t *testing.T) { + th := Setup().InitBasic() + Client := th.BasicClient + + EnableCustomEmoji := *utils.Cfg.ServiceSettings.EnableCustomEmoji + RestrictCustomEmojiCreation := *utils.Cfg.ServiceSettings.RestrictCustomEmojiCreation + defer func() { + *utils.Cfg.ServiceSettings.EnableCustomEmoji = EnableCustomEmoji + *utils.Cfg.ServiceSettings.RestrictCustomEmojiCreation = RestrictCustomEmojiCreation + }() + *utils.Cfg.ServiceSettings.EnableCustomEmoji = true + *utils.Cfg.ServiceSettings.RestrictCustomEmojiCreation = model.RESTRICT_EMOJI_CREATION_ALL + + emoji1 := &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + emoji1 = Client.MustGeneric(Client.CreateEmoji(emoji1, createTestGif(t, 10, 10), "image.gif")).(*model.Emoji) + defer func() { Client.MustGeneric(Client.DeleteEmoji(emoji1.Id)) }() + + *utils.Cfg.ServiceSettings.EnableCustomEmoji = false + + if _, err := Client.DoApiGet(Client.GetCustomEmojiImageUrl(emoji1.Id), "", ""); err == nil { + t.Fatal("should've failed to get emoji image when disabled") + } + + *utils.Cfg.ServiceSettings.EnableCustomEmoji = true + + if resp, err := Client.DoApiGet(Client.GetCustomEmojiImageUrl(emoji1.Id), "", ""); err != nil { + t.Fatal(err) + } else if resp.Header.Get("Content-Type") != "image/gif" { + t.Fatal("should've received a gif") + } else if _, imageType, err := image.DecodeConfig(resp.Body); err != nil { + t.Fatalf("unable to identify received image: %v", err.Error()) + } else if imageType != "gif" { + t.Fatal("should've received gif data") + } + + emoji2 := &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + emoji2 = Client.MustGeneric(Client.CreateEmoji(emoji2, createTestAnimatedGif(t, 10, 10, 10), "image.gif")).(*model.Emoji) + defer func() { Client.MustGeneric(Client.DeleteEmoji(emoji2.Id)) }() + + if resp, err := Client.DoApiGet(Client.GetCustomEmojiImageUrl(emoji2.Id), "", ""); err != nil { + t.Fatal(err) + } else if resp.Header.Get("Content-Type") != "image/gif" { + t.Fatal("should've received a gif") + } else if _, imageType, err := image.DecodeConfig(resp.Body); err != nil { + t.Fatalf("unable to identify received image: %v", err.Error()) + } else if imageType != "gif" { + t.Fatal("should've received gif data") + } + + emoji3 := &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + emoji3 = Client.MustGeneric(Client.CreateEmoji(emoji3, createTestJpeg(t, 10, 10), "image.jpeg")).(*model.Emoji) + defer func() { Client.MustGeneric(Client.DeleteEmoji(emoji3.Id)) }() + + if resp, err := Client.DoApiGet(Client.GetCustomEmojiImageUrl(emoji3.Id), "", ""); err != nil { + t.Fatal(err) + } else if resp.Header.Get("Content-Type") != "image/jpeg" { + t.Fatal("should've received a jpeg") + } else if _, imageType, err := image.DecodeConfig(resp.Body); err != nil { + t.Fatalf("unable to identify received image: %v", err.Error()) + } else if imageType != "jpeg" { + t.Fatal("should've received jpeg data") + } + + emoji4 := &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + emoji4 = Client.MustGeneric(Client.CreateEmoji(emoji4, createTestPng(t, 10, 10), "image.png")).(*model.Emoji) + defer func() { Client.MustGeneric(Client.DeleteEmoji(emoji4.Id)) }() + + if resp, err := Client.DoApiGet(Client.GetCustomEmojiImageUrl(emoji4.Id), "", ""); err != nil { + t.Fatal(err) + } else if resp.Header.Get("Content-Type") != "image/png" { + t.Fatal("should've received a png") + } else if _, imageType, err := image.DecodeConfig(resp.Body); err != nil { + t.Fatalf("unable to identify received image: %v", err.Error()) + } else if imageType != "png" { + t.Fatal("should've received png data") + } + + emoji5 := &model.Emoji{ + CreatorId: th.BasicUser.Id, + Name: model.NewId(), + } + emoji5 = Client.MustGeneric(Client.CreateEmoji(emoji5, createTestPng(t, 10, 10), "image.png")).(*model.Emoji) + Client.MustGeneric(Client.DeleteEmoji(emoji5.Id)) + + if _, err := Client.DoApiGet(Client.GetCustomEmojiImageUrl(emoji5.Id), "", ""); err == nil { + t.Fatal("should've failed to get image for deleted emoji") + } +} |