Function seedChatDependencies

enterprise/coderd/x/chatd/chatd_test.go:106–141

seedChatDependencies creates a user, organization, and chat model config in the database for use in relay tests.

(
	t *testing.T,
	db database.Store,
)

// seedChatDependencies creates a user, organization, and chat model
// config in the database for use in relay tests.
func seedChatDependencies(
	t *testing.T,
	db database.Store,
) (database.User, database.Organization, database.ChatModelConfig) {
	t.Helper()

	safetyNet := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
		rw.Header().Set("Content-Type", "application/json")
		rw.WriteHeader(http.StatusInternalServerError)
		_, _ = rw.Write([]byte(`{"error":{"message":"unexpected OpenAI request in chatd relay test safety net"}}`))
	}))
	t.Cleanup(safetyNet.Close)

	user := dbgen.User(t, db, database.User{})
	org := dbgen.Organization(t, db, database.Organization{})
	dbgen.OrganizationMember(t, db, database.OrganizationMember{
		UserID:         user.ID,
		OrganizationID: org.ID,
	})
	provider := dbgen.AIProvider(t, db, database.AIProvider{
		Type:    database.AiProviderTypeOpenai,
		Name:    "test-" + uuid.NewString(),
		BaseUrl: safetyNet.URL,
	})
	dbgen.AIProviderKey(t, db, database.AIProviderKey{
		ProviderID: provider.ID,
	})
	model := dbgen.ChatModelConfig(t, db, database.ChatModelConfig{
		Provider:     "openai",
		AIProviderID: uuid.NullUUID{UUID: provider.ID, Valid: true},
		CreatedBy:    uuid.NullUUID{UUID: user.ID, Valid: true},
		UpdatedBy:    uuid.NullUUID{UUID: user.ID, Valid: true},
		IsDefault:    true,
	})
	return user, org, model
}

func seedWaitingChat(
	t *testing.T,