// Copyright 2016 Google Inc. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include <iostream>

#include "zcpointer.h"

class C {
 public:
  ~C() {}

  void DoThing() {}
};

#define EXPECT(expr) do { if (!(expr)) { throw std::logic_error(#expr); } } while(0)

#define EXPECT_UAF(expr) do { \
    try { \
      (expr); \
      throw std::logic_error("Expected use-after-free: " #expr); \
    } catch (zc::UseAfterFreeError) {} \
  } while(0)

void TestReset() {
  zc::owned<C> c(new C());
  zc::ref<C> owned = c.get();
  zc::ref<C> owned2 = owned;
  c.reset();
  EXPECT_UAF(owned2->DoThing());
}

template <typename T>
void TestUnwrap() {
  zc::owned<T> t(new T());
  //T* unwrap = t.get();

  zc::ref<T> ref = t.get();
  T* unwrap2 = ref;
}

void TestMove() {
  zc::owned<C> c(new C());
  zc::ref<C> owned = c.get();

  zc::owned<C> c2(std::move(c));
  owned->DoThing();

  c2.reset();
  EXPECT_UAF(owned->DoThing());
}

void PtrHelper(zc::ref<C>* out) {
  zc::owned<C> c(new C());
  *out = c.get();
}

void TestPtr() {
  zc::ref<C> ref;
  PtrHelper(&ref);
  EXPECT_UAF(ref->DoThing());
}

void TestEquality() {
  zc::owned<C> a(new C());
  zc::owned<C> b(new C());

  EXPECT(a == a);
  EXPECT(b == b);
  EXPECT(a != b);

  zc::ref<C> ra = a.get();
  zc::ref<C> rb = b.get();

  EXPECT(ra == ra);
  EXPECT(ra == a.get());
  EXPECT(rb == rb);
  EXPECT(rb == b.get());

  EXPECT(rb != ra);

  zc::ref<C> r = a.get();
  EXPECT(r == ra);
  EXPECT(r == a.get());

  zc::owned<C> c;
  zc::owned<C> c2;
  zc::ref<C> rc = nullptr;

  EXPECT(rc == c.get());
  EXPECT(c == nullptr);
  EXPECT(rc == nullptr);
  EXPECT(a != c);
  EXPECT(c == c2);
}

void TestNulls() {
  zc::owned<C> l;
  zc::owned<C> r;

  zc::ref<C> rl = l.get();
  zc::ref<C> rr = r.get();

  r = std::move(l);
  rl = rr;

  EXPECT(l == nullptr);
  EXPECT(r == nullptr);
  EXPECT(rl == nullptr);
  EXPECT(rr == nullptr);
}

#define TEST_FUNC(fn) { #fn , Test##fn }

int main() {
  struct {
    const char* name;
    void (*test)();
  } kTests[] = {
    TEST_FUNC(Reset),
    TEST_FUNC(Move),
    TEST_FUNC(Ptr),
    TEST_FUNC(Equality),
    TEST_FUNC(Nulls),
  };

  bool passed = true;
  for (const auto& test : kTests) {
    std::cout << "=== BEGIN " << test.name << " ===" << std::endl;
    try {
      test.test();
      std::cout << "=== PASS " << test.name << " ===" << std::endl;
    } catch (const std::logic_error& e) {
      passed = false;
      std::cout << "=== FAIL " << test.name
                << ": Assertion failure: " << e.what() << " ===" << std::endl;
    }
  }

  return passed ? 0 : 1;
}