Add test ID to data table columns

[jacob-carlborg-apoex]

I would like to add a test ID (that is a custom attribute like data-test="foo") to the columns of v-data-table(-server). I want to have this in all my tables. I can use the [item.${string}] slot to do this. So I thought I would create a new component that wraps the data table component. I came up with this:

<template>
  <v-data-table v-bind="$attrs">
    <template v-for="(_, name) in $slots" #[name]="slotData">
      <slot :name="name" v-bind="slotData || {}"></slot>
    </template>

    <template v-for="{column, slot} in columns" #[slot]="props" :key="column">
      <span :data-test="column">
        <slot v-if="$slots[slot]" :name="slot" v-bind="props"></slot>
        <template v-else>
          {{ props.item.raw[column] }}
        </template>
      </span>
    </template>
  </v-data-table>
</template>

<script lang="ts" setup>
import {computed, useAttrs} from 'vue'

interface DataTableHeader {
  key: string
}

function headers(): DataTableHeader[] {
  return attrs['headers'] as DataTableHeader[]
}

const attrs = useAttrs()
const columns = computed(() =>
  headers().map(({key: column}: {key: string}) => ({
    column,
    slot: `item.${column}`
  }))
)
</script>

<script lang="ts">
export default {
  inheritAttrs: false
}
</script>

This code snippet adds the data-test attribute only to the columns of the regular rows and not to the header row. If I use the same technique to add support for the header columns as well I loose some things, like icon that indicates the sort order.

Is there a better way to do this so that I don't loose any functionality or styling? Or is this worth an enhancement request?