usage_test.go

// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

package metronome_test

import (
	"context"
	"errors"
	"os"
	"testing"
	"time"

	"github.com/Metronome-Industries/metronome-go"
	"github.com/Metronome-Industries/metronome-go/internal/testutil"
	"github.com/Metronome-Industries/metronome-go/option"
)

func TestUsageListWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := metronome.NewClient(
		option.WithBaseURL(baseURL),
		option.WithBearerToken("My Bearer Token"),
	)
	_, err := client.Usage.List(context.TODO(), metronome.UsageListParams{
		EndingBefore: metronome.F(time.Now()),
		StartingOn:   metronome.F(time.Now()),
		WindowSize:   metronome.F(metronome.UsageListParamsWindowSizeHour),
		NextPage:     metronome.F("next_page"),
		BillableMetrics: metronome.F([]metronome.UsageListParamsBillableMetric{{
			ID: metronome.F("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e"),
			GroupBy: metronome.F(metronome.UsageListParamsBillableMetricsGroupBy{
				Key:    metronome.F("key"),
				Values: metronome.F([]string{"x"}),
			}),
		}, {
			ID: metronome.F("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e"),
			GroupBy: metronome.F(metronome.UsageListParamsBillableMetricsGroupBy{
				Key:    metronome.F("key"),
				Values: metronome.F([]string{"x"}),
			}),
		}, {
			ID: metronome.F("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e"),
			GroupBy: metronome.F(metronome.UsageListParamsBillableMetricsGroupBy{
				Key:    metronome.F("key"),
				Values: metronome.F([]string{"x"}),
			}),
		}}),
		CustomerIDs: metronome.F([]string{"182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e", "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e", "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e"}),
	})
	if err != nil {
		var apierr *metronome.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestUsageIngest(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := metronome.NewClient(
		option.WithBaseURL(baseURL),
		option.WithBearerToken("My Bearer Token"),
	)
	err := client.Usage.Ingest(context.TODO(), metronome.UsageIngestParams{
		Usage: []metronome.UsageIngestParamsUsage{{
			CustomerID:    metronome.F("team@example.com"),
			EventType:     metronome.F("heartbeat"),
			Timestamp:     metronome.F("2021-01-01T00:00:00Z"),
			TransactionID: metronome.F("2021-01-01T00:00:00Z_cluster42"),
			Properties: metronome.F(map[string]interface{}{
				"cluster_id":  "bar",
				"cpu_seconds": "bar",
				"region":      "bar",
			}),
		}},
	})
	if err != nil {
		var apierr *metronome.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestUsageListWithGroupsWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := metronome.NewClient(
		option.WithBaseURL(baseURL),
		option.WithBearerToken("My Bearer Token"),
	)
	_, err := client.Usage.ListWithGroups(context.TODO(), metronome.UsageListWithGroupsParams{
		BillableMetricID: metronome.F("222796fd-d29c-429e-89b2-549fabda4ed6"),
		CustomerID:       metronome.F("04ca7e72-4229-4a6e-ab11-9f7376fccbcb"),
		WindowSize:       metronome.F(metronome.UsageListWithGroupsParamsWindowSizeHour),
		Limit:            metronome.F(int64(1)),
		NextPage:         metronome.F("next_page"),
		CurrentPeriod:    metronome.F(true),
		EndingBefore:     metronome.F(time.Now()),
		GroupBy: metronome.F(metronome.UsageListWithGroupsParamsGroupBy{
			Key:    metronome.F("region"),
			Values: metronome.F([]string{"US-East", "US-West", "EU-Central"}),
		}),
		StartingOn: metronome.F(time.Now()),
	})
	if err != nil {
		var apierr *metronome.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}