/* 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 "FileSystemMocks.h"
#include "fs/FileSystemChildFactory.h"
#include "gtest/gtest.h"
#include "mozilla/dom/FileSystemDirectoryHandle.h"
#include "mozilla/dom/FileSystemFileHandle.h"
#include "mozilla/dom/FileSystemHandle.h"
#include "mozilla/dom/FileSystemHandleBinding.h"
#include "mozilla/dom/FileSystemManager.h"
#include "mozilla/dom/StorageManager.h"
#include "nsIGlobalObject.h"

namespace mozilla::dom::fs::test {

class TestFileSystemHandle : public ::testing::Test {
 protected:
  void SetUp() override {
    // TODO: Fix the test to not depend on CreateFileSystemManagerParent
    // failure because of the pref set to false.
    nsCOMPtr<nsIPrefBranch> prefs = do_GetService(NS_PREFSERVICE_CONTRACTID);
    prefs->SetBoolPref("dom.fs.enabled", false);

    mDirMetadata = FileSystemEntryMetadata("dir"_ns, u"Directory"_ns,
                                           /* directory */ true);
    mFileMetadata =
        FileSystemEntryMetadata("file"_ns, u"File"_ns, /* directory */ false);
    mManager = MakeAndAddRef<FileSystemManager>(mGlobal, nullptr);
  }

  void TearDown() override {
    nsCOMPtr<nsIPrefBranch> prefs = do_GetService(NS_PREFSERVICE_CONTRACTID);
    prefs->SetBoolPref("dom.fs.enabled", true);

    if (!mManager->IsShutdown()) {
      mManager->Shutdown();
    }
  }

  nsIGlobalObject* mGlobal = GetGlobal();
  FileSystemEntryMetadata mDirMetadata;
  FileSystemEntryMetadata mFileMetadata;
  RefPtr<FileSystemManager> mManager;
};

TEST_F(TestFileSystemHandle, createAndDestroyHandles) {
  RefPtr<FileSystemHandle> dirHandle =
      new FileSystemDirectoryHandle(mGlobal, mManager, mDirMetadata);
  RefPtr<FileSystemHandle> fileHandle =
      new FileSystemFileHandle(mGlobal, mManager, mFileMetadata);

  EXPECT_TRUE(dirHandle);
  EXPECT_TRUE(fileHandle);
}

TEST_F(TestFileSystemHandle, areFileNamesAsExpected) {
  RefPtr<FileSystemHandle> dirHandle =
      new FileSystemDirectoryHandle(mGlobal, mManager, mDirMetadata);
  RefPtr<FileSystemHandle> fileHandle =
      new FileSystemFileHandle(mGlobal, mManager, mFileMetadata);

  auto GetEntryName = [](const RefPtr<FileSystemHandle>& aHandle) {
    DOMString domName;
    aHandle->GetName(domName);
    return domName;
  };

  const nsString& dirName = GetEntryName(dirHandle);
  EXPECT_TRUE(mDirMetadata.entryName().Equals(dirName));

  const nsString& fileName = GetEntryName(fileHandle);
  EXPECT_TRUE(mFileMetadata.entryName().Equals(fileName));
}

TEST_F(TestFileSystemHandle, isParentObjectReturned) {
  ASSERT_TRUE(mGlobal);
  RefPtr<FileSystemHandle> dirHandle =
      new FileSystemDirectoryHandle(mGlobal, mManager, mDirMetadata);

  ASSERT_EQ(mGlobal, dirHandle->GetParentObject());
}

TEST_F(TestFileSystemHandle, areHandleKindsAsExpected) {
  RefPtr<FileSystemHandle> dirHandle =
      new FileSystemDirectoryHandle(mGlobal, mManager, mDirMetadata);
  RefPtr<FileSystemHandle> fileHandle =
      new FileSystemFileHandle(mGlobal, mManager, mFileMetadata);

  EXPECT_EQ(FileSystemHandleKind::Directory, dirHandle->Kind());
  EXPECT_EQ(FileSystemHandleKind::File, fileHandle->Kind());
}

TEST_F(TestFileSystemHandle, isDifferentEntry) {
  RefPtr<FileSystemHandle> dirHandle =
      new FileSystemDirectoryHandle(mGlobal, mManager, mDirMetadata);
  RefPtr<FileSystemHandle> fileHandle =
      new FileSystemFileHandle(mGlobal, mManager, mFileMetadata);

  IgnoredErrorResult rv;
  RefPtr<Promise> promise = dirHandle->IsSameEntry(*fileHandle, rv);
  ASSERT_TRUE(rv.ErrorCodeIs(NS_OK));
  ASSERT_TRUE(promise);
  ASSERT_EQ(Promise::PromiseState::Resolved, promise->State());

  nsString result;
  ASSERT_NSEQ(NS_OK, GetAsString(promise, result));
  ASSERT_STREQ(u"false"_ns, result);
}

TEST_F(TestFileSystemHandle, isSameEntry) {
  RefPtr<FileSystemHandle> fileHandle =
      new FileSystemFileHandle(mGlobal, mManager, mFileMetadata);

  IgnoredErrorResult rv;
  RefPtr<Promise> promise = fileHandle->IsSameEntry(*fileHandle, rv);
  ASSERT_TRUE(rv.ErrorCodeIs(NS_OK));
  ASSERT_TRUE(promise);
  ASSERT_EQ(Promise::PromiseState::Resolved, promise->State());

  nsString result;
  ASSERT_NSEQ(NS_OK, GetAsString(promise, result));
  ASSERT_STREQ(u"true"_ns, result);
}

}  // namespace mozilla::dom::fs::test