Function expectUpsertTaskWorkspaceApp

coderd/wsbuilder/wsbuilder_test.go:1630–1648

(
	assertions func(database.UpsertTaskWorkspaceAppParams),
)

}

func expectUpsertTaskWorkspaceApp(
	assertions func(database.UpsertTaskWorkspaceAppParams),
) func(mTx *dbmock.MockStore) {
	return func(mTx *dbmock.MockStore) {
		mTx.EXPECT().UpsertTaskWorkspaceApp(gomock.Any(), gomock.Any()).
			Times(1).
			DoAndReturn(
				func(ctx context.Context, params database.UpsertTaskWorkspaceAppParams) (database.TaskWorkspaceApp, error) {
					assertions(params)
					return database.TaskWorkspaceApp{
						TaskID:               params.TaskID,
						WorkspaceBuildNumber: params.WorkspaceBuildNumber,
						WorkspaceAgentID:     params.WorkspaceAgentID,
						WorkspaceAppID:       params.WorkspaceAppID,
					}, nil
				},
			)
	}
}