/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
// vim:cindent:ts=4:et:sw=4:
/* 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 "nsAutoPtr.h"
#include "gtest/gtest.h"

class TestObjectBaseA {
  public:
    // Virtual dtor for deleting through base class pointer
    virtual ~TestObjectBaseA() { }
    void MemberFunction(int, int*, int&)
    {
    }
    virtual void VirtualMemberFunction(int, int*, int&) { };
    virtual void VirtualConstMemberFunction(int, int*, int&) const { };
    int fooA;
};

class TestObjectBaseB {
  public:
    // Virtual dtor for deleting through base class pointer
    virtual ~TestObjectBaseB() { }
    int fooB;
};

class TestObject : public TestObjectBaseA, public TestObjectBaseB {
  public:
    TestObject()
    {
    }

    // Virtual dtor for deleting through base class pointer
    virtual ~TestObject()
    {
      destructed++;
    }

    virtual void VirtualMemberFunction(int, int*, int&) override
    {
    }
    virtual void VirtualConstMemberFunction(int, int*, int&) const override
    {
    }

    static int destructed;
    static const void* last_ptr;
};

int TestObject::destructed = 0;
const void* TestObject::last_ptr = nullptr;

static void CreateTestObject(TestObject **aResult)
{
  *aResult = new TestObject();
}

static void DoSomethingWithTestObject(TestObject *aIn)
{
  TestObject::last_ptr = static_cast<void*>(aIn);
}

static void DoSomethingWithConstTestObject(const TestObject *aIn)
{
  TestObject::last_ptr = static_cast<const void*>(aIn);
}

static void DoSomethingWithTestObjectBaseB(TestObjectBaseB *aIn)
{
  TestObject::last_ptr = static_cast<void*>(aIn);
}

static void DoSomethingWithConstTestObjectBaseB(const TestObjectBaseB *aIn)
{
  TestObject::last_ptr = static_cast<const void*>(aIn);
}

TEST(AutoPtr, Assignment)
{
  TestObject::destructed = 0;

  {
    nsAutoPtr<TestObject> pobj( new TestObject() );
  }

  ASSERT_EQ(1, TestObject::destructed);

  {
    nsAutoPtr<TestObject> pobj( new TestObject() );
    pobj = new TestObject();
    ASSERT_EQ(2, TestObject::destructed);
  }

  ASSERT_EQ(3, TestObject::destructed);
}

TEST(AutoPtr, getter_Transfers)
{
  TestObject::destructed = 0;

  {
    nsAutoPtr<TestObject> ptr;
    CreateTestObject(getter_Transfers(ptr));
  }

  ASSERT_EQ(1, TestObject::destructed);
}

TEST(AutoPtr, Casting)
{
  // This comparison is always false, as it should be. The extra parens
  // suppress a -Wunreachable-code warning about printf being unreachable.
  ASSERT_NE(((void*)(TestObject*)0x1000),
             ((void*)(TestObjectBaseB*)(TestObject*)0x1000));

  {
    nsAutoPtr<TestObject> p1(new TestObject());
    TestObjectBaseB *p2 = p1;
    ASSERT_NE(static_cast<void*>(p1),
               static_cast<void*>(p2));
    ASSERT_EQ(p1, p2);
    ASSERT_FALSE(p1 != p2);
    ASSERT_EQ(p2, p1);
    ASSERT_FALSE(p2 != p1);
  }

  {
    TestObject *p1 = new TestObject();
    nsAutoPtr<TestObjectBaseB> p2(p1);
    ASSERT_EQ(p1, p2);
    ASSERT_FALSE(p1 != p2);
    ASSERT_EQ(p2, p1);
    ASSERT_FALSE(p2 != p1);
  }
}

TEST(AutoPtr, Forget)
{
  TestObject::destructed = 0;

  {
    nsAutoPtr<TestObject> pobj( new TestObject() );
    nsAutoPtr<TestObject> pobj2( pobj.forget() );
    ASSERT_EQ(0, TestObject::destructed);
  }

  ASSERT_EQ(1, TestObject::destructed);
}

TEST(AutoPtr, Construction)
{
  TestObject::destructed = 0;

  {
    nsAutoPtr<TestObject> pobj(new TestObject());
  }

  ASSERT_EQ(1, TestObject::destructed);
}

TEST(AutoPtr, ImplicitConversion)
{
  // This test is basically successful if it builds. We add a few assertions
  // to make gtest happy.
  TestObject::destructed = 0;

  {
    nsAutoPtr<TestObject> pobj(new TestObject());
    DoSomethingWithTestObject(pobj);
    DoSomethingWithConstTestObject(pobj);
  }

  ASSERT_EQ(1, TestObject::destructed);

  {
    nsAutoPtr<TestObject> pobj(new TestObject());
    DoSomethingWithTestObjectBaseB(pobj);
    DoSomethingWithConstTestObjectBaseB(pobj);
  }

  ASSERT_EQ(2, TestObject::destructed);

  {
    const nsAutoPtr<TestObject> pobj(new TestObject());
    DoSomethingWithTestObject(pobj);
    DoSomethingWithConstTestObject(pobj);
  }

  ASSERT_EQ(3, TestObject::destructed);

  {
    const nsAutoPtr<TestObject> pobj(new TestObject());
    DoSomethingWithTestObjectBaseB(pobj);
    DoSomethingWithConstTestObjectBaseB(pobj);
  }

  ASSERT_EQ(4, TestObject::destructed);
}

TEST(AutoPtr, ArrowOperator)
{
  // This test is basically successful if it builds. We add a few assertions
  // to make gtest happy.
  TestObject::destructed = 0;

  {
    int test = 1;
    void (TestObjectBaseA::*fPtr)( int, int*, int& ) = &TestObjectBaseA::MemberFunction;
    void (TestObjectBaseA::*fVPtr)( int, int*, int& ) = &TestObjectBaseA::VirtualMemberFunction;
    void (TestObjectBaseA::*fVCPtr)( int, int*, int& ) const = &TestObjectBaseA::VirtualConstMemberFunction;
    nsAutoPtr<TestObjectBaseA> pobj(new TestObject());
    (pobj->*fPtr)(test, &test, test);
    (pobj->*fVPtr)(test, &test, test);
    (pobj->*fVCPtr)(test, &test, test);
  }

  ASSERT_EQ(1, TestObject::destructed);
}