diff options
author | Saturnino Abril <saturnino.abril@gmail.com> | 2017-03-12 07:24:44 +0900 |
---|---|---|
committer | enahum <nahumhbl@gmail.com> | 2017-03-11 19:24:44 -0300 |
commit | d32334cdfb09b25a97c4731721ce9be836f95300 (patch) | |
tree | f9da2fa15b6b70d2b27e8a98b868b0377a6b70d9 /api4/file_test.go | |
parent | ce772b87da3266f7261986ed2c9048c3d3d5ebe0 (diff) | |
download | chat-d32334cdfb09b25a97c4731721ce9be836f95300.tar.gz chat-d32334cdfb09b25a97c4731721ce9be836f95300.tar.bz2 chat-d32334cdfb09b25a97c4731721ce9be836f95300.zip |
Endpoint for APIv4: /files/{file_id}/link (#5607)
* APIv4: /files/{file_id}/link
* updated public link
Diffstat (limited to 'api4/file_test.go')
-rw-r--r-- | api4/file_test.go | 76 |
1 files changed, 76 insertions, 0 deletions
diff --git a/api4/file_test.go b/api4/file_test.go index f8f57cdb8..d6de56b7d 100644 --- a/api4/file_test.go +++ b/api4/file_test.go @@ -10,6 +10,7 @@ import ( "github.com/mattermost/platform/app" "github.com/mattermost/platform/model" + "github.com/mattermost/platform/store" "github.com/mattermost/platform/utils" ) @@ -201,3 +202,78 @@ func TestGetFileThumbnail(t *testing.T) { _, resp = th.SystemAdminClient.GetFileThumbnail(fileId) CheckNoError(t, resp) } + +func TestGetFileLink(t *testing.T) { + th := Setup().InitBasic().InitSystemAdmin() + defer TearDown() + Client := th.Client + channel := th.BasicChannel + + if utils.Cfg.FileSettings.DriverName == "" { + t.Skip("skipping because no file driver is enabled") + } + + enablePublicLink := utils.Cfg.FileSettings.EnablePublicLink + publicLinkSalt := *utils.Cfg.FileSettings.PublicLinkSalt + defer func() { + utils.Cfg.FileSettings.EnablePublicLink = enablePublicLink + *utils.Cfg.FileSettings.PublicLinkSalt = publicLinkSalt + }() + utils.Cfg.FileSettings.EnablePublicLink = true + *utils.Cfg.FileSettings.PublicLinkSalt = model.NewId() + + fileId := "" + if data, err := readTestFile("test.png"); err != nil { + t.Fatal(err) + } else { + fileResp, resp := Client.UploadFile(data, channel.Id, "test.png") + CheckNoError(t, resp) + + fileId = fileResp.FileInfos[0].Id + } + + link, resp := Client.GetFileLink(fileId) + CheckBadRequestStatus(t, resp) + + // Hacky way to assign file to a post (usually would be done by CreatePost call) + store.Must(app.Srv.Store.FileInfo().AttachToPost(fileId, th.BasicPost.Id)) + + utils.Cfg.FileSettings.EnablePublicLink = false + _, resp = Client.GetFileLink(fileId) + CheckNotImplementedStatus(t, resp) + + // Wait a bit for files to ready + time.Sleep(2 * time.Second) + + utils.Cfg.FileSettings.EnablePublicLink = true + link, resp = Client.GetFileLink(fileId) + CheckNoError(t, resp) + if link == "" { + t.Fatal("should've received public link") + } + + _, resp = Client.GetFileLink("junk") + CheckBadRequestStatus(t, resp) + + _, resp = Client.GetFileLink(model.NewId()) + CheckNotFoundStatus(t, resp) + + Client.Logout() + _, resp = Client.GetFileLink(fileId) + CheckUnauthorizedStatus(t, resp) + + otherUser := th.CreateUser() + Client.Login(otherUser.Email, otherUser.Password) + _, resp = Client.GetFileLink(fileId) + CheckForbiddenStatus(t, resp) + + Client.Logout() + _, resp = th.SystemAdminClient.GetFileLink(fileId) + CheckNoError(t, resp) + + if result := <-app.Srv.Store.FileInfo().Get(fileId); result.Err != nil { + t.Fatal(result.Err) + } else { + cleanupTestFile(result.Data.(*model.FileInfo)) + } +} |