feat(billing): Update per-category on-demand for Logs (#95472)

Closes:
https://linear.app/getsentry/issue/BIL-1065/set-default-per-category-budget

Enables Logs to be tracked as a separate category in on-demand budgets

Author: jarrettscott

static/gsApp/types/index.tsx

@@ -244,6 +244,7 @@ export type PerCategoryOnDemandBudget = {
   errorsBudget: number;
   replaysBudget: number;
   transactionsBudget: number;
+  logBytesBudget?: number;
   monitorSeatsBudget?: number;
   profileDurationBudget?: number;
   profileDurationUIBudget?: number;

static/gsApp/views/amCheckout/steps/onDemandBudgets.spec.tsx

@@ -221,6 +221,7 @@ describe('OnDemandBudgets AM Checkout', function () {
             replaysBudget: 0,
             profileDurationBudget: 0,
             profileDurationUIBudget: 0,
+            logBytesBudget: 0,
             budgets: {
               errors: 1000,
               transactions: 2000,
@@ -230,6 +231,7 @@ describe('OnDemandBudgets AM Checkout', function () {
               uptime: 0,
               profileDuration: 0,
               profileDurationUI: 0,
+              logBytes: 0,
             },
           },
           onDemandMaxSpend: 10000,

static/gsApp/views/onDemandBudgets/utils.spec.tsx

@@ -5,9 +5,12 @@ import {
   SubscriptionFixture,
 } from 'getsentry-test/fixtures/subscription';
 
+import {DataCategory} from 'sentry/types/core';
+
 import {OnDemandBudgetMode, type OnDemandBudgets} from 'getsentry/types';
 import {
   exceedsInvoicedBudgetLimit,
+  getOnDemandBudget,
   getTotalBudget,
   parseOnDemandBudgetsFromSubscription,
 } from 'getsentry/views/onDemandBudgets/utils';
@@ -196,6 +199,7 @@ describe('parseOnDemandBudgetsFromSubscription', function () {
       replaysBudget: 0,
       profileDurationBudget: 0,
       profileDurationUIBudget: 0,
+      logBytesBudget: 0,
       budgets: {
         errors: 100,
         transactions: 200,
@@ -514,3 +518,69 @@ describe('exceedsInvoicedBudgetLimit', function () {
     expect(exceedsInvoicedBudgetLimit(subscription, ondemandBudget)).toBe(true);
   });
 });
+
+describe('getOnDemandBudget', function () {
+  it('returns 0 for category when in per-category mode without explicit budget', function () {
+    const budget: OnDemandBudgets = {
+      budgetMode: OnDemandBudgetMode.PER_CATEGORY,
+      errorsBudget: 100,
+      transactionsBudget: 200,
+      attachmentsBudget: 300,
+      replaysBudget: 0,
+      monitorSeatsBudget: 0,
+      profileDurationBudget: 0,
+      profileDurationUIBudget: 0,
+      uptimeBudget: 0,
+      logBytesBudget: 0,
+      budgets: {
+        errors: 100,
+        transactions: 200,
+        attachments: 300,
+        replays: 0,
+        monitorSeats: 0,
+        profileDuration: 0,
+        profileDurationUI: 0,
+        uptime: 0,
+      },
+    };
+
+    expect(getOnDemandBudget(budget, DataCategory.LOG_BYTE)).toBe(0);
+  });
+
+  it('returns correct value for LOG_BYTE category when in per-category mode with explicit budget', function () {
+    const budget: OnDemandBudgets = {
+      budgetMode: OnDemandBudgetMode.PER_CATEGORY,
+      errorsBudget: 100,
+      transactionsBudget: 200,
+      attachmentsBudget: 300,
+      replaysBudget: 0,
+      monitorSeatsBudget: 0,
+      profileDurationBudget: 0,
+      profileDurationUIBudget: 0,
+      uptimeBudget: 0,
+      logBytesBudget: 500,
+      budgets: {
+        errors: 100,
+        transactions: 200,
+        attachments: 300,
+        replays: 0,
+        monitorSeats: 0,
+        profileDuration: 0,
+        profileDurationUI: 0,
+        uptime: 0,
+        logBytes: 500,
+      },
+    };
+
+    expect(getOnDemandBudget(budget, DataCategory.LOG_BYTE)).toBe(500);
+  });
+
+  it('returns total budget for LOG_BYTE category when in shared mode', function () {
+    const budget: OnDemandBudgets = {
+      budgetMode: OnDemandBudgetMode.SHARED,
+      sharedMaxBudget: 1000,
+    };
+
+    expect(getOnDemandBudget(budget, DataCategory.LOG_BYTE)).toBe(1000);
+  });
+});

static/gsApp/views/onDemandBudgets/utils.tsx

@@ -63,6 +63,7 @@ export function parseOnDemandBudgets(
       profileDurationBudget: 0,
       profileDurationUIBudget: 0,
       uptimeBudget: 0,
+      logBytesBudget: 0,
       // Spread the calculated values over the defaults
       ...categoryBudgets,
       budgets: parsedBudgets,
@@ -150,10 +151,7 @@ function getBudgetMode(budget: OnDemandBudgets) {
 }
 
 export function getOnDemandBudget(budget: OnDemandBudgets, dataCategory: DataCategory) {
-  if (
-    budget.budgetMode === OnDemandBudgetMode.PER_CATEGORY &&
-    dataCategory in budget.budgets
-  ) {
+  if (budget.budgetMode === OnDemandBudgetMode.PER_CATEGORY) {
     return budget.budgets[dataCategory] ?? 0;
   }
   return getTotalBudget(budget);
@@ -256,6 +254,7 @@ export function convertOnDemandBudget(
       uptime: 0,
       profileDuration: 0,
       profileDurationUI: 0,
+      logBytes: 0,
     };
 
     if (currentOnDemandBudget.budgetMode === OnDemandBudgetMode.PER_CATEGORY) {
@@ -288,6 +287,7 @@ export function convertOnDemandBudget(
       uptimeBudget: 0,
       profileDurationBudget: 0,
       profileDurationUIBudget: 0,
+      logBytesBudget: 0,
       // Spread the calculated values over the defaults
       ...categoryBudgets,
       budgets: newBudgets,