From c2aaf3b7e6c505994af4d8a53f2a96c7ae04c871 Mon Sep 17 00:00:00 2001
From: Dariusz L <dl122@wp.pl>
Date: Sun, 2 Aug 2026 13:32:11 +0200
Subject: [PATCH] test(frontend): cover persisted search source flags

---
 tests/test_search_source_methods.mjs | 47 ++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/tests/test_search_source_methods.mjs b/tests/test_search_source_methods.mjs
index 65621a7..e2aa123 100644
--- a/tests/test_search_source_methods.mjs
+++ b/tests/test_search_source_methods.mjs
@@ -36,9 +36,31 @@ const getSearchSourceDefinitions = eval(`(${extractMethod(searchPanelMethodsSour
 const getSearchResultKeysForSources = eval(`(${extractMethod(searchPanelMethodsSource, 'getSearchResultKeysForSources')})`);
 const getHashLookupSourcesForSearchSources = eval(`(${extractMethod(searchPanelMethodsSource, 'getHashLookupSourcesForSearchSources')})`);
 const clearSearchResultsForSources = eval(`(${extractMethod(searchPanelMethodsSource, 'clearSearchResultsForSources')})`);
+const isSearchSourceEnabled = eval(`(${extractMethod(searchPanelMethodsSource, 'isSearchSourceEnabled')})`);
+const getSearchSourceDefinition = eval(`(${extractMethod(searchPanelMethodsSource, 'getSearchSourceDefinition')})`);
+const getSearchSourceEnabledMap = eval(`(${extractMethod(searchPanelMethodsSource, 'getSearchSourceEnabledMap')})`);
 const getEnabledSearchSources = eval(`(${extractMethod(searchPanelMethodsSource, 'getEnabledSearchSources')})`);
 const getSearchSourcesForSelection = eval(`(${extractMethod(searchPanelMethodsSource, 'getSearchSourcesForSelection')})`);
 
+function installStorage(values = {}) {
+  const previous = Object.getOwnPropertyDescriptor(globalThis, 'localStorage');
+  const storage = {
+    getItem(key) {
+      return Object.prototype.hasOwnProperty.call(values, key) ? values[key] : null;
+    },
+    setItem() {},
+    removeItem() {},
+  };
+  Object.defineProperty(globalThis, 'localStorage', {
+    configurable: true,
+    value: storage,
+  });
+  return () => {
+    if (previous) Object.defineProperty(globalThis, 'localStorage', previous);
+    else delete globalThis.localStorage;
+  };
+}
+
 test('search source labels and definitions remain stable', () => {
   const dialog = {};
 
@@ -106,6 +128,31 @@ test('clearing selected search sources preserves unrelated results and hash matc
   });
 });
 
+test('source enablement defaults to enabled and honors persisted false flags', () => {
+  const restoreStorage = installStorage({
+    'ModelResolver.searchSource.civitaiEnabled': 'false',
+  });
+  try {
+    const dialog = {
+      getSearchSourceDefinitions,
+      getSearchSourceDefinition,
+      isSearchSourceEnabled,
+    };
+    assert.equal(isSearchSourceEnabled.call(dialog, 'civitai'), false);
+    assert.equal(isSearchSourceEnabled.call(dialog, 'huggingface'), true);
+    assert.equal(isSearchSourceEnabled.call(dialog, 'all'), true);
+    assert.deepEqual(getSearchSourceEnabledMap.call(dialog), {
+      local: true,
+      huggingface: true,
+      civitai: false,
+      civarchive: true,
+      lora_manager_archive: true,
+    });
+  } finally {
+    restoreStorage();
+  }
+});
+
 test('enabled source selection falls back to local and respects available sources', () => {
   const dialog = {
     getSearchSourceDefinitions,