Function TestBytesReadPutAttachment

rest/bytes_read_public_api_test.go:124–168

(t *testing.T)

}

func TestBytesReadPutAttachment(t *testing.T) {
	rt := NewRestTester(t, &RestTesterConfig{SyncFn: channels.DocChannelsSyncFunction})
	defer rt.Close()

	// create user
	rt.CreateUser("alice", []string{"ABC"})

	// add a doc for an attachment to be added to
	resp := rt.SendAdminRequest(http.MethodPut, fmt.Sprintf("/%s/%s", rt.GetSingleKeyspace(), "doc1"), `{"channels":["ABC"]}`)
	RequireStatus(t, resp, http.StatusCreated)
	var body db.Body
	require.NoError(t, base.JSONUnmarshal(resp.Body.Bytes(), &body))
	require.Equal(t, true, body["ok"])
	revid := body["rev"].(string)

	resp = rt.SendUserRequest(http.MethodGet, fmt.Sprintf("/%s/%s", rt.GetSingleKeyspace(), "doc1"), ``, "alice")
	RequireStatus(t, resp, http.StatusOK)

	attachmentBody := "this is the body of attachment"
	attachmentContentType := "content/type"
	reqHeaders := map[string]string{
		"Content-Type": attachmentContentType,
	}
	byteArrayAttachmentBody := []byte(attachmentBody)

	// attach to existing document created above
	resp = rt.SendUserRequestWithHeaders("PUT", "/{{.keyspace}}/doc1/attach1?rev="+revid, attachmentBody, reqHeaders, "alice", "letmein")
	RequireStatus(t, resp, http.StatusCreated)

	// assert the stat has increased by the attachment endpoint input
	base.RequireWaitForStat(t, func() int64 {
		return rt.GetDatabase().DbStats.DatabaseStats.PublicRestBytesRead.Value()
	}, int64(len(byteArrayAttachmentBody)))

	// test incorrect user still increments count
	resp = rt.SendUserRequestWithHeaders("PUT", "/{{.keyspace}}/doc1/attach1?rev="+revid, attachmentBody, reqHeaders, "bob", "letmein")
	RequireStatus(t, resp, http.StatusUnauthorized)

	newStatNum := len(byteArrayAttachmentBody) * 2

	base.RequireWaitForStat(t, func() int64 {
		return rt.GetDatabase().DbStats.DatabaseStats.PublicRestBytesRead.Value()
	}, int64(newStatNum))

}

func TestBytesReadRevDiff(t *testing.T) {
	rt := NewRestTester(t, nil)