Function TestEntityClientOperations

lib/gds/client_test.go:172–287

(t *testing.T)

}

func TestEntityClientOperations(t *testing.T) {
	ctx := context.Background()
	client, cleanup := getTestDatabase(ctx, t)
	defer cleanup()
	c := entityClient[TestSample]{client}
	// Step 1. Make sure the entity is not there yet.
	// Step 1a. Do Get
	entity, err := c.get(ctx, sampleKey, nameFilter{name: "a"})
	if entity != nil {
		t.Error("expected no entity")
	}
	if !errors.Is(err, ErrEntityNotFound) {
		t.Error("expected ErrEntityNotFound")
	}
	// Step 1b. Do List
	pageEmpty, nextPageToken, err := c.list(ctx, sampleKey, nil)
	if err != nil {
		t.Errorf("list query failed. %s", err.Error())
	}
	if nextPageToken == nil {
		t.Error("expected next page token")
	}
	if pageEmpty != nil {
		t.Error("expected empty page")
	}
	// Step 2. Insert the entities
	insertEntities(ctx, t, c)
	// Step 3. Get the entity
	entity, err = c.get(ctx, sampleKey, nameFilter{name: "a"})
	if err != nil {
		t.Errorf("expected error %s", err.Error())
	}
	if entity == nil {
		t.Error("expected entity")
		t.FailNow()
	}
	if !reflect.DeepEqual(*entity, TestSample{
		Name:      "a",
		Value:     new(0),
		CreatedAt: time.Date(2000, time.January, 1, 0, 0, 0, 0, time.UTC),
	}) {
		t.Errorf("values not equal. received %+v", *entity)
	}
	// Step 4. Upsert the entity
	entity.Value = new(200)
	// CreatedAt should not update due to the Mergeable policy
	entity.CreatedAt = time.Date(3000, time.March, 1, 0, 0, 0, 0, time.UTC)
	err = c.upsert(ctx, sampleKey, entity, testSampleMerge{}, nameFilter{name: "a"})
	if err != nil {
		t.Errorf("upsert failed %s", err.Error())
	}
	// Step 5. Get the updated entity
	entity, err = c.get(ctx, sampleKey, nameFilter{name: "a"})
	if err != nil {
		t.Errorf("expected error %s", err.Error())
	}
	if entity == nil {
		t.Error("expected entity")