def test_fit_allocate_no_double_entry(self):
roots = list()
x = self.bump_allocate(16 * 3)
roots.append(x)
self.bump_allocate(16 * 1)
roots.append(self.bump_allocate(16 * 1))
#
for r in roots:
self.push_root(r)
lib.bump_ptr_reset()
lib.qcgc_collect()
for _ in roots:
self.pop_root()
#
self.assertEqual(lib.small_free_list(0).count, 1)
#
del roots[0]
for r in roots:
self.push_root(r)
lib.qcgc_collect()
for _ in roots:
self.pop_root()
#
self.assertEqual(lib.small_free_list(3).count, 1)
#
y = lib.qcgc_fit_allocate(16 * 3) # Create double entry
self.assertEqual(lib.small_free_list(3).count, 0)
self.assertEqual(lib.small_free_list(0).count, 1)
self.assertEqual(x, y)
def test_collected_weakref_no_err(self):
alive = self.allocate(1)
self.push_root(alive)
wr = self.allocate_weakref(alive)
# Collect
lib.bump_ptr_reset()
lib.qcgc_collect()
def test_fit_allocate_no_double_entry(self):
roots = list()
x = self.bump_allocate(16 * 3)
roots.append(x)
self.bump_allocate(16 * 1)
roots.append(self.bump_allocate(16 * 1))
#
for r in roots:
self.push_root(r)
lib.bump_ptr_reset()
lib.qcgc_collect()
for _ in roots:
self.pop_root()
#
self.assertEqual(lib.small_free_list(0).count, 1)
#
del roots[0]
for r in roots:
self.push_root(r)
lib.qcgc_collect()
for _ in roots:
self.pop_root()
#
self.assertEqual(lib.small_free_list(3).count, 1)
#
y = lib.qcgc_fit_allocate(16 * 3) # Create double entry
self.assertEqual(lib.small_free_list(3).count, 0)
self.assertEqual(lib.small_free_list(0).count, 1)
self.assertEqual(x, y)
def test_color_transitions(self):
"""Test all possible color transitions"""
reachable = list()
unreachable = list()
for i in range(2 * lib.QCGC_INC_MARK_MIN):
o = self.allocate_ref(1)
self.push_root(o)
reachable.append(o)
self.assertEqual(lib.qcgc_get_mark_color(ffi.cast("object_t *",o)), lib.MARK_COLOR_LIGHT_GRAY)
lib.qcgc_incmark() # Marks ALL root objects
self.assertEqual(lib.qcgc_state.phase, lib.GC_MARK)
for o in reachable:
self.assertIn(lib.qcgc_get_mark_color(ffi.cast("object_t *", o)), [lib.MARK_COLOR_DARK_GRAY, lib.MARK_COLOR_BLACK])
if (lib.qcgc_get_mark_color(ffi.cast("object_t *", o)) == lib.MARK_COLOR_BLACK):
# Trigger write barrier and add object
lib.qcgc_write(ffi.cast("object_t *", o))
self.assertEqual(lib.qcgc_get_mark_color(ffi.cast("object_t *", o)), lib.MARK_COLOR_DARK_GRAY)
lib.qcgc_mark()
for o in reachable:
self.assertEqual(lib.qcgc_get_mark_color(ffi.cast("object_t *", o)), lib.MARK_COLOR_BLACK)
lib.bump_ptr_reset()
lib.qcgc_sweep()
for o in reachable:
self.assertEqual(lib.qcgc_get_mark_color(ffi.cast("object_t *", o)), lib.MARK_COLOR_WHITE)
def test_bump_remain_allocated(self):
for _ in range(10):
self.push_root(self.allocate(1))
lib.bump_ptr_reset()
lib.qcgc_collect()
self.assertEqual(lib.qcgc_state.free_cells, self.arena_blocks - 10)
lib.qcgc_collect()
self.assertEqual(lib.qcgc_state.free_cells, self.arena_blocks - 10)
def test_reuse_old_free_space(self):
arena = lib.qcgc_arena_create()
first_cell = lib.arena_cells(arena)[lib.qcgc_arena_first_cell_index]
size = lib.qcgc_arena_cells_count - lib.qcgc_arena_first_cell_index
lib.qcgc_fit_allocator_add(ffi.addressof(first_cell), size)
lib.bump_ptr_reset()
p = self.bump_allocate(16)
self.assertEqual(ffi.addressof(first_cell), p)
def test_bump_allocate(self):
for _ in range(2 ** 9):
self.push_root(self.allocate(1))
lib.bump_ptr_reset()
lib.qcgc_collect()
self.assertEqual(lib.qcgc_state.free_cells, self.arena_blocks - 2 ** 9)
self.assertEqual(lib.qcgc_state.largest_free_block, self.arena_blocks - 2 ** 9)
def test_reuse_old_free_space(self):
arena = lib.qcgc_arena_create()
first_cell = lib.arena_cells(arena)[lib.qcgc_arena_first_cell_index]
size = lib.qcgc_arena_cells_count - lib.qcgc_arena_first_cell_index
lib.qcgc_fit_allocator_add(ffi.addressof(first_cell), size)
lib.bump_ptr_reset()
p = self.bump_allocate(16)
self.assertEqual(ffi.addressof(first_cell), p)
def test_weakref_to_prebuilt(self):
prebuilt = self.allocate_prebuilt(1)
wr = self.allocate_weakref(prebuilt)
self.push_root(wr)
# Collect
lib.bump_ptr_reset()
lib.qcgc_collect()
#
self.assertEqual(self.get_ref(wr, 0), prebuilt)
def test_write_barrier_after_sweep(self):
o = self.allocate_ref(1)
self.push_root(o)
#
lib.qcgc_collect()
#
p = self.allocate(1)
self.set_ref(o, 0, p)
#
lib.bump_ptr_reset()
lib.qcgc_collect()
#
self.assertIn(self.get_blocktype(ffi.cast("cell_t *", p)), [lib.BLOCK_WHITE, lib.BLOCK_BLACK])
def test_fit_all_free(self):
for _ in range(10):
self.allocate(1)
self.push_root(self.allocate(1)) # Make it non coalesced
lib.bump_ptr_reset()
lib.qcgc_collect()
self.assertEqual(lib.qcgc_state.free_cells, self.arena_blocks - 10)
for _ in range(10):
self.push_root(lib.qcgc_fit_allocate(1))
lib.qcgc_collect()
self.assertEqual(lib.qcgc_state.free_cells, self.arena_blocks - 20)
def test_bump_all_free(self):
for _ in range(10):
self.push_root(self.allocate(1))
lib.bump_ptr_reset()
lib.qcgc_collect()
self.assertEqual(lib.qcgc_state.free_cells, self.arena_blocks - 10)
#
for _ in range(10):
self.pop_root()
#
lib.qcgc_collect()
self.assertEqual(lib.qcgc_state.free_cells, 0) # As the arena is free now
self.assertEqual(lib.qcgc_state.largest_free_block, 0)
def weakref_to_alive_and_dead(self, alive, dead):
"""Utility to reduce code duplication in test cases. Client is responsible
that alive object remains alive"""
wr_to_alive = self.allocate_weakref(alive)
wr_to_dead = self.allocate_weakref(dead)
self.push_root(wr_to_alive)
self.push_root(wr_to_dead)
# Collect
lib.bump_ptr_reset()
lib.qcgc_collect()
#
self.assertEqual(self.get_ref(wr_to_alive, 0), alive)
self.assertEqual(self.get_ref(wr_to_dead, 0), ffi.NULL)
def test_prebuilt(self):
p = self.allocate_prebuilt(1)
q = self.allocate_prebuilt_ref(4)
r = self.allocate(1)
s = self.allocate_ref(4)
self.set_ref(q, 0, p)
self.set_ref(q, 1, q)
self.set_ref(q, 2, r)
self.set_ref(q, 3, s)
self.set_ref(s, 0, p)
self.set_ref(s, 1, q)
self.set_ref(s, 2, r)
self.set_ref(s, 3, s)
lib.bump_ptr_reset()
lib.qcgc_collect()
self.assertEqual(self.get_blocktype(ffi.cast("cell_t *", r)), lib.BLOCK_WHITE)
self.assertEqual(self.get_blocktype(ffi.cast("cell_t *", s)), lib.BLOCK_WHITE)
def test_prebuilt(self):
p = self.allocate_prebuilt(1)
q = self.allocate_prebuilt_ref(4)
r = self.allocate(1)
s = self.allocate_ref(4)
self.set_ref(q, 0, p)
self.set_ref(q, 1, q)
self.set_ref(q, 2, r)
self.set_ref(q, 3, s)
self.set_ref(s, 0, p)
self.set_ref(s, 1, q)
self.set_ref(s, 2, r)
self.set_ref(s, 3, s)
lib.bump_ptr_reset()
lib.qcgc_collect()
self.assertEqual(self.get_blocktype(ffi.cast("cell_t *", r)),
lib.BLOCK_WHITE)
self.assertEqual(self.get_blocktype(ffi.cast("cell_t *", s)),
lib.BLOCK_WHITE)
def test_mark_large(self):
o = ffi.cast("object_t *", self.allocate(lib.qcgc_arena_size))
self.push_root(o)
p = ffi.cast("object_t *", self.allocate_ref(lib.qcgc_arena_size // ffi.sizeof("myobject_t *")))
self.push_root(p)
q = ffi.cast("object_t *", self.allocate(lib.qcgc_arena_size))
self.push_root(q)
r = ffi.cast("object_t *", self.allocate(1))
self.push_root(r)
s = ffi.cast("object_t *", self.allocate_ref(1))
self.push_root(s)
t = ffi.cast("object_t *", self.allocate(lib.qcgc_arena_size))
self.push_root(t)
self.set_ref(p, 0, q)
self.set_ref(p, 1, r)
self.set_ref(p, 2, s)
self.set_ref(s, 0, p)
#
for _ in range(6):
self.pop_root()
self.push_root(o)
self.push_root(s)
#
lib.qcgc_mark()
#
self.assertTrue(self.hbtable_has(o))
self.assertTrue(self.hbtable_marked(o))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(p))
self.assertTrue(self.hbtable_marked(p))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(q))
self.assertTrue(self.hbtable_marked(q))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(t))
self.assertFalse(self.hbtable_marked(t))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertEqual(self.get_blocktype(
ffi.cast("cell_t *", s)), lib.BLOCK_BLACK)
self.assertEqual(self.get_blocktype(
ffi.cast("cell_t *", r)), lib.BLOCK_BLACK)
#
lib.bump_ptr_reset()
lib.qcgc_sweep()
#
self.assertTrue(self.hbtable_has(o))
self.assertFalse(self.hbtable_marked(o))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(p))
self.assertFalse(self.hbtable_marked(p))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(q))
self.assertFalse(self.hbtable_marked(q))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertFalse(self.hbtable_has(t))
self.assertFalse(self.hbtable_marked(t))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertEqual(self.get_blocktype(
ffi.cast("cell_t *", s)), lib.BLOCK_WHITE)
self.assertEqual(self.get_blocktype(
ffi.cast("cell_t *", r)), lib.BLOCK_WHITE)
def test_mark_large(self):
o = ffi.cast("object_t *", self.allocate(lib.qcgc_arena_size))
self.push_root(o)
p = ffi.cast(
"object_t *",
self.allocate_ref(lib.qcgc_arena_size //
ffi.sizeof("myobject_t *")))
self.push_root(p)
q = ffi.cast("object_t *", self.allocate(lib.qcgc_arena_size))
self.push_root(q)
r = ffi.cast("object_t *", self.allocate(1))
self.push_root(r)
s = ffi.cast("object_t *", self.allocate_ref(1))
self.push_root(s)
t = ffi.cast("object_t *", self.allocate(lib.qcgc_arena_size))
self.push_root(t)
self.set_ref(p, 0, q)
self.set_ref(p, 1, r)
self.set_ref(p, 2, s)
self.set_ref(s, 0, p)
#
for _ in range(6):
self.pop_root()
self.push_root(o)
self.push_root(s)
#
lib.qcgc_mark()
#
self.assertTrue(self.hbtable_has(o))
self.assertTrue(self.hbtable_marked(o))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(p))
self.assertTrue(self.hbtable_marked(p))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(q))
self.assertTrue(self.hbtable_marked(q))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(t))
self.assertFalse(self.hbtable_marked(t))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertEqual(self.get_blocktype(ffi.cast("cell_t *", s)),
lib.BLOCK_BLACK)
self.assertEqual(self.get_blocktype(ffi.cast("cell_t *", r)),
lib.BLOCK_BLACK)
#
lib.bump_ptr_reset()
lib.qcgc_sweep()
#
self.assertTrue(self.hbtable_has(o))
self.assertFalse(self.hbtable_marked(o))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(p))
self.assertFalse(self.hbtable_marked(p))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertTrue(self.hbtable_has(q))
self.assertFalse(self.hbtable_marked(q))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertFalse(self.hbtable_has(t))
self.assertFalse(self.hbtable_marked(t))
self.assertEqual(lib.qcgc_state.gray_stack_size, 0)
#
self.assertEqual(self.get_blocktype(ffi.cast("cell_t *", s)),
lib.BLOCK_WHITE)
self.assertEqual(self.get_blocktype(ffi.cast("cell_t *", r)),
lib.BLOCK_WHITE)