Function TestRepositoryConcurrentAppend

internal/docsbatch/repository_test.go:151–211

(t *testing.T)

}

func TestRepositoryConcurrentAppend(t *testing.T) {
	t.Parallel()

	repository := testRepository(t, nil)

	state, err := repository.Create(State{
		Service:    ServiceDocs,
		DocumentID: "doc1",
		Account:    "user@example.com",
		Client:     "default",
	})
	if err != nil {
		t.Fatalf("Create: %v", err)
	}

	const appendCount = 20
	var wg sync.WaitGroup

	errs := make(chan error, appendCount)
	for index := range appendCount {
		wg.Add(1)

		go func() {
			defer wg.Done()

			request, marshalErr := json.Marshal(map[string]int{"index": index})
			if marshalErr != nil {
				errs <- marshalErr

				return
			}

			_, appendErr := repository.Append(AppendOptions{
				BatchID:    state.BatchID,
				Command:    "docs.insert",
				Identity:   testIdentity("doc1"),
				RevisionID: "rev1",
				Requests:   []json.RawMessage{request},
			})
			errs <- appendErr
		}()
	}

	wg.Wait()
	close(errs)

	for err := range errs {
		if err != nil {
			t.Fatalf("Append: %v", err)
		}
	}

	loaded, err := repository.Get(state.BatchID)
	if err != nil {
		t.Fatalf("Get: %v", err)
	}

	if len(loaded.Requests) != appendCount {