// Copyright 2018 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "core/fxcrt/autorestorer.h"
#include "core/fxcrt/retain_ptr.h"
#include "core/fxcrt/unowned_ptr.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/pseudo_retainable.h"

TEST(fxcrt, AutoRestorer) {
  int x = 5;
  {
    AutoRestorer<int> restorer(&x);
    x = 6;
    EXPECT_EQ(6, x);
  }
  EXPECT_EQ(5, x);
}

TEST(fxcrt, AutoRestorerNoOp) {
  int x = 5;
  {
    AutoRestorer<int> restorer(&x);
    EXPECT_EQ(5, x);
  }
  EXPECT_EQ(5, x);
}

TEST(fxcrt, AutoRestorerAbandon) {
  int x = 5;
  {
    AutoRestorer<int> restorer(&x);
    x = 6;
    EXPECT_EQ(6, x);
    restorer.AbandonRestoration();
  }
  EXPECT_EQ(6, x);
}

TEST(fxcrt, AutoRestorerUnownedPtr) {
  int x = 5;
  int y = 6;
  UnownedPtr<int> ptr(&x);
  {
    AutoRestorer<UnownedPtr<int>> restorer(&ptr);
    ptr = &y;
    EXPECT_EQ(&y, ptr);
  }
  EXPECT_EQ(&x, ptr);
}

TEST(fxcrt, AutoRestorerUnownedNoOp) {
  int x = 5;
  UnownedPtr<int> ptr(&x);
  {
    AutoRestorer<UnownedPtr<int>> restorer(&ptr);
    EXPECT_EQ(&x, ptr);
  }
  EXPECT_EQ(&x, ptr);
}

TEST(fxcrt, AutoRestorerUnownedPtrAbandon) {
  int x = 5;
  int y = 6;
  UnownedPtr<int> ptr(&x);
  {
    AutoRestorer<UnownedPtr<int>> restorer(&ptr);
    ptr = &y;
    EXPECT_EQ(&y, ptr);
    restorer.AbandonRestoration();
  }
  EXPECT_EQ(&y, ptr);
}

TEST(fxcrt, AutoRestorerRetainPtr) {
  PseudoRetainable obj1;
  PseudoRetainable obj2;
  RetainPtr<PseudoRetainable> ptr(&obj1);
  EXPECT_EQ(1, obj1.retain_count());
  EXPECT_EQ(0, obj1.release_count());
  {
    AutoRestorer<RetainPtr<PseudoRetainable>> restorer(&ptr);
    // |obj1| is kept alive by restorer in case it need to be restored.
    EXPECT_EQ(2, obj1.retain_count());
    EXPECT_EQ(0, obj1.release_count());
    ptr.Reset(&obj2);
    EXPECT_EQ(&obj2, ptr.Get());

    // |obj1| released by |ptr| assignment.
    EXPECT_TRUE(obj1.alive());
    EXPECT_EQ(2, obj1.retain_count());
    EXPECT_EQ(1, obj1.release_count());

    // |obj2| now kept alive by |ptr|.
    EXPECT_TRUE(obj1.alive());
    EXPECT_EQ(1, obj2.retain_count());
    EXPECT_EQ(0, obj2.release_count());
  }
  EXPECT_EQ(&obj1, ptr.Get());

  // |obj1| now kept alive again by |ptr|.
  EXPECT_TRUE(obj1.alive());
  EXPECT_EQ(3, obj1.retain_count());
  EXPECT_EQ(2, obj1.release_count());

  // |obj2| dead.
  EXPECT_FALSE(obj2.alive());
  EXPECT_EQ(1, obj2.retain_count());
  EXPECT_EQ(1, obj2.release_count());
}

TEST(fxcrt, AutoRestorerRetainPtrNoOp) {
  PseudoRetainable obj1;
  RetainPtr<PseudoRetainable> ptr(&obj1);
  EXPECT_EQ(1, obj1.retain_count());
  EXPECT_EQ(0, obj1.release_count());
  {
    AutoRestorer<RetainPtr<PseudoRetainable>> restorer(&ptr);
    EXPECT_EQ(2, obj1.retain_count());
    EXPECT_EQ(0, obj1.release_count());
  }
  EXPECT_EQ(&obj1, ptr.Get());

  // Self-reassignement avoided ref churn.
  EXPECT_TRUE(obj1.alive());
  EXPECT_EQ(2, obj1.retain_count());
  EXPECT_EQ(1, obj1.release_count());
}

TEST(fxcrt, AutoRestorerRetainPtrAbandon) {
  PseudoRetainable obj1;
  PseudoRetainable obj2;
  RetainPtr<PseudoRetainable> ptr(&obj1);
  EXPECT_EQ(1, obj1.retain_count());
  EXPECT_EQ(0, obj1.release_count());
  {
    AutoRestorer<RetainPtr<PseudoRetainable>> restorer(&ptr);
    // |obj1| is kept alive by restorer in case it need to be restored.
    EXPECT_EQ(2, obj1.retain_count());
    EXPECT_EQ(0, obj1.release_count());
    ptr.Reset(&obj2);
    EXPECT_EQ(&obj2, ptr.Get());

    // |obj1| released by |ptr| assignment.
    EXPECT_EQ(2, obj1.retain_count());
    EXPECT_EQ(1, obj1.release_count());

    // |obj2| now kept alive by |ptr|.
    EXPECT_EQ(1, obj2.retain_count());
    EXPECT_EQ(0, obj2.release_count());

    restorer.AbandonRestoration();
  }
  EXPECT_EQ(&obj2, ptr.Get());

  // |obj1| now dead as a result of abandonment.
  EXPECT_FALSE(obj1.alive());
  EXPECT_EQ(2, obj1.retain_count());
  EXPECT_EQ(2, obj1.release_count());

  // |obj2| kept alive by |ptr|.
  EXPECT_TRUE(obj2.alive());
  EXPECT_EQ(1, obj2.retain_count());
  EXPECT_EQ(0, obj2.release_count());
}