Function TestBasicAttachmentRemoval

rest/attachment_test.go:1081–2054

(t *testing.T)

}

func TestBasicAttachmentRemoval(t *testing.T) {
	rt := NewRestTester(t, &RestTesterConfig{GuestEnabled: true})
	defer rt.Close()

	retrieveAttachment := func(t *testing.T, docID, attName string) (attBody string) {
		resource := fmt.Sprintf("/{{.keyspace}}/%s/%s", docID, attName)
		response := rt.SendRequest(http.MethodGet, resource, "")
		RequireStatus(t, response, http.StatusOK)
		return response.Body.String()
	}

	retrieveAttachmentKey := func(t *testing.T, docID, attName string) string {
		resource := fmt.Sprintf("/{{.keyspace}}/%s/%s?meta=true", docID, attName)
		response := rt.SendRequest(http.MethodGet, resource, "")
		RequireStatus(t, response, http.StatusOK)
		return getAttachmentMetaFromREST(t, response.BodyBytes()).Key
	}

	requireAttachmentNotFound := func(t *testing.T, docID, attName string) {
		resource := fmt.Sprintf("/{{.keyspace}}/%s/%s", docID, attName)
		response := rt.SendRequest(http.MethodGet, resource, "")
		RequireStatus(t, response, http.StatusNotFound)
	}

	retrieveAttachmentMeta := func(t *testing.T, docID string) db.AttachmentMap {
		response := rt.SendRequest(http.MethodGet, fmt.Sprintf("/{{.keyspace}}/%s", docID), "")
		RequireStatus(t, response, http.StatusOK)
		return db.GetAttachmentsFromInlineBody(t, response.BodyBytes())
	}

	dataStore := rt.GetSingleDataStore()
	requireAttachmentFound := func(attKey string, attBodyExpected []byte) {
		var attBodyActual []byte
		_, err := dataStore.Get(attKey, &attBodyActual)
		require.NoError(t, err)
		assert.Equal(t, attBodyExpected, attBodyActual)
	}

	rt.Run("single attachment removal upon document update", func(t *testing.T) {
		// Create a document.
		docID := "foo"
		version := rt.CreateTestDoc(docID)
		// Add an attachment to the document.
		attName := "foo.txt"
		attBody := "this is the body of attachment foo.txt"
		version = rt.storeAttachment(docID, version, attName, attBody)

		// Retrieve the attachment added to the document.
		actualAttBody := retrieveAttachment(t, docID, attName)
		require.Equal(t, attBody, actualAttBody)

		// Get the document and check the attachment metadata.
		attachments := retrieveAttachmentMeta(t, docID)
		require.Equal(t, db.AttachmentMap{
			attName: {
				Stub:        true,
				ContentType: "content/type",
				Digest:      "sha1-CTJaowVFZ4ozgmvBageTH9w+OKU=",