Function TestGetTablesInfo

backend/plugins/gh-copilot/models/models_test.go:22–62

(t *testing.T)

import "testing"

func TestGetTablesInfo(t *testing.T) {
	tables := GetTablesInfo()
	expected := map[string]bool{
		(&GhCopilotConnection{}).TableName():      false,
		(&GhCopilotScope{}).TableName():           false,
		(&GhCopilotScopeConfig{}).TableName():     false,
		(&GhCopilotOrgMetrics{}).TableName():      false,
		(&GhCopilotLanguageMetrics{}).TableName(): false,
		(&GhCopilotEnterpriseDailyMetrics{}).TableName():      false,
		(&GhCopilotMetricsByIde{}).TableName():               false,
		(&GhCopilotMetricsByFeature{}).TableName():           false,
		(&GhCopilotMetricsByLanguageFeature{}).TableName():   false,
		(&GhCopilotMetricsByLanguageModel{}).TableName():     false,
		(&GhCopilotMetricsByModelFeature{}).TableName():      false,
		(&GhCopilotUserDailyMetrics{}).TableName():           false,
		(&GhCopilotUserMetricsByIde{}).TableName():           false,
		(&GhCopilotUserMetricsByFeature{}).TableName():       false,
		(&GhCopilotUserMetricsByLanguageFeature{}).TableName(): false,
		(&GhCopilotUserMetricsByLanguageModel{}).TableName(): false,
		(&GhCopilotUserMetricsByModelFeature{}).TableName():  false,
		(&GhCopilotSeat{}).TableName():            false,
	}

	if len(tables) != len(expected) {
		t.Fatalf("unexpected number of tables: want %d, got %d", len(expected), len(tables))
	}

	for _, table := range tables {
		tableName := table.TableName()
		if _, ok := expected[tableName]; !ok {
			t.Fatalf("unexpected table registered: %s", tableName)
		}
		expected[tableName] = true
	}

	for name, seen := range expected {
		if !seen {
			t.Fatalf("table not registered: %s", name)
		}
	}
}