universe@850: /* universe@850: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. universe@850: * universe@850: * Copyright 2024 Mike Becker, Olaf Wintermann All rights reserved. universe@850: * universe@850: * Redistribution and use in source and binary forms, with or without universe@850: * modification, are permitted provided that the following conditions are met: universe@850: * universe@850: * 1. Redistributions of source code must retain the above copyright universe@850: * notice, this list of conditions and the following disclaimer. universe@850: * universe@850: * 2. Redistributions in binary form must reproduce the above copyright universe@850: * notice, this list of conditions and the following disclaimer in the universe@850: * documentation and/or other materials provided with the distribution. universe@850: * universe@850: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" universe@850: * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE universe@850: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE universe@850: * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE universe@850: * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR universe@850: * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF universe@850: * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS universe@850: * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN universe@850: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) universe@850: * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE universe@850: * POSSIBILITY OF SUCH DAMAGE. universe@850: */ universe@850: universe@850: #include "cx/test.h" universe@850: universe@850: #include "cx/iterator.h" universe@850: universe@850: CX_TEST(test_iterator_create) { universe@850: size_t size = 20; universe@850: unsigned array[size]; universe@850: for (unsigned i = 0 ; i < size ; i++) array[i] = i; universe@850: universe@850: CxMutIterator iter = cxIterator(array, sizeof(unsigned), size, false); universe@850: CX_TEST_DO { universe@850: CX_TEST_ASSERT(iter.index == 0); universe@850: CX_TEST_ASSERT(iter.elem_size == sizeof(unsigned)); universe@850: CX_TEST_ASSERT(iter.elem_count == size); universe@850: CX_TEST_ASSERT(iter.src_handle == array); universe@850: CX_TEST_ASSERT(iter.elem_handle == &array[0]); universe@850: CX_TEST_ASSERT(cxIteratorValid(iter)); universe@850: } universe@850: } universe@850: universe@850: CX_TEST(test_iterator_create_null) { universe@850: CxMutIterator iter = cxIterator(NULL, sizeof(unsigned), 47, false); universe@850: CX_TEST_DO { universe@850: CX_TEST_ASSERT(iter.index == 0); universe@850: CX_TEST_ASSERT(iter.elem_size == sizeof(unsigned)); universe@850: CX_TEST_ASSERT(iter.elem_count == 0); universe@850: CX_TEST_ASSERT(iter.src_handle == NULL); universe@850: CX_TEST_ASSERT(iter.elem_handle == NULL); universe@850: CX_TEST_ASSERT(!cxIteratorValid(iter)); universe@850: } universe@850: } universe@850: universe@850: CX_TEST(test_iterator_iterate) { universe@850: size_t size = 20; universe@850: unsigned array[size]; universe@850: for (unsigned i = 0 ; i < size ; i++) array[i] = i; universe@850: universe@850: CxMutIterator iter = cxIterator(array, sizeof(unsigned), size, false); universe@850: CX_TEST_DO { universe@850: unsigned expected = 0; universe@850: cx_foreach(unsigned *, e, iter) { universe@850: CX_TEST_ASSERT(iter.index == expected); universe@850: CX_TEST_ASSERT(*e == expected); universe@850: CX_TEST_ASSERT(iter.elem_size == sizeof(unsigned)); universe@850: CX_TEST_ASSERT(iter.elem_count == size); universe@850: CX_TEST_ASSERT(iter.src_handle == array); universe@850: CX_TEST_ASSERT(iter.elem_handle == &array[expected]); universe@850: expected++; universe@850: } universe@850: CX_TEST_ASSERT(expected == size); universe@850: } universe@850: } universe@850: universe@850: CX_TEST(test_iterator_with_slow_remove) { universe@850: size_t size = 20; universe@850: unsigned array[size]; universe@850: for (unsigned i = 0 ; i < size ; i++) array[i] = i; universe@850: universe@850: size_t elem_counts[] = { universe@850: 20, 20, 19, 19, 18, 18, 17, 17, 16, 16, universe@850: 15, 15, 14, 14, 13, 13, 12, 12, 11, 11 universe@850: }; universe@850: size_t indices[] = { universe@850: 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, universe@850: 6, 6, 7, 7, 8, 8, 9, 9, 10 universe@850: }; universe@850: unsigned expected_result[] = { universe@850: 0, 2, 4, 6, 8, 10, 12, 14, 16, 18 universe@850: }; universe@850: universe@850: CxMutIterator iter = cxIterator(array, sizeof(unsigned), size, true); universe@850: CX_TEST_DO { universe@850: unsigned expected = 0; universe@850: cx_foreach(unsigned *, e, iter) { universe@850: CX_TEST_ASSERT(*e == expected); universe@850: CX_TEST_ASSERT(iter.index == indices[expected]); universe@850: CX_TEST_ASSERT(iter.elem_size == sizeof(unsigned)); universe@850: CX_TEST_ASSERT(iter.elem_count == elem_counts[expected]); universe@850: CX_TEST_ASSERT(iter.src_handle == array); universe@850: CX_TEST_ASSERT(iter.elem_handle == &array[indices[expected]]); universe@850: expected++; universe@850: if (expected % 2 == 0) { universe@850: cxIteratorFlagRemoval(iter); universe@850: } universe@850: } universe@850: CX_TEST_ASSERT(expected == 20); universe@850: CX_TEST_ASSERT(iter.index == 10); universe@850: CX_TEST_ASSERT(iter.elem_count == 10); universe@850: for (unsigned i = 0 ; i < 9 ; i++) { universe@850: CX_TEST_ASSERT(array[i] == expected_result[i]); universe@850: } universe@850: } universe@850: } universe@850: universe@850: CX_TEST(test_iterator_with_fast_remove) { universe@850: size_t size = 20; universe@850: unsigned array[size]; universe@850: for (unsigned i = 0 ; i < size ; i++) array[i] = i; universe@850: universe@850: size_t elem_counts[] = { universe@850: 20, 20, 19, 19, 18, 18, 17, 17, 16, 16, universe@850: 15, 15, 14, 14, 13, 13, 12, 12, 11, 11 universe@850: }; universe@850: size_t indices[] = { universe@850: 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, universe@850: 6, 6, 7, 7, 8, 8, 9, 9, 10 universe@850: }; universe@850: unsigned expected_result[] = { universe@850: 0, 19, 18, 17, 16, 15, 14, 13, 12, 11 universe@850: }; universe@850: unsigned expected_visits[] = { universe@850: 0, 1, 19, 2, 18, 3, 17, 4, 16, 5, universe@850: 15, 6, 14, 7, 13, 8, 12, 9, 11, 10 universe@850: }; universe@850: universe@850: CxMutIterator iter = cxIterator(array, sizeof(unsigned), size, false); universe@850: CX_TEST_DO { universe@850: unsigned expected = 0; universe@850: cx_foreach(unsigned *, e, iter) { universe@850: CX_TEST_ASSERT(*e == expected_visits[expected]); universe@850: CX_TEST_ASSERT(iter.index == indices[expected]); universe@850: CX_TEST_ASSERT(iter.elem_size == sizeof(unsigned)); universe@850: CX_TEST_ASSERT(iter.elem_count == elem_counts[expected]); universe@850: CX_TEST_ASSERT(iter.src_handle == array); universe@850: CX_TEST_ASSERT(iter.elem_handle == &array[indices[expected]]); universe@850: expected++; universe@850: if (expected % 2 == 0) { universe@850: cxIteratorFlagRemoval(iter); universe@850: } universe@850: } universe@850: CX_TEST_ASSERT(expected == 20); universe@850: CX_TEST_ASSERT(iter.index == 10); universe@850: CX_TEST_ASSERT(iter.elem_count == 10); universe@850: for (unsigned i = 0 ; i < 9 ; i++) { universe@850: CX_TEST_ASSERT(array[i] == expected_result[i]); universe@850: } universe@850: } universe@850: } universe@850: universe@850: CxTestSuite *cx_test_suite_iterator(void) { universe@850: CxTestSuite *suite = cx_test_suite_new("iterator"); universe@850: universe@850: cx_test_register(suite, test_iterator_create); universe@850: cx_test_register(suite, test_iterator_iterate); universe@850: cx_test_register(suite, test_iterator_with_slow_remove); universe@850: cx_test_register(suite, test_iterator_with_fast_remove); universe@850: universe@850: return suite; universe@850: } universe@850: