Files
dingtalk-workspace-cli/internal/cli/fortest.go
T
玉澜 ea43db1d64 refactor(schema): drop cli shim packages, add testseam swap, consolidate ForTest helpers
- import 统一:删除 cli/contractfinal 与 cli/runtimeannotate 垫片包,
  26 个消费文件一律直引 corecmd/*;cli 根仅保留 runtime_schema_seam.go
  包内别名,依赖图保持单向无环
- 新增 internal/testseam.Swap[T]:包级 var 注入缝置换由 t.Cleanup
  结构性恢复;迁移 pipeline*/stdinIsTerminalFn/loadReviewedCommandRegistry/
  schemaCommandCatalogError/schemaParameterBindingData/finalSchemaAgentMetadata
  六组核心缝(26+ 处)
- ForTest 辅助归拢到 per-package fortest.go(corecmd/contract、
  corecmd/contractfinal、shortcut、cli),生产文件只留真逻辑
- 文档同步:runtime_schema_seam.go / runtimeannotate/doc.go 注释、
  CHANGELOG、RFC §278-279、AGENTS.md
2026-08-03 00:50:16 +08:00

104 lines
3.7 KiB
Go

// Copyright 2026 Alibaba Group
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package cli
import (
"sync"
"github.com/spf13/cobra"
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/corecmd/contract"
)
// Test helpers for this package's lazy delivery state. Production code must
// not call these; the ForTest suffix is the boundary. Cross-package tests use
// the exported ones; in-package tests may use the unexported resetters.
func resetDeliverySchemaCatalogStateForTest() {
runtimeDeliverySchemaCatalogOnce = sync.Once{}
runtimeDeliverySchemaCatalog = loadedSchemaCatalog{}
runtimeDeliverySchemaCatalogErr = nil
runtimeDeliverySchemaCatalogLazyCount.Store(0)
runtimeDeliverySchemaCatalogMapsOnce = sync.Once{}
runtimeDeliverySchemaCatalogMapsErr = nil
}
func resetMetaByCLIPathStateForTest() {
metaByCLIPathOnce = sync.Once{}
metaByCLIPath = nil
runtimeDeliverySchemaMetaIndexErr = nil
runtimeDeliverySchemaMetaIndexLazyCount.Store(0)
resetDeliverySchemaCatalogStateForTest()
}
// clearDeclaredDryRunCapabilitiesForTest resets the declared index (tests only).
func clearDeclaredDryRunCapabilitiesForTest() {
declaredDryRunCapabilities.Range(func(key, _ any) bool {
declaredDryRunCapabilities.Delete(key)
return true
})
}
func resetReviewedDryRunCapabilitiesLazyForTest() {
reviewedDryRunCapabilitiesLazy = struct {
once sync.Once
byCanonical map[string]contract.DryRunSpec
err error
}{}
}
// InstallProductionSchemaAssemblyForTest installs the production
// assembleSchemaCatalogFromRoot delivery path with factory and clears lazy
// caches. Tests must Cleanup via RegisterSchemaSourceRoot(nil) or a restore
// helper that reinstalls their prior delivery stub.
func InstallProductionSchemaAssemblyForTest(factory func() *cobra.Command) {
RegisterSchemaSourceRoot(factory)
assembleDeliverySchemaCatalogFn = assembleSchemaCatalogFromRoot
resetDeliverySchemaCatalogStateForTest()
}
// MaterializeDeliverySchemaCatalogMapsForTest exercises the lazy Catalog/Tools
// materialize path used by map-based consumers.
func MaterializeDeliverySchemaCatalogMapsForTest() (sourceHash string, err error) {
loaded, err := materializeDeliverySchemaCatalogMaps()
if err != nil {
return "", err
}
return loaded.Snapshot.SourceHash, nil
}
// DeliverySchemaAllPayloadForTest returns schema --all through the installed
// delivery loader (catalog_hash comes from Snapshot.SourceHash).
func DeliverySchemaAllPayloadForTest() (map[string]any, error) {
return deliverySchemaAllPayload()
}
// RestorePackageCLISchemaDeliveryForTest reinstalls the package-cli TestMain
// assembled-delivery stub after a production-assembly exercise. Outside package
// cli TestMain it clears the factory and resets lazy delivery state.
func RestorePackageCLISchemaDeliveryForTest() {
if restorePackageCLISchemaDeliveryHook != nil {
restorePackageCLISchemaDeliveryHook()
return
}
schemaSourceRootFn = nil
assembleDeliverySchemaCatalogFn = assembleSchemaCatalogFromRoot
resetDeliverySchemaCatalogStateForTest()
resetMetaByCLIPathStateForTest()
}
// restorePackageCLISchemaDeliveryHook is installed by package-cli TestMain so
// external cli_test helpers can restore the assembled-delivery stub.
var restorePackageCLISchemaDeliveryHook func()