// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/extensions/menu_manager_test_observer.h"

#include "content/public/browser/browser_thread.h"

namespace extensions {

MenuManagerTestObserver::MenuManagerTestObserver(MenuManager* menu_manager)
    : menu_manager_(menu_manager) {
  observation_.Observe(menu_manager_.get());
}

MenuManagerTestObserver::~MenuManagerTestObserver() = default;

void MenuManagerTestObserver::WaitForExtension(
    const ExtensionId& extension_id) {
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);

  // The extension's menus may have already been loaded before we were
  // able to observe it.
  if (MenusItemsFound(extension_id))
    return;

  if (ids_with_reads_.count(extension_id) == 0) {
    waiting_for_id_ = extension_id;
    run_loop_.Run();
    DCHECK(MenusItemsFound(extension_id));
  }
}

void MenuManagerTestObserver::DidReadFromStorage(
    const ExtensionId& extension_id) {
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  ids_with_reads_.insert(extension_id);
  if (extension_id == waiting_for_id_) {
    run_loop_.Quit();
  }
}

void MenuManagerTestObserver::WillWriteToStorage(
    const ExtensionId& extension_id) {
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  ids_with_writes_.insert(extension_id);
  if (extension_id == waiting_for_id_) {
    run_loop_.Quit();
  }
}

bool MenuManagerTestObserver::MenusItemsFound(const ExtensionId& extension_id) {
  const MenuItem::ExtensionKey key(extension_id);
  return menu_manager_->MenuItems(key) &&
         !menu_manager_->MenuItems(key)->empty();
}

}  // namespace extensions