/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "nsVariant.h"
#include "storage_test_harness.h"
#include "nsThreadUtils.h"
#include "nsIURI.h"
#include "nsIFileURL.h"
#include "nsIVariant.h"
#include "nsNetUtil.h"

////////////////////////////////////////////////////////////////////////////////
//// Tests

TEST(storage_async_thread_naming, MemoryDatabase)
{
  HookSqliteMutex hook;

  nsCOMPtr<mozIStorageConnection> db(getMemoryDatabase());

  nsCOMPtr<nsIThread> target(get_conn_async_thread(db));
  do_check_true(target);
  PRThread* prThread;
  target->GetPRThread(&prThread);
  do_check_true(prThread);
  nsAutoCString name(PR_GetThreadName(prThread));
  printf("%s", name.get());
  do_check_true(StringBeginsWith(name, "sqldb:memory"_ns));

  blocking_async_close(db);
}

TEST(storage_async_thread_naming, FileDatabase)
{
  HookSqliteMutex hook;

  nsAutoString filename(u"test_thread_name.sqlite"_ns);
  nsCOMPtr<nsIFile> dbFile;
  do_check_success(NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR,
                                          getter_AddRefs(dbFile)));
  do_check_success(dbFile->Append(filename));
  nsCOMPtr<mozIStorageService> ss = getService();
  nsCOMPtr<mozIStorageConnection> conn;
  do_check_success(ss->OpenDatabase(dbFile, 0, getter_AddRefs(conn)));

  nsCOMPtr<nsIThread> target(get_conn_async_thread(conn));
  do_check_true(target);
  PRThread* prThread;
  target->GetPRThread(&prThread);
  do_check_true(prThread);
  nsAutoCString name(PR_GetThreadName(prThread));
  nsAutoCString expected("sqldb:"_ns);
  expected.Append(NS_ConvertUTF16toUTF8(filename));
  do_check_true(StringBeginsWith(name, expected));

  {
    nsCOMPtr<mozIStorageConnection> clone;
    do_check_success(conn->Clone(true, getter_AddRefs(clone)));
    nsCOMPtr<nsIThread> target(get_conn_async_thread(conn));
    do_check_true(target);
    PRThread* prThread;
    target->GetPRThread(&prThread);
    do_check_true(prThread);
    nsAutoCString name(PR_GetThreadName(prThread));
    nsAutoCString expected("sqldb:"_ns);
    expected.Append(NS_ConvertUTF16toUTF8(filename));
    do_check_true(StringBeginsWith(name, expected));
    blocking_async_close(clone);
  }

  blocking_async_close(conn);
}

TEST(storage_async_thread_naming, FileUnsharedDatabase)
{
  HookSqliteMutex hook;

  nsAutoString filename(u"test_thread_name.sqlite"_ns);
  nsCOMPtr<nsIFile> dbFile;
  do_check_success(NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR,
                                          getter_AddRefs(dbFile)));
  do_check_success(dbFile->Append(filename));
  nsCOMPtr<mozIStorageService> ss = getService();
  nsCOMPtr<mozIStorageConnection> conn;
  do_check_success(ss->OpenUnsharedDatabase(dbFile, 0, getter_AddRefs(conn)));

  nsCOMPtr<nsIThread> target(get_conn_async_thread(conn));
  do_check_true(target);
  PRThread* prThread;
  target->GetPRThread(&prThread);
  do_check_true(prThread);
  nsAutoCString name(PR_GetThreadName(prThread));
  nsAutoCString expected("sqldb:"_ns);
  expected.Append(NS_ConvertUTF16toUTF8(filename));
  do_check_true(StringBeginsWith(name, expected));

  blocking_async_close(conn);
}

TEST(storage_async_thread_naming, FileURLDatabase)
{
  HookSqliteMutex hook;

  nsAutoString filename(u"test_thread_name.sqlite"_ns);
  nsCOMPtr<nsIFile> dbFile;
  do_check_success(NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR,
                                          getter_AddRefs(dbFile)));
  do_check_success(dbFile->Append(filename));
  nsCOMPtr<nsIURI> uri;
  do_check_success(NS_NewFileURI(getter_AddRefs(uri), dbFile));
  nsCOMPtr<nsIFileURL> fileUrl = do_QueryInterface(uri);
  nsCOMPtr<mozIStorageService> ss = getService();
  nsCOMPtr<mozIStorageConnection> conn;
  do_check_success(ss->OpenDatabaseWithFileURL(fileUrl, EmptyCString(), 0,
                                               getter_AddRefs(conn)));

  nsCOMPtr<nsIThread> target(get_conn_async_thread(conn));
  do_check_true(target);
  PRThread* prThread;
  target->GetPRThread(&prThread);
  do_check_true(prThread);
  nsAutoCString name(PR_GetThreadName(prThread));
  nsAutoCString expected("sqldb:"_ns);
  expected.Append(NS_ConvertUTF16toUTF8(filename));
  do_check_true(StringBeginsWith(name, expected));

  blocking_async_close(conn);
}

TEST(storage_async_thread_naming, OverrideFileURLDatabase)
{
  HookSqliteMutex hook;

  nsAutoString filename(u"test_thread_name.sqlite"_ns);
  nsCOMPtr<nsIFile> dbFile;
  do_check_success(NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR,
                                          getter_AddRefs(dbFile)));
  do_check_success(dbFile->Append(filename));
  nsCOMPtr<nsIURI> uri;
  do_check_success(NS_NewFileURI(getter_AddRefs(uri), dbFile));
  nsCOMPtr<nsIFileURL> fileUrl = do_QueryInterface(uri);
  nsCOMPtr<mozIStorageService> ss = getService();
  nsCOMPtr<mozIStorageConnection> conn;
  nsAutoCString override("override"_ns);
  do_check_success(
      ss->OpenDatabaseWithFileURL(fileUrl, override, 0, getter_AddRefs(conn)));

  nsCOMPtr<nsIThread> target(get_conn_async_thread(conn));
  do_check_true(target);
  PRThread* prThread;
  target->GetPRThread(&prThread);
  do_check_true(prThread);
  nsAutoCString name(PR_GetThreadName(prThread));
  nsAutoCString expected("sqldb:"_ns);
  expected.Append(override);
  do_check_true(StringBeginsWith(name, expected));

  blocking_async_close(conn);
}

TEST(storage_async_thread_naming, AsyncOpenDatabase)
{
  HookSqliteMutex hook;

  nsAutoString filename(u"test_thread_name.sqlite"_ns);
  nsCOMPtr<nsIFile> dbFile;
  do_check_success(NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR,
                                          getter_AddRefs(dbFile)));
  do_check_success(dbFile->Append(filename));
  nsCOMPtr<mozIStorageService> ss = getService();

  auto completionSpinner = mozilla::MakeRefPtr<AsyncCompletionSpinner>();
  auto variant = mozilla::MakeRefPtr<nsVariant>();
  variant->SetAsInterface(NS_GET_IID(nsIFile), dbFile);
  do_check_success(ss->OpenAsyncDatabase(variant, 0, 0, completionSpinner));
  completionSpinner->SpinUntilCompleted();
  nsCOMPtr<mozIStorageConnection> conn(
      do_QueryInterface(completionSpinner->mCompletionValue));
  nsCOMPtr<nsIThread> target(get_conn_async_thread(conn));
  do_check_true(target);
  PRThread* prThread;
  target->GetPRThread(&prThread);
  do_check_true(prThread);
  nsAutoCString name(PR_GetThreadName(prThread));
  nsAutoCString expected("sqldb:"_ns);
  expected.Append(NS_ConvertUTF16toUTF8(filename));
  do_check_true(StringBeginsWith(name, expected));

  blocking_async_close(conn);
}

TEST(storage_async_thread_naming, AsyncClone)
{
  HookSqliteMutex hook;

  nsAutoString filename(u"test_thread_name.sqlite"_ns);
  nsCOMPtr<nsIFile> dbFile;
  do_check_success(NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR,
                                          getter_AddRefs(dbFile)));
  do_check_success(dbFile->Append(filename));
  nsCOMPtr<mozIStorageService> ss = getService();

  nsCOMPtr<mozIStorageConnection> conn;
  do_check_success(ss->OpenDatabase(dbFile, 0, getter_AddRefs(conn)));

  auto completionSpinner = mozilla::MakeRefPtr<AsyncCompletionSpinner>();
  auto variant = mozilla::MakeRefPtr<nsVariant>();
  variant->SetAsInterface(NS_GET_IID(nsIFile), dbFile);
  do_check_success(conn->AsyncClone(true, completionSpinner));
  completionSpinner->SpinUntilCompleted();
  nsCOMPtr<mozIStorageConnection> clone(
      do_QueryInterface(completionSpinner->mCompletionValue));
  nsCOMPtr<nsIThread> target(get_conn_async_thread(clone));
  do_check_true(target);
  PRThread* prThread;
  target->GetPRThread(&prThread);
  do_check_true(prThread);
  nsAutoCString name(PR_GetThreadName(prThread));
  nsAutoCString expected("sqldb:"_ns);
  expected.Append(NS_ConvertUTF16toUTF8(filename));
  do_check_true(StringBeginsWith(name, expected));

  blocking_async_close(conn);
  blocking_async_close(clone);
}