Function TestProveAttachmentNotFound

rest/attachment_test.go:2460–2511

CBG-2004: Test that prove attachment over Blip works correctly when receiving a ErrAttachmentNotFound

(t *testing.T)

// CBG-2004: Test that prove attachment over Blip works correctly when receiving a ErrAttachmentNotFound
func TestProveAttachmentNotFound(t *testing.T) {
	rt := NewRestTester(t, &RestTesterConfig{PersistentConfig: true})
	defer rt.Close()

	dbConfig := rt.NewDbConfig()
	RequireStatus(t, rt.CreateDatabase("db", dbConfig), http.StatusCreated)
	rt.GetDatabase().EnableAllowConflicts(rt.TB())
	rt.SetAdminParty(true)

	bt := NewBlipTesterFromSpecWithRT(rt, nil)
	defer bt.Close()

	attachmentData := []byte("attachmentA")
	attachmentDataEncoded := base64.StdEncoding.EncodeToString(attachmentData)

	bt.blipContext.HandlerForProfile[db.MessageProveAttachment] = func(msg *blip.Message) {
		status, errMsg := base.ErrorAsHTTPStatus(db.ErrAttachmentNotFound)
		msg.Response().SetError("HTTP", status, errMsg)
	}

	// Handler for when full attachment is requested
	bt.blipContext.HandlerForProfile[db.MessageGetAttachment] = func(msg *blip.Message) {
		resp := msg.Response()
		resp.SetBody(attachmentData)
		resp.SetCompressed(msg.Properties[db.BlipCompress] == "true")
	}

	// Initial set up
	sent, _, _, err := bt.SendRev("doc1", "1-abc", []byte(`{"key": "val", "_attachments": {"attachment": {"data": "`+attachmentDataEncoded+`"}}}`), blip.Properties{})
	require.True(t, sent)
	require.NoError(t, err)

	rt.WaitForPendingChanges()

	// Should log:
	// "Peer sent prove attachment error 404 attachment not found, falling back to getAttachment for proof in doc <ud>doc1</ud> (digest sha1-wzp8ZyykdEuZ9GuqmxQ7XDrY7Co=)"
	base.SetUpTestLogging(t, base.LevelInfo, base.KeyAll)

	// Use different attachment name to bypass digest check in ForEachStubAttachment() which skips prove attachment code
	// Set attachment to V2 so it can be retrieved by RT successfully
	sent, _, _, err = bt.SendRev("doc1", "2-abc", []byte(`{"key": "val", "_attachments":{"attach":{"digest":"sha1-wzp8ZyykdEuZ9GuqmxQ7XDrY7Co=","length":11,"stub":true,"revpos":1,"ver":2}}}`), blip.Properties{})
	require.True(t, sent)
	require.NoError(t, err)

	rt.WaitForPendingChanges()
	// Check attachment is on the document
	body := rt.GetDocBody("doc1")
	assert.Equal(t, "2-abc", body.ExtractRev())
	resp := rt.SendAdminRequest("GET", "/{{.keyspace}}/doc1/attach", "")
	RequireStatus(t, resp, 200)
	assert.EqualValues(t, attachmentData, resp.BodyBytes())
}

// Test adding / retrieving attachments
func TestAttachments(t *testing.T) {