column_test.go

package honeycombio

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestColumns(t *testing.T) {
	ctx := context.Background()

	c := newTestClient(t)
	dataset := testDataset(t)

	var column *Column
	var err error

	t.Run("Create", func(t *testing.T) {
		data := &Column{
			KeyName:     "column_test",
			Hidden:      BoolPtr(false),
			Description: "This column is created by a test",
			Type:        ColumnTypePtr(ColumnTypeFloat),
		}
		column, err = c.Columns.Create(ctx, dataset, data)

		assert.NoError(t, err)

		data.ID = column.ID
		assert.Equal(t, data, column)
	})

	t.Run("List", func(t *testing.T) {
		columns, err := c.Columns.List(ctx, dataset)

		assert.NoError(t, err)
		assert.Contains(t, columns, *column, "could not find column with List")
	})

	t.Run("Get", func(t *testing.T) {
		result, err := c.Columns.Get(ctx, dataset, column.ID)

		assert.NoError(t, err)
		assert.Equal(t, *column, *result)
	})

	t.Run("GetByKeyName", func(t *testing.T) {
		result, err := c.Columns.GetByKeyName(ctx, dataset, "column_test")

		assert.NoError(t, err)
		assert.Equal(t, *column, *result)
	})

	t.Run("Update", func(t *testing.T) {
		// change all the fields to test
		data := &Column{
			ID:          column.ID,
			KeyName:     "a-new-keyname",
			Hidden:      BoolPtr(true),
			Description: "This is a new description",
			Type:        ColumnTypePtr(ColumnTypeBoolean),
		}
		column, err = c.Columns.Update(ctx, dataset, data)

		assert.NoError(t, err)

		data.ID = column.ID
		assert.Equal(t, data, column)
	})

	t.Run("Delete", func(t *testing.T) {
		err = c.Columns.Delete(ctx, dataset, column.ID)

		assert.NoError(t, err)
	})

	t.Run("Get_notFound", func(t *testing.T) {
		_, err := c.Columns.Get(ctx, dataset, column.ID)

		assert.Equal(t, ErrNotFound, err)
	})
}