From ace228c4e52bd25dca24d1a5b35eff97740e5ea2 Mon Sep 17 00:00:00 2001
From: Joram Wilander <jwawilander@gmail.com>
Date: Fri, 24 Feb 2017 06:15:13 -0500
Subject: Implement GET /posts/{post_id}/files/info endpoint for APIv4 (#5519)

---
 api4/post_test.go | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

(limited to 'api4/post_test.go')

diff --git a/api4/post_test.go b/api4/post_test.go
index 5c224cb06..67f815b7f 100644
--- a/api4/post_test.go
+++ b/api4/post_test.go
@@ -504,3 +504,63 @@ func TestSearchPostsFromUser(t *testing.T) {
 		t.Fatalf("wrong number of posts returned %v", len(posts.Order))
 	}
 }
+
+func TestGetFileInfosForPost(t *testing.T) {
+	th := Setup().InitBasic().InitSystemAdmin()
+	defer TearDown()
+	Client := th.Client
+
+	fileIds := make([]string, 3, 3)
+	if data, err := readTestFile("test.png"); err != nil {
+		t.Fatal(err)
+	} else {
+		for i := 0; i < 3; i++ {
+			fileResp, _ := Client.UploadFile(data, th.BasicChannel.Id, "test.png")
+			fileIds[i] = fileResp.FileInfos[0].Id
+		}
+	}
+
+	post := &model.Post{ChannelId: th.BasicChannel.Id, Message: "a" + model.NewId() + "a", FileIds: fileIds}
+	post, _ = Client.CreatePost(post)
+
+	infos, resp := Client.GetFileInfosForPost(post.Id, "")
+	CheckNoError(t, resp)
+
+	if len(infos) != 3 {
+		t.Fatal("missing file infos")
+	}
+
+	found := false
+	for _, info := range infos {
+		if info.Id == fileIds[0] {
+			found = true
+		}
+	}
+
+	if !found {
+		t.Fatal("missing file info")
+	}
+
+	infos, resp = Client.GetFileInfosForPost(post.Id, resp.Etag)
+	CheckEtag(t, infos, resp)
+
+	infos, resp = Client.GetFileInfosForPost(th.BasicPost.Id, "")
+	CheckNoError(t, resp)
+
+	if len(infos) != 0 {
+		t.Fatal("should have no file infos")
+	}
+
+	_, resp = Client.GetFileInfosForPost("junk", "")
+	CheckBadRequestStatus(t, resp)
+
+	_, resp = Client.GetFileInfosForPost(model.NewId(), "")
+	CheckForbiddenStatus(t, resp)
+
+	Client.Logout()
+	_, resp = Client.GetFileInfosForPost(model.NewId(), "")
+	CheckUnauthorizedStatus(t, resp)
+
+	_, resp = th.SystemAdminClient.GetFileInfosForPost(th.BasicPost.Id, "")
+	CheckNoError(t, resp)
+}
-- 
cgit v1.2.3-1-g7c22