michael@0: #include <stdio.h>
michael@0: 
michael@0: #include "gtest/gtest.h"
michael@0: 
michael@0: #include "gfxASurface.h"
michael@0: #include "gfxImageSurface.h"
michael@0: 
michael@0: #include "cairo/cairo.h"
michael@0: 
michael@0: int
michael@0: GetASurfaceRefCount(gfxASurface *s) {
michael@0:     NS_ADDREF(s);
michael@0:     return s->Release();
michael@0: }
michael@0: 
michael@0: int
michael@0: CheckInt (int value, int expected) {
michael@0:     if (value != expected) {
michael@0:         fprintf (stderr, "Expected %d got %d\n", expected, value);
michael@0:         return 1;
michael@0:     }
michael@0: 
michael@0:     return 0;
michael@0: }
michael@0: 
michael@0: int
michael@0: CheckPointer (void *value, void *expected) {
michael@0:     if (value != expected) {
michael@0:         fprintf (stderr, "Expected %p got %p\n", expected, value);
michael@0:         return 1;
michael@0:     }
michael@0: 
michael@0:     return 0;
michael@0: }
michael@0: 
michael@0: static cairo_user_data_key_t destruction_key;
michael@0: void
michael@0: SurfaceDestroyNotifier (void *data) {
michael@0:     *(int *)data = 1;
michael@0: }
michael@0: 
michael@0: int
michael@0: TestNewSurface () {
michael@0:     int failures = 0;
michael@0:     int destroyed = 0;
michael@0: 
michael@0:     nsRefPtr<gfxASurface> s = new gfxImageSurface (gfxIntSize(10, 10), gfxImageFormat::ARGB32);
michael@0:     cairo_surface_t *cs = s->CairoSurface();
michael@0: 
michael@0:     cairo_surface_set_user_data (cs, &destruction_key, &destroyed, SurfaceDestroyNotifier);
michael@0: 
michael@0:     failures += CheckInt (GetASurfaceRefCount(s.get()), 1);
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 1);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     cairo_surface_reference(cs);
michael@0: 
michael@0:     failures += CheckInt (GetASurfaceRefCount(s.get()), 2);
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 2);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     gfxASurface *savedWrapper = s.get();
michael@0: 
michael@0:     s = nullptr;
michael@0: 
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 1);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     s = gfxASurface::Wrap(cs);
michael@0: 
michael@0:     failures += CheckPointer (s.get(), savedWrapper);
michael@0:     failures += CheckInt (GetASurfaceRefCount(s.get()), 2);
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 2);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     cairo_surface_destroy(cs);
michael@0: 
michael@0:     failures += CheckInt (GetASurfaceRefCount(s.get()), 1);
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 1);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     s = nullptr;
michael@0: 
michael@0:     failures += CheckInt (destroyed, 1);
michael@0: 
michael@0:     return failures;
michael@0: }
michael@0: 
michael@0: int
michael@0: TestExistingSurface () {
michael@0:     int failures = 0;
michael@0:     int destroyed = 0;
michael@0: 
michael@0:     cairo_surface_t *cs = cairo_image_surface_create (CAIRO_FORMAT_ARGB32, 10, 10);
michael@0: 
michael@0:     cairo_surface_set_user_data (cs, &destruction_key, &destroyed, SurfaceDestroyNotifier);
michael@0: 
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 1);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     nsRefPtr<gfxASurface> s = gfxASurface::Wrap(cs);
michael@0: 
michael@0:     failures += CheckInt (GetASurfaceRefCount(s.get()), 2);
michael@0: 
michael@0:     cairo_surface_reference(cs);
michael@0: 
michael@0:     failures += CheckInt (GetASurfaceRefCount(s.get()), 3);
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 3);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     gfxASurface *savedWrapper = s.get();
michael@0: 
michael@0:     s = nullptr;
michael@0: 
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 2);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     s = gfxASurface::Wrap(cs);
michael@0: 
michael@0:     failures += CheckPointer (s.get(), savedWrapper);
michael@0:     failures += CheckInt (GetASurfaceRefCount(s.get()), 3);
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 3);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     cairo_surface_destroy(cs);
michael@0: 
michael@0:     failures += CheckInt (GetASurfaceRefCount(s.get()), 2);
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 2);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     s = nullptr;
michael@0: 
michael@0:     failures += CheckInt (cairo_surface_get_reference_count(cs), 1);
michael@0:     failures += CheckInt (destroyed, 0);
michael@0: 
michael@0:     cairo_surface_destroy(cs);
michael@0: 
michael@0:     failures += CheckInt (destroyed, 1);
michael@0: 
michael@0:     return failures;
michael@0: }
michael@0: 
michael@0: TEST(Gfx, SurfaceRefCount) {
michael@0:     int fail;
michael@0: 
michael@0:     fail = TestNewSurface();
michael@0:     EXPECT_TRUE(fail == 0) << "TestNewSurface: " << fail << " failures";
michael@0:     fail = TestExistingSurface();
michael@0:     EXPECT_TRUE(fail == 0) << "TestExistingSurface: " << fail << " failures";
michael@0: }
michael@0: