661 lines
23 KiB
Go
661 lines
23 KiB
Go
package api
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-pkgz/lgr"
|
|
R "github.com/go-pkgz/rest"
|
|
"github.com/pkg/errors"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/umputun/remark/backend/app/rest"
|
|
|
|
"github.com/umputun/remark/backend/app/store"
|
|
"github.com/umputun/remark/backend/app/store/image"
|
|
)
|
|
|
|
func TestRest_Create(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
resp, err := post(t, ts.URL+"/api/v1/comment",
|
|
`{"text": "test 123", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`)
|
|
assert.Nil(t, err)
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
require.Equal(t, http.StatusCreated, resp.StatusCode, string(b))
|
|
|
|
t.Log(string(b))
|
|
c := R.JSON{}
|
|
err = json.Unmarshal(b, &c)
|
|
assert.Nil(t, err)
|
|
loc := c["locator"].(map[string]interface{})
|
|
assert.Equal(t, "radio-t", loc["site"])
|
|
assert.Equal(t, "https://radio-t.com/blah1", loc["url"])
|
|
assert.True(t, len(c["id"].(string)) > 8)
|
|
}
|
|
|
|
func TestRest_CreateOldPost(t *testing.T) {
|
|
ts, srv, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
// make old, but not too old comment
|
|
old := store.Comment{Text: "test test old", ParentID: "", Timestamp: time.Now().AddDate(0, 0, -5),
|
|
Locator: store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah1"}, User: store.User{ID: "u1"}}
|
|
_, err := srv.DataService.Create(old)
|
|
assert.Nil(t, err)
|
|
|
|
comments, err := srv.DataService.Find(store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah1"}, "time")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, len(comments))
|
|
|
|
// try to add new comment to the same old post
|
|
resp, err := post(t, ts.URL+"/api/v1/comment",
|
|
`{"text": "test 123", "locator":{"site": "radio-t","url": "https://radio-t.com/blah1"}}`)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, http.StatusCreated, resp.StatusCode)
|
|
|
|
assert.Nil(t, srv.DataService.DeleteAll("radio-t"))
|
|
// make too old comment
|
|
old = store.Comment{Text: "test test old", ParentID: "", Timestamp: time.Now().AddDate(0, 0, -15),
|
|
Locator: store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah1"}, User: store.User{ID: "u1"}}
|
|
_, err = srv.DataService.Create(old)
|
|
assert.Nil(t, err)
|
|
|
|
resp, err = post(t, ts.URL+"/api/v1/comment",
|
|
`{"text": "test 123", "locator":{"site": "radio-t","url": "https://radio-t.com/blah1"}}`)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, http.StatusForbidden, resp.StatusCode)
|
|
}
|
|
|
|
func TestRest_CreateTooBig(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
longComment := fmt.Sprintf(`{"text": "%4001s", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`, "Щ")
|
|
|
|
resp, err := post(t, ts.URL+"/api/v1/comment", longComment)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
c := R.JSON{}
|
|
err = json.Unmarshal(b, &c)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "comment text exceeded max allowed size 4000 (4001)", c["error"])
|
|
assert.Equal(t, "invalid comment", c["details"])
|
|
|
|
veryLongComment := fmt.Sprintf(`{"text": "%70000s", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`, "Щ")
|
|
resp, err = post(t, ts.URL+"/api/v1/comment", veryLongComment)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
|
|
b, err = ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
c = R.JSON{}
|
|
err = json.Unmarshal(b, &c)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "http: request body too large", c["error"])
|
|
assert.Equal(t, "can't bind comment", c["details"])
|
|
}
|
|
|
|
func TestRest_CreateWithRestrictedWord(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
badComment := fmt.Sprintf(`{"text": "What the duck is that?", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`)
|
|
|
|
resp, err := post(t, ts.URL+"/api/v1/comment", badComment)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
c := R.JSON{}
|
|
err = json.Unmarshal(b, &c)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "comment contains restricted words", c["error"])
|
|
assert.Equal(t, "invalid comment", c["details"])
|
|
}
|
|
|
|
func TestRest_CreateRejected(t *testing.T) {
|
|
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
body := `{"text": "test 123", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`
|
|
|
|
// try to create without auth
|
|
resp, err := http.Post(ts.URL+"/api/v1/comment", "", strings.NewReader(body))
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 401, resp.StatusCode)
|
|
}
|
|
|
|
func TestRest_CreateAndGet(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
// create comment
|
|
resp, err := post(t, ts.URL+"/api/v1/comment",
|
|
`{"text": "**test** *123*\n\n http://radio-t.com", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`)
|
|
require.Nil(t, err)
|
|
require.Equal(t, http.StatusCreated, resp.StatusCode)
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
c := R.JSON{}
|
|
err = json.Unmarshal(b, &c)
|
|
assert.Nil(t, err)
|
|
|
|
id := c["id"].(string)
|
|
|
|
// get created comment by id as admin
|
|
res, code := getWithAdminAuth(t, fmt.Sprintf("%s/api/v1/id/%s?site=radio-t&url=https://radio-t.com/blah1", ts.URL, id))
|
|
assert.Equal(t, 200, code)
|
|
comment := store.Comment{}
|
|
err = json.Unmarshal([]byte(res), &comment)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "<p><strong>test</strong> <em>123</em></p>\n\n<p><a href=\"http://radio-t.com\" rel=\"nofollow\">http://radio-t.com</a></p>\n", comment.Text)
|
|
assert.Equal(t, "**test** *123*\n\n http://radio-t.com", comment.Orig)
|
|
assert.Equal(t, store.User{Name: "admin", ID: "admin", Admin: true, Blocked: false,
|
|
IP: "dbc7c999343f003f189f70aaf52cc04443f90790"},
|
|
comment.User)
|
|
t.Logf("%+v", comment)
|
|
|
|
// get created comment by id as non-admin
|
|
res, code = getWithDevAuth(t, fmt.Sprintf("%s/api/v1/id/%s?site=radio-t&url=https://radio-t.com/blah1", ts.URL, id))
|
|
assert.Equal(t, 200, code)
|
|
comment = store.Comment{}
|
|
err = json.Unmarshal([]byte(res), &comment)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, store.User{Name: "admin", ID: "admin", Admin: true, Blocked: false, IP: ""}, comment.User, "no ip")
|
|
}
|
|
|
|
func TestRest_Update(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
c1 := store.Comment{Text: "test test #1", ParentID: "p1",
|
|
Locator: store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah1"}}
|
|
id := addComment(t, c1, ts)
|
|
|
|
client := http.Client{}
|
|
req, err := http.NewRequest(http.MethodPut, ts.URL+"/api/v1/comment/"+id+"?site=radio-t&url=https://radio-t.com/blah1",
|
|
strings.NewReader(`{"text":"updated text", "summary":"my edit"}`))
|
|
assert.Nil(t, err)
|
|
req.Header.Add("X-JWT", devToken)
|
|
b, err := client.Do(req)
|
|
assert.Nil(t, err)
|
|
body, err := ioutil.ReadAll(b.Body)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 200, b.StatusCode, string(body))
|
|
|
|
// comments returned by update
|
|
c2 := store.Comment{}
|
|
err = json.Unmarshal(body, &c2)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, id, c2.ID)
|
|
assert.Equal(t, "<p>updated text</p>\n", c2.Text)
|
|
assert.Equal(t, "updated text", c2.Orig)
|
|
assert.Equal(t, "my edit", c2.Edit.Summary)
|
|
assert.True(t, time.Since(c2.Edit.Timestamp) < 1*time.Second)
|
|
|
|
// read updated comment
|
|
res, code := getWithAdminAuth(t, fmt.Sprintf("%s/api/v1/id/%s?site=radio-t&url=https://radio-t.com/blah1", ts.URL, id))
|
|
assert.Equal(t, 200, code)
|
|
c3 := store.Comment{}
|
|
err = json.Unmarshal([]byte(res), &c3)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, c2, c3, "same as response from update")
|
|
}
|
|
|
|
func TestRest_UpdateDelete(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
c1 := store.Comment{Text: "test test #1", ParentID: "p1",
|
|
Locator: store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah1"}}
|
|
id := addComment(t, c1, ts)
|
|
|
|
// check multi count updated
|
|
resp, err := post(t, ts.URL+"/api/v1/counts?site=radio-t", `["https://radio-t.com/blah1","https://radio-t.com/blah2"]`)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
bb, err := ioutil.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
j := []store.PostInfo{}
|
|
err = json.Unmarshal(bb, &j)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, []store.PostInfo([]store.PostInfo{{URL: "https://radio-t.com/blah1", Count: 1},
|
|
{URL: "https://radio-t.com/blah2", Count: 0}}), j)
|
|
|
|
// delete a comment
|
|
client := http.Client{}
|
|
req, err := http.NewRequest(http.MethodPut, ts.URL+"/api/v1/comment/"+id+"?site=radio-t&url=https://radio-t.com/blah1",
|
|
strings.NewReader(`{"delete": true, "summary":"removed by user"}`))
|
|
require.NoError(t, err)
|
|
req.Header.Add("X-JWT", devToken)
|
|
b, err := client.Do(req)
|
|
require.NoError(t, err)
|
|
body, err := ioutil.ReadAll(b.Body)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 200, b.StatusCode, string(body))
|
|
|
|
// comments returned by update
|
|
c2 := store.Comment{}
|
|
err = json.Unmarshal(body, &c2)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, id, c2.ID)
|
|
assert.True(t, c2.Deleted)
|
|
|
|
// read updated comment
|
|
res, code := getWithDevAuth(t, fmt.Sprintf("%s/api/v1/id/%s?site=radio-t&url=https://radio-t.com/blah1", ts.URL, id))
|
|
assert.Equal(t, 200, code)
|
|
c3 := store.Comment{}
|
|
err = json.Unmarshal([]byte(res), &c3)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "", c3.Text)
|
|
assert.Equal(t, "", c3.Orig)
|
|
assert.True(t, c3.Deleted)
|
|
|
|
// check multi count updated
|
|
resp, err = post(t, ts.URL+"/api/v1/counts?site=radio-t", `["https://radio-t.com/blah1","https://radio-t.com/blah2"]`)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
bb, err = ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
j = []store.PostInfo{}
|
|
err = json.Unmarshal(bb, &j)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, []store.PostInfo([]store.PostInfo{{URL: "https://radio-t.com/blah1", Count: 0},
|
|
{URL: "https://radio-t.com/blah2", Count: 0}}), j)
|
|
}
|
|
|
|
func TestRest_UpdateNotOwner(t *testing.T) {
|
|
ts, srv, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
c1 := store.Comment{Text: "test test #1", ParentID: "p1",
|
|
Locator: store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah1"}, User: store.User{ID: "xyz"}}
|
|
id1, err := srv.DataService.Create(c1)
|
|
assert.Nil(t, err)
|
|
|
|
client := http.Client{}
|
|
req, err := http.NewRequest(http.MethodPut, ts.URL+"/api/v1/comment/"+id1+
|
|
"?site=radio-t&url=https://radio-t.com/blah1", strings.NewReader(`{"text":"updated text", "summary":"my edit"}`))
|
|
assert.Nil(t, err)
|
|
req.Header.Add("X-JWT", devToken)
|
|
b, err := client.Do(req)
|
|
assert.Nil(t, err)
|
|
body, err := ioutil.ReadAll(b.Body)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 403, b.StatusCode, string(body), "update from non-owner")
|
|
assert.Equal(t, `{"code":3,"details":"can not edit comments for other users","error":"rejected"}`+"\n", string(body))
|
|
|
|
client = http.Client{}
|
|
req, err = http.NewRequest(http.MethodPut, ts.URL+"/api/v1/comment/"+id1+
|
|
"?site=radio-t&url=https://radio-t.com/blah1", strings.NewReader(`ERRR "text":"updated text", "summary":"my"}`))
|
|
assert.Nil(t, err)
|
|
req.Header.Add("X-JWT", devToken)
|
|
b, err = client.Do(req)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 400, b.StatusCode, string(body), "update is not json")
|
|
}
|
|
|
|
func TestRest_UpdateWithRestrictedWords(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
c1 := store.Comment{Text: "What the quack is that?", ParentID: "p1",
|
|
Locator: store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah1"}}
|
|
id := addComment(t, c1, ts)
|
|
|
|
client := http.Client{}
|
|
req, err := http.NewRequest(http.MethodPut, ts.URL+"/api/v1/comment/"+id+"?site=radio-t&url=https://radio-t.com/blah1",
|
|
strings.NewReader(`{"text":"What the duck is that?", "summary":"my edit"}`))
|
|
assert.Nil(t, err)
|
|
req.Header.Add("X-JWT", devToken)
|
|
b, err := client.Do(req)
|
|
assert.Nil(t, err)
|
|
body, err := ioutil.ReadAll(b.Body)
|
|
assert.Nil(t, err)
|
|
c := R.JSON{}
|
|
err = json.Unmarshal(body, &c)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 400, b.StatusCode, string(body))
|
|
assert.Equal(t, "comment contains restricted words", c["error"])
|
|
assert.Equal(t, "invalid comment", c["details"])
|
|
}
|
|
|
|
func TestRest_Vote(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
c1 := store.Comment{Text: "test test #1",
|
|
Locator: store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah"}}
|
|
c2 := store.Comment{Text: "test test #2", ParentID: "p1",
|
|
Locator: store.Locator{SiteID: "radio-t", URL: "https://radio-t.com/blah"}}
|
|
|
|
id1 := addComment(t, c1, ts)
|
|
addComment(t, c2, ts)
|
|
|
|
vote := func(val int) int {
|
|
client := http.Client{}
|
|
req, err := http.NewRequest(http.MethodPut,
|
|
fmt.Sprintf("%s/api/v1/vote/%s?site=radio-t&url=https://radio-t.com/blah&vote=%d", ts.URL, id1, val), nil)
|
|
assert.Nil(t, err)
|
|
req.SetBasicAuth("admin", "password")
|
|
resp, err := client.Do(req)
|
|
assert.Nil(t, err)
|
|
return resp.StatusCode
|
|
}
|
|
|
|
assert.Equal(t, 200, vote(1), "first vote allowed")
|
|
assert.Equal(t, 400, vote(1), "second vote rejected")
|
|
body, code := get(t, fmt.Sprintf("%s/api/v1/id/%s?site=radio-t&url=https://radio-t.com/blah", ts.URL, id1))
|
|
assert.Equal(t, 200, code)
|
|
cr := store.Comment{}
|
|
err := json.Unmarshal([]byte(body), &cr)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, cr.Score)
|
|
assert.Equal(t, map[string]bool{"admin": true}, cr.Votes)
|
|
|
|
assert.Equal(t, 200, vote(-1), "opposite vote allowed")
|
|
body, code = get(t, fmt.Sprintf("%s/api/v1/id/%s?site=radio-t&url=https://radio-t.com/blah", ts.URL, id1))
|
|
assert.Equal(t, 200, code)
|
|
cr = store.Comment{}
|
|
err = json.Unmarshal([]byte(body), &cr)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 0, cr.Score)
|
|
assert.Equal(t, map[string]bool{}, cr.Votes)
|
|
}
|
|
|
|
func TestRest_UserAllData(t *testing.T) {
|
|
ts, srv, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
// write 3 comments
|
|
user := store.User{ID: "dev", Name: "user name 1"}
|
|
c1 := store.Comment{User: user, Text: "test test #1", Locator: store.Locator{SiteID: "radio-t",
|
|
URL: "https://radio-t.com/blah1"}, Timestamp: time.Date(2018, 05, 27, 1, 14, 10, 0, time.Local)}
|
|
c2 := store.Comment{User: user, Text: "test test #2", ParentID: "p1", Locator: store.Locator{SiteID: "radio-t",
|
|
URL: "https://radio-t.com/blah1"}, Timestamp: time.Date(2018, 05, 27, 1, 14, 20, 0, time.Local)}
|
|
c3 := store.Comment{User: user, Text: "test test #3", ParentID: "p1", Locator: store.Locator{SiteID: "radio-t",
|
|
URL: "https://radio-t.com/blah1"}, Timestamp: time.Date(2018, 05, 27, 1, 14, 25, 0, time.Local)}
|
|
_, err := srv.DataService.Create(c1)
|
|
require.Nil(t, err, "%+v", err)
|
|
_, err = srv.DataService.Create(c2)
|
|
require.Nil(t, err)
|
|
_, err = srv.DataService.Create(c3)
|
|
require.Nil(t, err)
|
|
|
|
client := &http.Client{Timeout: 1 * time.Second}
|
|
req, err := http.NewRequest("GET", ts.URL+"/api/v1/userdata?site=radio-t", nil)
|
|
require.Nil(t, err)
|
|
req.Header.Add("X-JWT", devToken)
|
|
resp, err := client.Do(req)
|
|
require.Nil(t, err)
|
|
require.Equal(t, 200, resp.StatusCode)
|
|
require.Equal(t, "application/gzip", resp.Header.Get("Content-Type"))
|
|
|
|
ungzReader, err := gzip.NewReader(resp.Body)
|
|
assert.NoError(t, err)
|
|
ungzBody, err := ioutil.ReadAll(ungzReader)
|
|
assert.NoError(t, err)
|
|
assert.True(t, strings.HasPrefix(string(ungzBody),
|
|
`{"info": {"name":"developer one","id":"dev","picture":"http://example.com/pic.png","ip":"127.0.0.1","admin":false}, "comments":[{`))
|
|
assert.Equal(t, 3, strings.Count(string(ungzBody), `"text":`), "3 comments inside")
|
|
t.Logf("%s", string(ungzBody))
|
|
|
|
parsed := struct {
|
|
Info store.User `json:"info"`
|
|
Comments []store.Comment `json:"comments"`
|
|
}{}
|
|
|
|
err = json.Unmarshal(ungzBody, &parsed)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, store.User{Name: "developer one", ID: "dev",
|
|
Picture: "http://example.com/pic.png", IP: "127.0.0.1"}, parsed.Info)
|
|
assert.Equal(t, 3, len(parsed.Comments))
|
|
|
|
req, err = http.NewRequest("GET", ts.URL+"/api/v1/userdata?site=radio-t", nil)
|
|
require.Nil(t, err)
|
|
resp, err = client.Do(req)
|
|
require.Nil(t, err)
|
|
require.Equal(t, 401, resp.StatusCode)
|
|
}
|
|
|
|
func TestRest_UserAllDataManyComments(t *testing.T) {
|
|
ts, srv, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
user := store.User{ID: "dev", Name: "user name 1"}
|
|
c := store.Comment{User: user, Text: "test test #1", Locator: store.Locator{SiteID: "radio-t",
|
|
URL: "https://radio-t.com/blah1"}, Timestamp: time.Date(2018, 05, 27, 1, 14, 10, 0, time.Local)}
|
|
|
|
for i := 0; i < 51; i++ {
|
|
c.ID = fmt.Sprintf("id-%03d", i)
|
|
c.Timestamp = c.Timestamp.Add(time.Second)
|
|
_, err := srv.DataService.Create(c)
|
|
require.Nil(t, err)
|
|
}
|
|
client := &http.Client{Timeout: 1 * time.Second}
|
|
req, err := http.NewRequest("GET", ts.URL+"/api/v1/userdata?site=radio-t", nil)
|
|
require.Nil(t, err)
|
|
req.Header.Add("X-JWT", devToken)
|
|
resp, err := client.Do(req)
|
|
require.Nil(t, err)
|
|
require.Equal(t, 200, resp.StatusCode)
|
|
require.Equal(t, "application/gzip", resp.Header.Get("Content-Type"))
|
|
|
|
ungzReader, err := gzip.NewReader(resp.Body)
|
|
assert.NoError(t, err)
|
|
ungzBody, err := ioutil.ReadAll(ungzReader)
|
|
assert.NoError(t, err)
|
|
assert.True(t, strings.HasPrefix(string(ungzBody),
|
|
`{"info": {"name":"developer one","id":"dev","picture":"http://example.com/pic.png","ip":"127.0.0.1","admin":false}, "comments":[{`))
|
|
assert.Equal(t, 51, strings.Count(string(ungzBody), `"text":`), "51 comments inside")
|
|
}
|
|
|
|
func TestRest_DeleteMe(t *testing.T) {
|
|
ts, srv, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
client := http.Client{}
|
|
req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("%s/api/v1/deleteme?site=radio-t", ts.URL), nil)
|
|
assert.Nil(t, err)
|
|
req.Header.Add("X-JWT", devToken)
|
|
resp, err := client.Do(req)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 200, resp.StatusCode)
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
|
|
m := map[string]string{}
|
|
err = json.Unmarshal(body, &m)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "radio-t", m["site"])
|
|
assert.Equal(t, "dev", m["user_id"])
|
|
|
|
token := m["token"]
|
|
claims, err := srv.Authenticator.TokenService().Parse(token)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "dev", claims.User.ID)
|
|
assert.Equal(t, "https://demo.remark42.com/web/deleteme.html?token="+token, m["link"])
|
|
|
|
req, err = http.NewRequest(http.MethodPost, fmt.Sprintf("%s/api/v1/deleteme?site=radio-t", ts.URL), nil)
|
|
assert.Nil(t, err)
|
|
resp, err = client.Do(req)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 401, resp.StatusCode)
|
|
}
|
|
|
|
func TestRest_SavePictureCtrl(t *testing.T) {
|
|
ts, _, teardown := startupT(t)
|
|
defer teardown()
|
|
|
|
// save picture
|
|
savePic := func(name string) (id string) {
|
|
r := strings.NewReader("file content 123")
|
|
bodyBuf := &bytes.Buffer{}
|
|
bodyWriter := multipart.NewWriter(bodyBuf)
|
|
fileWriter, err := bodyWriter.CreateFormFile("file", name)
|
|
require.NoError(t, err)
|
|
_, err = io.Copy(fileWriter, r)
|
|
require.NoError(t, err)
|
|
contentType := bodyWriter.FormDataContentType()
|
|
require.NoError(t, bodyWriter.Close())
|
|
|
|
client := http.Client{}
|
|
req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("%s/api/v1/picture", ts.URL), bodyBuf)
|
|
require.NoError(t, err)
|
|
req.Header.Add("Content-Type", contentType)
|
|
req.Header.Add("X-JWT", devToken)
|
|
resp, err := client.Do(req)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 200, resp.StatusCode)
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
require.Nil(t, err)
|
|
|
|
m := map[string]string{}
|
|
err = json.Unmarshal(body, &m)
|
|
assert.NoError(t, err)
|
|
assert.True(t, m["id"] != "")
|
|
return m["id"]
|
|
}
|
|
|
|
id := savePic("picture.png")
|
|
resp, err := http.Get(fmt.Sprintf("%s/api/v1/picture/%s", ts.URL, id))
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 200, resp.StatusCode)
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, "file content 123", string(body))
|
|
assert.Equal(t, "image/png", resp.Header.Get("Content-Type"))
|
|
|
|
id = savePic("picture.gif")
|
|
resp, err = http.Get(fmt.Sprintf("%s/api/v1/picture/%s", ts.URL, id))
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 200, resp.StatusCode)
|
|
assert.Equal(t, "image/gif", resp.Header.Get("Content-Type"))
|
|
|
|
id = savePic("picture.jpg")
|
|
resp, err = http.Get(fmt.Sprintf("%s/api/v1/picture/%s", ts.URL, id))
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 200, resp.StatusCode)
|
|
assert.Equal(t, "image/jpeg", resp.Header.Get("Content-Type"))
|
|
|
|
id = savePic("picture.blah")
|
|
resp, err = http.Get(fmt.Sprintf("%s/api/v1/picture/%s", ts.URL, id))
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 200, resp.StatusCode)
|
|
assert.Equal(t, "image/*", resp.Header.Get("Content-Type"))
|
|
|
|
resp, err = http.Get(fmt.Sprintf("%s/api/v1/picture/blah/pic.blah", ts.URL))
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 400, resp.StatusCode)
|
|
}
|
|
|
|
func TestRest_CreateWithPictures(t *testing.T) {
|
|
ts, svc, teardown := startupT(t)
|
|
defer func() {
|
|
teardown()
|
|
os.RemoveAll("/tmp/remark42")
|
|
}()
|
|
lgr.Setup(lgr.Debug, lgr.CallerFile, lgr.CallerFunc)
|
|
|
|
svc.ImageService = &image.Service{
|
|
Store: &image.FileSystem{
|
|
Staging: "/tmp/remark42/images.staging",
|
|
Location: "/tmp/remark42/images",
|
|
MaxSize: 1000,
|
|
},
|
|
TTL: time.Millisecond * 100,
|
|
}
|
|
svc.DataService.EditDuration = time.Millisecond * 100
|
|
svc.DataService.ImageService = svc.ImageService
|
|
|
|
uploadPicture := func(file, content string) (id string) {
|
|
r := strings.NewReader(content)
|
|
bodyBuf := &bytes.Buffer{}
|
|
bodyWriter := multipart.NewWriter(bodyBuf)
|
|
fileWriter, err := bodyWriter.CreateFormFile("file", file)
|
|
require.NoError(t, err)
|
|
_, err = io.Copy(fileWriter, r)
|
|
require.NoError(t, err)
|
|
contentType := bodyWriter.FormDataContentType()
|
|
require.NoError(t, bodyWriter.Close())
|
|
client := http.Client{}
|
|
req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("%s/api/v1/picture", ts.URL), bodyBuf)
|
|
require.NoError(t, err)
|
|
req.Header.Add("Content-Type", contentType)
|
|
req.Header.Add("X-JWT", devToken)
|
|
resp, err := client.Do(req)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 200, resp.StatusCode)
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
require.Nil(t, err)
|
|
m := map[string]string{}
|
|
err = json.Unmarshal(body, &m)
|
|
assert.NoError(t, err)
|
|
assert.Contains(t, m["id"], ".png")
|
|
return m["id"]
|
|
}
|
|
|
|
id1 := uploadPicture("pic1.png", "file content 123")
|
|
id2 := uploadPicture("pic2.png", "file content 12345")
|
|
id3 := uploadPicture("pic3.png", "file content xyz12365789")
|
|
|
|
text := fmt.Sprintf(`text 123  *xxx*  `, id1, id2, id3)
|
|
body := fmt.Sprintf(`{"text": "%s", "locator":{"url": "https://radio-t.com/blah1", "site": "radio-t"}}`, text)
|
|
|
|
resp, err := post(t, ts.URL+"/api/v1/comment", body)
|
|
assert.Nil(t, err)
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
assert.Nil(t, err)
|
|
require.Equal(t, http.StatusCreated, resp.StatusCode, string(b))
|
|
|
|
_, err = os.Stat("/tmp/remark42/images/" + id1)
|
|
assert.NotNil(t, err, "not moved from staging yet")
|
|
|
|
time.Sleep(300 * time.Millisecond)
|
|
_, err = os.Stat("/tmp/remark42/images/" + id1)
|
|
assert.NoError(t, err, "moved from staging")
|
|
_, err = os.Stat("/tmp/remark42/images/" + id2)
|
|
assert.NoError(t, err, "moved from staging")
|
|
_, err = os.Stat("/tmp/remark42/images/" + id3)
|
|
assert.NoError(t, err, "moved from staging")
|
|
}
|
|
|
|
func TestRest_parseError(t *testing.T) {
|
|
tbl := []struct {
|
|
err error
|
|
res int
|
|
}{
|
|
{errors.New("can not vote for his own comment"), rest.ErrVoteSelf},
|
|
{errors.New("already voted for"), rest.ErrVoteDbl},
|
|
{errors.New("maximum number of votes exceeded for comment"), rest.ErrVoteMax},
|
|
{errors.New("minimal score reached for comment"), rest.ErrVoteMinScore},
|
|
{errors.New("too late to edit"), rest.ErrCommentEditExpired},
|
|
{errors.New("parent comment with reply can't be edited"), rest.ErrCommentEditChanged},
|
|
{errors.New("blah blah"), rest.ErrInternal},
|
|
}
|
|
|
|
svc := Rest{}
|
|
for n, tt := range tbl {
|
|
t.Run(strconv.Itoa(n), func(t *testing.T) {
|
|
res := svc.parseError(tt.err, rest.ErrInternal)
|
|
assert.Equal(t, tt.res, res)
|
|
})
|
|
}
|
|
}
|