mirror of
https://github.com/pacnpal/Roo-Code.git
synced 2025-12-21 04:41:16 -05:00
Clean up the settings page
This commit is contained in:
@@ -64,7 +64,7 @@ export class ConfigManager {
|
||||
/**
|
||||
* List all available configs with metadata
|
||||
*/
|
||||
async ListConfig(): Promise<ApiConfigMeta[]> {
|
||||
async listConfig(): Promise<ApiConfigMeta[]> {
|
||||
try {
|
||||
const config = await this.readConfig()
|
||||
return Object.entries(config.apiConfigs).map(([name, apiConfig]) => ({
|
||||
@@ -80,7 +80,7 @@ export class ConfigManager {
|
||||
/**
|
||||
* Save a config with the given name
|
||||
*/
|
||||
async SaveConfig(name: string, config: ApiConfiguration): Promise<void> {
|
||||
async saveConfig(name: string, config: ApiConfiguration): Promise<void> {
|
||||
try {
|
||||
const currentConfig = await this.readConfig()
|
||||
const existingConfig = currentConfig.apiConfigs[name]
|
||||
@@ -97,7 +97,7 @@ export class ConfigManager {
|
||||
/**
|
||||
* Load a config by name
|
||||
*/
|
||||
async LoadConfig(name: string): Promise<ApiConfiguration> {
|
||||
async loadConfig(name: string): Promise<ApiConfiguration> {
|
||||
try {
|
||||
const config = await this.readConfig()
|
||||
const apiConfig = config.apiConfigs[name]
|
||||
@@ -118,7 +118,7 @@ export class ConfigManager {
|
||||
/**
|
||||
* Delete a config by name
|
||||
*/
|
||||
async DeleteConfig(name: string): Promise<void> {
|
||||
async deleteConfig(name: string): Promise<void> {
|
||||
try {
|
||||
const currentConfig = await this.readConfig()
|
||||
if (!currentConfig.apiConfigs[name]) {
|
||||
@@ -140,7 +140,7 @@ export class ConfigManager {
|
||||
/**
|
||||
* Set the current active API configuration
|
||||
*/
|
||||
async SetCurrentConfig(name: string): Promise<void> {
|
||||
async setCurrentConfig(name: string): Promise<void> {
|
||||
try {
|
||||
const currentConfig = await this.readConfig()
|
||||
if (!currentConfig.apiConfigs[name]) {
|
||||
@@ -157,7 +157,7 @@ export class ConfigManager {
|
||||
/**
|
||||
* Check if a config exists by name
|
||||
*/
|
||||
async HasConfig(name: string): Promise<boolean> {
|
||||
async hasConfig(name: string): Promise<boolean> {
|
||||
try {
|
||||
const config = await this.readConfig()
|
||||
return name in config.apiConfigs
|
||||
@@ -169,7 +169,7 @@ export class ConfigManager {
|
||||
/**
|
||||
* Set the API config for a specific mode
|
||||
*/
|
||||
async SetModeConfig(mode: Mode, configId: string): Promise<void> {
|
||||
async setModeConfig(mode: Mode, configId: string): Promise<void> {
|
||||
try {
|
||||
const currentConfig = await this.readConfig()
|
||||
if (!currentConfig.modeApiConfigs) {
|
||||
@@ -185,7 +185,7 @@ export class ConfigManager {
|
||||
/**
|
||||
* Get the API config ID for a specific mode
|
||||
*/
|
||||
async GetModeConfigId(mode: Mode): Promise<string | undefined> {
|
||||
async getModeConfigId(mode: Mode): Promise<string | undefined> {
|
||||
try {
|
||||
const config = await this.readConfig()
|
||||
return config.modeApiConfigs?.[mode]
|
||||
@@ -194,10 +194,23 @@ export class ConfigManager {
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the key used for storing config in secrets
|
||||
*/
|
||||
private getConfigKey(): string {
|
||||
return `${this.SCOPE_PREFIX}api_config`
|
||||
}
|
||||
|
||||
/**
|
||||
* Reset all configuration by deleting the stored config from secrets
|
||||
*/
|
||||
public async resetAllConfigs(): Promise<void> {
|
||||
await this.context.secrets.delete(this.getConfigKey())
|
||||
}
|
||||
|
||||
private async readConfig(): Promise<ApiConfigData> {
|
||||
try {
|
||||
const configKey = `${this.SCOPE_PREFIX}api_config`
|
||||
const content = await this.context.secrets.get(configKey)
|
||||
const content = await this.context.secrets.get(this.getConfigKey())
|
||||
|
||||
if (!content) {
|
||||
return this.defaultConfig
|
||||
@@ -211,9 +224,8 @@ export class ConfigManager {
|
||||
|
||||
private async writeConfig(config: ApiConfigData): Promise<void> {
|
||||
try {
|
||||
const configKey = `${this.SCOPE_PREFIX}api_config`
|
||||
const content = JSON.stringify(config, null, 2)
|
||||
await this.context.secrets.store(configKey, content)
|
||||
await this.context.secrets.store(this.getConfigKey(), content)
|
||||
} catch (error) {
|
||||
throw new Error(`Failed to write config to secrets: ${error}`)
|
||||
}
|
||||
|
||||
@@ -106,7 +106,7 @@ describe("ConfigManager", () => {
|
||||
|
||||
mockSecrets.get.mockResolvedValue(JSON.stringify(existingConfig))
|
||||
|
||||
const configs = await configManager.ListConfig()
|
||||
const configs = await configManager.listConfig()
|
||||
expect(configs).toEqual([
|
||||
{ name: "default", id: "default", apiProvider: undefined },
|
||||
{ name: "test", id: "test-id", apiProvider: "anthropic" },
|
||||
@@ -126,14 +126,14 @@ describe("ConfigManager", () => {
|
||||
|
||||
mockSecrets.get.mockResolvedValue(JSON.stringify(emptyConfig))
|
||||
|
||||
const configs = await configManager.ListConfig()
|
||||
const configs = await configManager.listConfig()
|
||||
expect(configs).toEqual([])
|
||||
})
|
||||
|
||||
it("should throw error if reading from secrets fails", async () => {
|
||||
mockSecrets.get.mockRejectedValue(new Error("Read failed"))
|
||||
|
||||
await expect(configManager.ListConfig()).rejects.toThrow(
|
||||
await expect(configManager.listConfig()).rejects.toThrow(
|
||||
"Failed to list configs: Error: Failed to read config from secrets: Error: Read failed",
|
||||
)
|
||||
})
|
||||
@@ -160,7 +160,7 @@ describe("ConfigManager", () => {
|
||||
apiKey: "test-key",
|
||||
}
|
||||
|
||||
await configManager.SaveConfig("test", newConfig)
|
||||
await configManager.saveConfig("test", newConfig)
|
||||
|
||||
// Get the actual stored config to check the generated ID
|
||||
const storedConfig = JSON.parse(mockSecrets.store.mock.calls[0][1])
|
||||
@@ -207,7 +207,7 @@ describe("ConfigManager", () => {
|
||||
apiKey: "new-key",
|
||||
}
|
||||
|
||||
await configManager.SaveConfig("test", updatedConfig)
|
||||
await configManager.saveConfig("test", updatedConfig)
|
||||
|
||||
const expectedConfig = {
|
||||
currentApiConfigName: "default",
|
||||
@@ -235,7 +235,7 @@ describe("ConfigManager", () => {
|
||||
)
|
||||
mockSecrets.store.mockRejectedValueOnce(new Error("Storage failed"))
|
||||
|
||||
await expect(configManager.SaveConfig("test", {})).rejects.toThrow(
|
||||
await expect(configManager.saveConfig("test", {})).rejects.toThrow(
|
||||
"Failed to save config: Error: Failed to write config to secrets: Error: Storage failed",
|
||||
)
|
||||
})
|
||||
@@ -258,7 +258,7 @@ describe("ConfigManager", () => {
|
||||
|
||||
mockSecrets.get.mockResolvedValue(JSON.stringify(existingConfig))
|
||||
|
||||
await configManager.DeleteConfig("test")
|
||||
await configManager.deleteConfig("test")
|
||||
|
||||
// Get the stored config to check the ID
|
||||
const storedConfig = JSON.parse(mockSecrets.store.mock.calls[0][1])
|
||||
@@ -275,7 +275,7 @@ describe("ConfigManager", () => {
|
||||
}),
|
||||
)
|
||||
|
||||
await expect(configManager.DeleteConfig("nonexistent")).rejects.toThrow("Config 'nonexistent' not found")
|
||||
await expect(configManager.deleteConfig("nonexistent")).rejects.toThrow("Config 'nonexistent' not found")
|
||||
})
|
||||
|
||||
it("should throw error when trying to delete last remaining config", async () => {
|
||||
@@ -290,7 +290,7 @@ describe("ConfigManager", () => {
|
||||
}),
|
||||
)
|
||||
|
||||
await expect(configManager.DeleteConfig("default")).rejects.toThrow(
|
||||
await expect(configManager.deleteConfig("default")).rejects.toThrow(
|
||||
"Cannot delete the last remaining configuration.",
|
||||
)
|
||||
})
|
||||
@@ -311,7 +311,7 @@ describe("ConfigManager", () => {
|
||||
|
||||
mockSecrets.get.mockResolvedValue(JSON.stringify(existingConfig))
|
||||
|
||||
const config = await configManager.LoadConfig("test")
|
||||
const config = await configManager.loadConfig("test")
|
||||
|
||||
expect(config).toEqual({
|
||||
apiProvider: "anthropic",
|
||||
@@ -342,7 +342,7 @@ describe("ConfigManager", () => {
|
||||
}),
|
||||
)
|
||||
|
||||
await expect(configManager.LoadConfig("nonexistent")).rejects.toThrow("Config 'nonexistent' not found")
|
||||
await expect(configManager.loadConfig("nonexistent")).rejects.toThrow("Config 'nonexistent' not found")
|
||||
})
|
||||
|
||||
it("should throw error if secrets storage fails", async () => {
|
||||
@@ -361,7 +361,7 @@ describe("ConfigManager", () => {
|
||||
)
|
||||
mockSecrets.store.mockRejectedValueOnce(new Error("Storage failed"))
|
||||
|
||||
await expect(configManager.LoadConfig("test")).rejects.toThrow(
|
||||
await expect(configManager.loadConfig("test")).rejects.toThrow(
|
||||
"Failed to load config: Error: Failed to write config to secrets: Error: Storage failed",
|
||||
)
|
||||
})
|
||||
@@ -384,7 +384,7 @@ describe("ConfigManager", () => {
|
||||
|
||||
mockSecrets.get.mockResolvedValue(JSON.stringify(existingConfig))
|
||||
|
||||
await configManager.SetCurrentConfig("test")
|
||||
await configManager.setCurrentConfig("test")
|
||||
|
||||
// Get the stored config to check the structure
|
||||
const storedConfig = JSON.parse(mockSecrets.store.mock.calls[0][1])
|
||||
@@ -404,7 +404,7 @@ describe("ConfigManager", () => {
|
||||
}),
|
||||
)
|
||||
|
||||
await expect(configManager.SetCurrentConfig("nonexistent")).rejects.toThrow(
|
||||
await expect(configManager.setCurrentConfig("nonexistent")).rejects.toThrow(
|
||||
"Config 'nonexistent' not found",
|
||||
)
|
||||
})
|
||||
@@ -420,12 +420,34 @@ describe("ConfigManager", () => {
|
||||
)
|
||||
mockSecrets.store.mockRejectedValueOnce(new Error("Storage failed"))
|
||||
|
||||
await expect(configManager.SetCurrentConfig("test")).rejects.toThrow(
|
||||
await expect(configManager.setCurrentConfig("test")).rejects.toThrow(
|
||||
"Failed to set current config: Error: Failed to write config to secrets: Error: Storage failed",
|
||||
)
|
||||
})
|
||||
})
|
||||
|
||||
describe("ResetAllConfigs", () => {
|
||||
it("should delete all stored configs", async () => {
|
||||
// Setup initial config
|
||||
mockSecrets.get.mockResolvedValue(
|
||||
JSON.stringify({
|
||||
currentApiConfigName: "test",
|
||||
apiConfigs: {
|
||||
test: {
|
||||
apiProvider: "anthropic",
|
||||
id: "test-id",
|
||||
},
|
||||
},
|
||||
}),
|
||||
)
|
||||
|
||||
await configManager.resetAllConfigs()
|
||||
|
||||
// Should have called delete with the correct config key
|
||||
expect(mockSecrets.delete).toHaveBeenCalledWith("roo_cline_config_api_config")
|
||||
})
|
||||
})
|
||||
|
||||
describe("HasConfig", () => {
|
||||
it("should return true for existing config", async () => {
|
||||
const existingConfig: ApiConfigData = {
|
||||
@@ -443,7 +465,7 @@ describe("ConfigManager", () => {
|
||||
|
||||
mockSecrets.get.mockResolvedValue(JSON.stringify(existingConfig))
|
||||
|
||||
const hasConfig = await configManager.HasConfig("test")
|
||||
const hasConfig = await configManager.hasConfig("test")
|
||||
expect(hasConfig).toBe(true)
|
||||
})
|
||||
|
||||
@@ -455,14 +477,14 @@ describe("ConfigManager", () => {
|
||||
}),
|
||||
)
|
||||
|
||||
const hasConfig = await configManager.HasConfig("nonexistent")
|
||||
const hasConfig = await configManager.hasConfig("nonexistent")
|
||||
expect(hasConfig).toBe(false)
|
||||
})
|
||||
|
||||
it("should throw error if secrets storage fails", async () => {
|
||||
mockSecrets.get.mockRejectedValue(new Error("Storage failed"))
|
||||
|
||||
await expect(configManager.HasConfig("test")).rejects.toThrow(
|
||||
await expect(configManager.hasConfig("test")).rejects.toThrow(
|
||||
"Failed to check config existence: Error: Failed to read config from secrets: Error: Storage failed",
|
||||
)
|
||||
})
|
||||
|
||||
@@ -446,7 +446,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
})
|
||||
|
||||
this.configManager
|
||||
.ListConfig()
|
||||
.listConfig()
|
||||
.then(async (listApiConfig) => {
|
||||
if (!listApiConfig) {
|
||||
return
|
||||
@@ -456,7 +456,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
// check if first time init then sync with exist config
|
||||
if (!checkExistKey(listApiConfig[0])) {
|
||||
const { apiConfiguration } = await this.getState()
|
||||
await this.configManager.SaveConfig(
|
||||
await this.configManager.saveConfig(
|
||||
listApiConfig[0].name ?? "default",
|
||||
apiConfiguration,
|
||||
)
|
||||
@@ -467,11 +467,11 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
let currentConfigName = (await this.getGlobalState("currentApiConfigName")) as string
|
||||
|
||||
if (currentConfigName) {
|
||||
if (!(await this.configManager.HasConfig(currentConfigName))) {
|
||||
if (!(await this.configManager.hasConfig(currentConfigName))) {
|
||||
// current config name not valid, get first config in list
|
||||
await this.updateGlobalState("currentApiConfigName", listApiConfig?.[0]?.name)
|
||||
if (listApiConfig?.[0]?.name) {
|
||||
const apiConfig = await this.configManager.LoadConfig(
|
||||
const apiConfig = await this.configManager.loadConfig(
|
||||
listApiConfig?.[0]?.name,
|
||||
)
|
||||
|
||||
@@ -726,8 +726,8 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
await this.updateGlobalState("mode", newMode)
|
||||
|
||||
// Load the saved API config for the new mode if it exists
|
||||
const savedConfigId = await this.configManager.GetModeConfigId(newMode)
|
||||
const listApiConfig = await this.configManager.ListConfig()
|
||||
const savedConfigId = await this.configManager.getModeConfigId(newMode)
|
||||
const listApiConfig = await this.configManager.listConfig()
|
||||
|
||||
// Update listApiConfigMeta first to ensure UI has latest data
|
||||
await this.updateGlobalState("listApiConfigMeta", listApiConfig)
|
||||
@@ -736,7 +736,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
if (savedConfigId) {
|
||||
const config = listApiConfig?.find((c) => c.id === savedConfigId)
|
||||
if (config?.name) {
|
||||
const apiConfig = await this.configManager.LoadConfig(config.name)
|
||||
const apiConfig = await this.configManager.loadConfig(config.name)
|
||||
await Promise.all([
|
||||
this.updateGlobalState("currentApiConfigName", config.name),
|
||||
this.updateApiConfiguration(apiConfig),
|
||||
@@ -748,7 +748,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
if (currentApiConfigName) {
|
||||
const config = listApiConfig?.find((c) => c.name === currentApiConfigName)
|
||||
if (config?.id) {
|
||||
await this.configManager.SetModeConfig(newMode, config.id)
|
||||
await this.configManager.setModeConfig(newMode, config.id)
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -913,7 +913,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
if (enhancementApiConfigId) {
|
||||
const config = listApiConfigMeta?.find((c) => c.id === enhancementApiConfigId)
|
||||
if (config?.name) {
|
||||
const loadedConfig = await this.configManager.LoadConfig(config.name)
|
||||
const loadedConfig = await this.configManager.loadConfig(config.name)
|
||||
if (loadedConfig.apiProvider) {
|
||||
configToUse = loadedConfig
|
||||
}
|
||||
@@ -1004,8 +1004,8 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
case "upsertApiConfiguration":
|
||||
if (message.text && message.apiConfiguration) {
|
||||
try {
|
||||
await this.configManager.SaveConfig(message.text, message.apiConfiguration)
|
||||
let listApiConfig = await this.configManager.ListConfig()
|
||||
await this.configManager.saveConfig(message.text, message.apiConfiguration)
|
||||
let listApiConfig = await this.configManager.listConfig()
|
||||
|
||||
await Promise.all([
|
||||
this.updateGlobalState("listApiConfigMeta", listApiConfig),
|
||||
@@ -1025,10 +1025,10 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
try {
|
||||
const { oldName, newName } = message.values
|
||||
|
||||
await this.configManager.SaveConfig(newName, message.apiConfiguration)
|
||||
await this.configManager.DeleteConfig(oldName)
|
||||
await this.configManager.saveConfig(newName, message.apiConfiguration)
|
||||
await this.configManager.deleteConfig(oldName)
|
||||
|
||||
let listApiConfig = await this.configManager.ListConfig()
|
||||
let listApiConfig = await this.configManager.listConfig()
|
||||
const config = listApiConfig?.find((c) => c.name === newName)
|
||||
|
||||
// Update listApiConfigMeta first to ensure UI has latest data
|
||||
@@ -1046,8 +1046,8 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
case "loadApiConfiguration":
|
||||
if (message.text) {
|
||||
try {
|
||||
const apiConfig = await this.configManager.LoadConfig(message.text)
|
||||
const listApiConfig = await this.configManager.ListConfig()
|
||||
const apiConfig = await this.configManager.loadConfig(message.text)
|
||||
const listApiConfig = await this.configManager.listConfig()
|
||||
|
||||
await Promise.all([
|
||||
this.updateGlobalState("listApiConfigMeta", listApiConfig),
|
||||
@@ -1075,8 +1075,8 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
}
|
||||
|
||||
try {
|
||||
await this.configManager.DeleteConfig(message.text)
|
||||
const listApiConfig = await this.configManager.ListConfig()
|
||||
await this.configManager.deleteConfig(message.text)
|
||||
const listApiConfig = await this.configManager.listConfig()
|
||||
|
||||
// Update listApiConfigMeta first to ensure UI has latest data
|
||||
await this.updateGlobalState("listApiConfigMeta", listApiConfig)
|
||||
@@ -1084,7 +1084,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
// If this was the current config, switch to first available
|
||||
let currentApiConfigName = await this.getGlobalState("currentApiConfigName")
|
||||
if (message.text === currentApiConfigName && listApiConfig?.[0]?.name) {
|
||||
const apiConfig = await this.configManager.LoadConfig(listApiConfig[0].name)
|
||||
const apiConfig = await this.configManager.loadConfig(listApiConfig[0].name)
|
||||
await Promise.all([
|
||||
this.updateGlobalState("currentApiConfigName", listApiConfig[0].name),
|
||||
this.updateApiConfiguration(apiConfig),
|
||||
@@ -1100,7 +1100,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
break
|
||||
case "getListApiConfiguration":
|
||||
try {
|
||||
let listApiConfig = await this.configManager.ListConfig()
|
||||
let listApiConfig = await this.configManager.listConfig()
|
||||
await this.updateGlobalState("listApiConfigMeta", listApiConfig)
|
||||
this.postMessageToWebview({ type: "listApiConfig", listApiConfig })
|
||||
} catch (error) {
|
||||
@@ -1127,10 +1127,10 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
const { mode } = await this.getState()
|
||||
if (mode) {
|
||||
const currentApiConfigName = await this.getGlobalState("currentApiConfigName")
|
||||
const listApiConfig = await this.configManager.ListConfig()
|
||||
const listApiConfig = await this.configManager.listConfig()
|
||||
const config = listApiConfig?.find((c) => c.name === currentApiConfigName)
|
||||
if (config?.id) {
|
||||
await this.configManager.SetModeConfig(mode, config.id)
|
||||
await this.configManager.setModeConfig(mode, config.id)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -2077,7 +2077,16 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
// dev
|
||||
|
||||
async resetState() {
|
||||
vscode.window.showInformationMessage("Resetting state...")
|
||||
const answer = await vscode.window.showInformationMessage(
|
||||
"Are you sure you want to reset all state and secret storage in the extension? This cannot be undone.",
|
||||
{ modal: true },
|
||||
"Yes",
|
||||
)
|
||||
|
||||
if (answer !== "Yes") {
|
||||
return
|
||||
}
|
||||
|
||||
for (const key of this.context.globalState.keys()) {
|
||||
await this.context.globalState.update(key, undefined)
|
||||
}
|
||||
@@ -2097,11 +2106,11 @@ export class ClineProvider implements vscode.WebviewViewProvider {
|
||||
for (const key of secretKeys) {
|
||||
await this.storeSecret(key, undefined)
|
||||
}
|
||||
await this.configManager.resetAllConfigs()
|
||||
if (this.cline) {
|
||||
this.cline.abortTask()
|
||||
this.cline = undefined
|
||||
}
|
||||
vscode.window.showInformationMessage("State reset")
|
||||
await this.postStateToWebview()
|
||||
await this.postMessageToWebview({ type: "action", action: "chatButtonClicked" })
|
||||
}
|
||||
|
||||
@@ -443,18 +443,18 @@ describe("ClineProvider", () => {
|
||||
|
||||
// Mock ConfigManager methods
|
||||
provider.configManager = {
|
||||
GetModeConfigId: jest.fn().mockResolvedValue("test-id"),
|
||||
ListConfig: jest.fn().mockResolvedValue([{ name: "test-config", id: "test-id", apiProvider: "anthropic" }]),
|
||||
LoadConfig: jest.fn().mockResolvedValue({ apiProvider: "anthropic" }),
|
||||
SetModeConfig: jest.fn(),
|
||||
getModeConfigId: jest.fn().mockResolvedValue("test-id"),
|
||||
listConfig: jest.fn().mockResolvedValue([{ name: "test-config", id: "test-id", apiProvider: "anthropic" }]),
|
||||
loadConfig: jest.fn().mockResolvedValue({ apiProvider: "anthropic" }),
|
||||
setModeConfig: jest.fn(),
|
||||
} as any
|
||||
|
||||
// Switch to architect mode
|
||||
await messageHandler({ type: "mode", text: "architect" })
|
||||
|
||||
// Should load the saved config for architect mode
|
||||
expect(provider.configManager.GetModeConfigId).toHaveBeenCalledWith("architect")
|
||||
expect(provider.configManager.LoadConfig).toHaveBeenCalledWith("test-config")
|
||||
expect(provider.configManager.getModeConfigId).toHaveBeenCalledWith("architect")
|
||||
expect(provider.configManager.loadConfig).toHaveBeenCalledWith("test-config")
|
||||
expect(mockContext.globalState.update).toHaveBeenCalledWith("currentApiConfigName", "test-config")
|
||||
})
|
||||
|
||||
@@ -464,11 +464,11 @@ describe("ClineProvider", () => {
|
||||
|
||||
// Mock ConfigManager methods
|
||||
provider.configManager = {
|
||||
GetModeConfigId: jest.fn().mockResolvedValue(undefined),
|
||||
ListConfig: jest
|
||||
getModeConfigId: jest.fn().mockResolvedValue(undefined),
|
||||
listConfig: jest
|
||||
.fn()
|
||||
.mockResolvedValue([{ name: "current-config", id: "current-id", apiProvider: "anthropic" }]),
|
||||
SetModeConfig: jest.fn(),
|
||||
setModeConfig: jest.fn(),
|
||||
} as any
|
||||
|
||||
// Mock current config name
|
||||
@@ -483,7 +483,7 @@ describe("ClineProvider", () => {
|
||||
await messageHandler({ type: "mode", text: "architect" })
|
||||
|
||||
// Should save current config as default for architect mode
|
||||
expect(provider.configManager.SetModeConfig).toHaveBeenCalledWith("architect", "current-id")
|
||||
expect(provider.configManager.setModeConfig).toHaveBeenCalledWith("architect", "current-id")
|
||||
})
|
||||
|
||||
test("saves config as default for current mode when loading config", async () => {
|
||||
@@ -491,10 +491,10 @@ describe("ClineProvider", () => {
|
||||
const messageHandler = (mockWebviewView.webview.onDidReceiveMessage as jest.Mock).mock.calls[0][0]
|
||||
|
||||
provider.configManager = {
|
||||
LoadConfig: jest.fn().mockResolvedValue({ apiProvider: "anthropic", id: "new-id" }),
|
||||
ListConfig: jest.fn().mockResolvedValue([{ name: "new-config", id: "new-id", apiProvider: "anthropic" }]),
|
||||
SetModeConfig: jest.fn(),
|
||||
GetModeConfigId: jest.fn().mockResolvedValue(undefined),
|
||||
loadConfig: jest.fn().mockResolvedValue({ apiProvider: "anthropic", id: "new-id" }),
|
||||
listConfig: jest.fn().mockResolvedValue([{ name: "new-config", id: "new-id", apiProvider: "anthropic" }]),
|
||||
setModeConfig: jest.fn(),
|
||||
getModeConfigId: jest.fn().mockResolvedValue(undefined),
|
||||
} as any
|
||||
|
||||
// First set the mode
|
||||
@@ -504,7 +504,7 @@ describe("ClineProvider", () => {
|
||||
await messageHandler({ type: "loadApiConfiguration", text: "new-config" })
|
||||
|
||||
// Should save new config as default for architect mode
|
||||
expect(provider.configManager.SetModeConfig).toHaveBeenCalledWith("architect", "new-id")
|
||||
expect(provider.configManager.setModeConfig).toHaveBeenCalledWith("architect", "new-id")
|
||||
})
|
||||
|
||||
test("handles request delay settings messages", async () => {
|
||||
@@ -678,8 +678,8 @@ describe("ClineProvider", () => {
|
||||
const messageHandler = (mockWebviewView.webview.onDidReceiveMessage as jest.Mock).mock.calls[0][0]
|
||||
|
||||
provider.configManager = {
|
||||
ListConfig: jest.fn().mockResolvedValue([{ name: "test-config", id: "test-id", apiProvider: "anthropic" }]),
|
||||
SetModeConfig: jest.fn(),
|
||||
listConfig: jest.fn().mockResolvedValue([{ name: "test-config", id: "test-id", apiProvider: "anthropic" }]),
|
||||
setModeConfig: jest.fn(),
|
||||
} as any
|
||||
|
||||
// Update API configuration
|
||||
@@ -689,7 +689,7 @@ describe("ClineProvider", () => {
|
||||
})
|
||||
|
||||
// Should save config as default for current mode
|
||||
expect(provider.configManager.SetModeConfig).toHaveBeenCalledWith("code", "test-id")
|
||||
expect(provider.configManager.setModeConfig).toHaveBeenCalledWith("code", "test-id")
|
||||
})
|
||||
|
||||
test("file content includes line numbers", async () => {
|
||||
|
||||
Reference in New Issue
Block a user