Function TestUpdateExistingAttachment

rest/attachment_test.go:2324–2378

(t *testing.T)

}

func TestUpdateExistingAttachment(t *testing.T) {
	rtConfig := &RestTesterConfig{
		GuestEnabled: true,
	}

	btcRunner := NewBlipTesterClientRunner(t)
	const (
		doc1ID = "doc1"
		doc2ID = "doc2"
	)

	btcRunner.Run(func(t *testing.T) {
		rt := NewRestTester(t, rtConfig)
		defer rt.Close()

		btc := btcRunner.NewBlipTesterClientOptsWithRT(rt, nil)
		defer btc.Close()

		btcRunner.StartPull(btc.id)
		btcRunner.StartPush(btc.id)

		doc1Version := rt.PutDoc(doc1ID, `{}`)
		doc2Version := rt.PutDoc(doc2ID, `{}`)

		btcRunner.WaitForVersion(btc.id, doc1ID, doc1Version)
		btcRunner.WaitForVersion(btc.id, doc2ID, doc2Version)

		attachmentAData := base64.StdEncoding.EncodeToString([]byte("attachmentA"))
		attachmentBData := base64.StdEncoding.EncodeToString([]byte("attachmentB"))

		doc1Version = btcRunner.AddRev(btc.id, doc1ID, &doc1Version, []byte(`{"key": "val", "_attachments": {"attachment": {"data": "`+attachmentAData+`"}}}`))
		doc2Version = btcRunner.AddRev(btc.id, doc2ID, &doc2Version, []byte(`{"key": "val", "_attachments": {"attachment": {"data": "`+attachmentBData+`"}}}`))

		rt.WaitForVersion(doc1ID, doc1Version)
		rt.WaitForVersion(doc2ID, doc2Version)

		collection, ctx := rt.GetSingleTestDatabaseCollection()
		_, err := collection.GetDocument(ctx, "doc1", db.DocUnmarshalAll)
		require.NoError(t, err)
		_, err = collection.GetDocument(ctx, "doc2", db.DocUnmarshalAll)
		require.NoError(t, err)

		doc1Version = btcRunner.AddRev(btc.id, doc1ID, &doc1Version, []byte(`{"key": "val", "_attachments":{"attachment":{"digest":"sha1-SKk0IV40XSHW37d3H0xpv2+z9Ck=","length":11,"content_type":"","stub":true,"revpos":3}}}`))

		rt.WaitForVersion(doc1ID, doc1Version)

		doc1, err := collection.GetDocument(ctx, "doc1", db.DocUnmarshalAll)
		require.NoError(t, err)

		assert.Equal(t, "sha1-SKk0IV40XSHW37d3H0xpv2+z9Ck=", doc1.Attachments()["attachment"].(map[string]interface{})["digest"])

		req := rt.SendAdminRequest("GET", "/{{.keyspace}}/doc1/attachment", "")
		assert.Equal(t, "attachmentB", string(req.BodyBytes()))
	})
}

// TestPushUnknownAttachmentAsStub sets revpos to an older generation, for an attachment that doesn't exist on the server.
// Verifies that getAttachment is triggered, and attachment is properly persisted.