Function TestStoreAttachments

db/attachment_test.go:669–769

(t *testing.T)

}

func TestStoreAttachments(t *testing.T) {
	db, ctx := setupTestDB(t)
	defer db.Close(ctx)
	collection, ctx := GetSingleDatabaseCollectionWithUser(ctx, t, db)

	var revBody Body
	// Simulate Invalid _attachments scenario; try to put a document with bad
	// attachment metadata. It should throw "Invalid _attachments" error.
	revText := `{"key1": "value1", "_attachments": {"att1.txt": "YXR0MS50eHQ="}}`
	assert.NoError(t, base.JSONUnmarshal([]byte(revText), &revBody))
	revId, doc, err := collection.Put(ctx, "doc1", revBody)
	assert.Empty(t, revId, "The revId should be empty since the request has attachment")
	assert.Empty(t, doc, "The doc should be empty since the request has attachment")
	assert.Error(t, err, "It should throw 400 Invalid _attachments")
	assert.Contains(t, err.Error(), strconv.Itoa(http.StatusBadRequest))

	// Simulate illegal base64 data error while storing attachments in Couchbase database.
	revText = `{"key1": "value1", "_attachments": {"att1.txt": {"data": "%$^&**"}}}`
	assert.NoError(t, base.JSONUnmarshal([]byte(revText), &revBody))
	revId, doc, err = collection.Put(ctx, "doc1", revBody)
	assert.Empty(t, revId, "The revId should be empty since illegal base64 data in attachment")
	assert.Empty(t, doc, "The doc should be empty since illegal base64 data in attachment")
	assert.Error(t, err, "It should throw illegal base64 data at input byte 0 error")
	assert.Contains(t, err.Error(), "illegal base64 data at input byte 0")

	// Simulate a valid scenario; attachment contains data, so store it in the database.
	// Include content type, encoding, attachment length  in attachment metadata.
	revText = `{"key1": "value1", "_attachments": {"att1.txt": {"data": "YXR0MS50eHQ=", "content_type": "text/plain", "encoding": "utf-8", "length": 12}}}`
	assert.NoError(t, base.JSONUnmarshal([]byte(revText), &revBody))
	revId, doc, err = collection.Put(ctx, "doc1", revBody)
	assert.NoError(t, err, "Couldn't update document")
	assert.NotEmpty(t, revId, "Document revision id should be generated")
	require.NotNil(t, doc)
	assert.NotEmpty(t, doc.Attachments(), "Attachment metadata should be populated")
	attachment := doc.Attachments()["att1.txt"].(map[string]interface{})
	assert.Equal(t, "text/plain", attachment["content_type"])
	assert.Equal(t, "sha1-crv3IVNxp3JXbP6bizTHt3GB3O0=", attachment["digest"])
	assert.Equal(t, 8, attachment["encoded_length"])
	assert.Equal(t, "utf-8", attachment["encoding"])
	assert.Equal(t, float64(12), attachment["length"])
	assert.NotEmpty(t, attachment["revpos"])
	assert.True(t, attachment["stub"].(bool))

	// Simulate a valid scenario; attachment contains data, so store it in the database.
	// Include content type, encoding in attachment metadata but no attachment length.
	revText = `{"key1": "value1", "_attachments": {"att1.txt": {"data": "YXR0MS50eHQ=", "content_type": "text/plain", "encoding": "utf-8"}}}`
	assert.NoError(t, base.JSONUnmarshal([]byte(revText), &revBody))
	revId, doc, err = collection.Put(ctx, "doc2", revBody)
	assert.NoError(t, err, "Couldn't update document")
	assert.NotEmpty(t, revId, "Document revision id should be generated")
	require.NotNil(t, doc)
	assert.NotEmpty(t, doc.Attachments(), "Attachment metadata should be populated")
	attachment = doc.Attachments()["att1.txt"].(map[string]interface{})
	assert.Equal(t, "text/plain", attachment["content_type"])
	assert.Equal(t, "sha1-crv3IVNxp3JXbP6bizTHt3GB3O0=", attachment["digest"])
	assert.Equal(t, 8, attachment["encoded_length"])
	assert.Equal(t, "utf-8", attachment["encoding"])
	assert.Empty(t, attachment["length"], "Attachment length should be empty")