Function TupleWritingAndReadingTest

pkg/storage/test/tuples.go:624–1788

(t *testing.T, datastore storage.OpenFGADatastore)

}

func TupleWritingAndReadingTest(t *testing.T, datastore storage.OpenFGADatastore) {
	ctx := context.Background()

	t.Run("lots_of_writes_and_read_returns_everything", func(t *testing.T) {
		storeID := ulid.Make().String()

		var writtenTuples []*openfgav1.TupleKey
		for i := 0; i < storage.DefaultPageSize*50; i++ {
			newTuple := tuple.NewTupleKey(fmt.Sprintf("document:%d", i), "viewer", "user:jon")
			err := datastore.Write(context.Background(), storeID, nil, []*openfgav1.TupleKey{newTuple})
			require.NoError(t, err)
			writtenTuples = append(writtenTuples, newTuple)
		}

		t.Run("read_returns_everything", func(t *testing.T) {
			tupleIterator, err := datastore.Read(ctx, storeID, storage.ReadFilter{}, storage.ReadOptions{})
			require.NoError(t, err)
			defer tupleIterator.Stop()

			seenTuples := iterateThroughAllTuples(t, tupleIterator)
			if diff := cmp.Diff(writtenTuples, seenTuples, cmpSortTupleKeys...); diff != "" {
				t.Fatalf("mismatch (-want +got):\n%s", diff)
			}
		})

		t.Run("read_page_size_1_returns_everything", func(t *testing.T) {
			seenTuples := testutils.ConvertTuplesToTupleKeys(readWithPageSize(t, datastore, storeID, 1, storage.ReadFilter{}))
			if diff := cmp.Diff(writtenTuples, seenTuples, cmpSortTupleKeys...); diff != "" {
				t.Fatalf("mismatch (-want +got):\n%s", diff)
			}
		})

		t.Run("read_page_size_2_returns_everything", func(t *testing.T) {
			seenTuples := testutils.ConvertTuplesToTupleKeys(readWithPageSize(t, datastore, storeID, 2, storage.ReadFilter{}))
			if diff := cmp.Diff(writtenTuples, seenTuples, cmpSortTupleKeys...); diff != "" {
				t.Fatalf("mismatch (-want +got):\n%s", diff)
			}
		})

		t.Run("read_page_size_default_returns_everything", func(t *testing.T) {
			seenTuples := testutils.ConvertTuplesToTupleKeys(readWithPageSize(t, datastore, storeID, storage.DefaultPageSize, storage.ReadFilter{}))
			if diff := cmp.Diff(writtenTuples, seenTuples, cmpSortTupleKeys...); diff != "" {
				t.Fatalf("mismatch (-want +got):\n%s", diff)
			}
		})

		t.Run("read_page_size_infinite_returns_everything", func(t *testing.T) {
			seenTuples := testutils.ConvertTuplesToTupleKeys(readWithPageSize(t, datastore, storeID, storage.DefaultPageSize*50000, storage.ReadFilter{}))
			if diff := cmp.Diff(writtenTuples, seenTuples, cmpSortTupleKeys...); diff != "" {
				t.Fatalf("mismatch (-want +got):\n%s", diff)
			}
		})
	})

	t.Run("deletes_would_succeed_and_write_would_fail,_fails_and_introduces_no_changes", func(t *testing.T) {
		storeID := ulid.Make().String()
		tks := []*openfgav1.TupleKey{
			{