Function TestBatchUpdateMetadata

coderd/agentapi/metadata_test.go:25–285

(t *testing.T)

)

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

	agent := database.WorkspaceAgent{
		ID: uuid.New(),
	}

	t.Run("OK", func(t *testing.T) {
		t.Parallel()

		ctx := testutil.Context(t, testutil.WaitShort)

		ctrl := gomock.NewController(t)
		store := dbmock.NewMockStore(ctrl)
		ps := pubsub.NewInMemory()
		reg := prometheus.NewRegistry()

		now := dbtime.Now()
		req := &agentproto.BatchUpdateMetadataRequest{
			Metadata: []*agentproto.Metadata{
				{
					Key: "awesome key",
					Result: &agentproto.WorkspaceAgentMetadata_Result{
						CollectedAt: timestamppb.New(now.Add(-10 * time.Second)),
						Age:         10,
						Value:       "awesome value",
						Error:       "",
					},
				},
				{
					Key: "uncool key",
					Result: &agentproto.WorkspaceAgentMetadata_Result{
						CollectedAt: timestamppb.New(now.Add(-3 * time.Second)),
						Age:         3,
						Value:       "",
						Error:       "\t uncool error  ",
					},
				},
			},
		}
		batchSize := len(req.Metadata)
		// This test sends 2 metadata entries (one clean, one with
		// whitespace padding). With batch size 2 we expect exactly
		// 1 capacity flush. The matcher verifies that stored values
		// are trimmed while clean values pass through unchanged.
		expectedValues := map[string]string{
			"awesome key": "awesome value",
			"uncool key":  "",
		}
		expectedErrors := map[string]string{
			"awesome key": "",
			"uncool key":  "uncool error",
		}
		store.EXPECT().
			BatchUpdateWorkspaceAgentMetadata(
				gomock.Any(),
				gomock.Cond(func(arg database.BatchUpdateWorkspaceAgentMetadataParams) bool {
					if len(arg.Key) != len(expectedValues) {