diff options
Diffstat (limited to 'tests/pagedfile_tests.cpp')
| -rw-r--r-- | tests/pagedfile_tests.cpp | 352 |
1 files changed, 352 insertions, 0 deletions
diff --git a/tests/pagedfile_tests.cpp b/tests/pagedfile_tests.cpp new file mode 100644 index 0000000..f54c630 --- /dev/null +++ b/tests/pagedfile_tests.cpp @@ -0,0 +1,352 @@ +#include <string> + +#include "testing.h" +#include "io/PagedFile.h" + +#include <check.h> + +using namespace de; + +std::string existing_file1 = "tests/data/test_file1.dat"; +std::string nonexisting_file = "tests/data/nonexisting_file.dat"; +std::string new_file = "tests/data/new_file.dat"; + + +START_TEST(t_create) +{ + auto pfile = PagedFile::create(new_file, true); + + ck_assert_ptr_nonnull(pfile); + ck_assert_int_eq(pfile->get_page_count(), 0); + ck_assert_int_eq(pfile->get_file_size(), PAGE_SIZE); + + std::string fname = pfile->get_fname(); + ck_assert_str_eq(fname.c_str(), new_file.c_str()); + + delete pfile; +} +END_TEST + + +START_TEST(t_create_fail) +{ + auto pfile = PagedFile::create(nonexisting_file, false); + + ck_assert_ptr_null(pfile); +} +END_TEST + + +START_TEST(t_create_open) +{ + size_t pg_cnt = 10; + ck_assert(initialize_test_file(existing_file1, pg_cnt)); + + auto pfile = PagedFile::create(existing_file1, false); + ck_assert_ptr_nonnull(pfile); + ck_assert_int_eq(pfile->get_page_count(), pg_cnt); + ck_assert_int_eq(pfile->get_file_size(), PAGE_SIZE * (pg_cnt + 1)); + + std::string fname = pfile->get_fname(); + ck_assert_str_eq(fname.c_str(), existing_file1.c_str()); + + delete pfile; +} +END_TEST + + +START_TEST(t_read_page) +{ + size_t pg_cnt = 10; + ck_assert(initialize_test_file(existing_file1, pg_cnt)); + auto pfile = PagedFile::create(existing_file1, false); + ck_assert_ptr_nonnull(pfile); + + char *buffer = (char *) aligned_alloc(SECTOR_SIZE, PAGE_SIZE); + + for (size_t i=1; i<=pg_cnt; i++) { + ck_assert_int_eq(pfile->read_page(i, buffer), 1); + ck_assert_int_eq(*((int *) buffer), i); + } + + ck_assert_int_eq(pfile->read_page(0, buffer), 0); + ck_assert_int_eq(pfile->read_page(11, buffer), 0); + + free(buffer); + delete pfile; +} +END_TEST + + +START_TEST(t_read_pages_sg) +{ + size_t pg_cnt = 20; + ck_assert(initialize_test_file(existing_file1, pg_cnt)); + auto pfile = PagedFile::create(existing_file1, false); + ck_assert_ptr_nonnull(pfile); + + size_t buf_cnt = 10; + std::vector<char *> buffers(buf_cnt); + std::vector<std::pair<PageNum, char*>> reads(buf_cnt); + std::vector<PageNum> to_read = {1, 2, 3, 5, 9, 7, 8, 11, 12, 15}; + for (size_t i=0; i<buf_cnt; i++) { + buffers[i] = (char *) std::aligned_alloc(SECTOR_SIZE, PAGE_SIZE); + reads[i] = {to_read[i], buffers[i]}; + } + + ck_assert_int_eq(pfile->read_pages(reads), 1); + + for (size_t i=0; i<buf_cnt; i++) { + ck_assert_int_eq(*((int*) buffers[i]), to_read[i]); + } + + for (size_t i=0; i<buf_cnt; i++) { + free(buffers[i]); + } + + delete pfile; +} +END_TEST + + +START_TEST(t_read_pages_seq) +{ + size_t pg_cnt = 20; + ck_assert(initialize_test_file(existing_file1, pg_cnt)); + auto pfile = PagedFile::create(existing_file1, false); + ck_assert_ptr_nonnull(pfile); + + PageNum read_cnt = 10; + PageNum start_pg = 5; + char *buffer = (char *) aligned_alloc(SECTOR_SIZE, PAGE_SIZE*read_cnt); + ck_assert_ptr_nonnull(buffer); + + ck_assert_int_eq(pfile->read_pages(start_pg, read_cnt, buffer), 1); + for (size_t i=0; i<read_cnt; i++) { + ck_assert_int_eq(i + start_pg, *((int*) (buffer + PAGE_SIZE * i))); + } + + read_cnt = 25; + ck_assert_int_eq(pfile->read_pages(start_pg, start_pg + read_cnt, buffer), 0); + + // FIXME: This throws a double free error, but otherwise works. + // I'm definitely not freeing this anywhere else, so not sure + // what to make of that. + free(buffer); + delete pfile; +} +END_TEST + +START_TEST(t_allocate_pages) +{ + auto pfile = PagedFile::create(new_file, true); + ck_assert_ptr_nonnull(pfile); + + ck_assert_int_eq(pfile->get_page_count(), 0); + + ck_assert_int_eq(pfile->allocate_pages(1), 1); + ck_assert_int_eq(pfile->get_page_count(), 1); + ck_assert_int_eq(pfile->get_file_size(), 2*PAGE_SIZE); + + ck_assert_int_eq(pfile->allocate_pages(10), 2); + ck_assert_int_eq(pfile->get_page_count(), 11); + ck_assert_int_eq(pfile->get_file_size(), 12*PAGE_SIZE); + + delete pfile; +} +END_TEST + +START_TEST(t_write) +{ + auto pfile = PagedFile::create(new_file, true); + ck_assert_ptr_nonnull(pfile); + + char *buffer = (char *) aligned_alloc(SECTOR_SIZE, PAGE_SIZE); + *((int*) buffer) = 123; + + // writing to an unallocated page fails + ck_assert_int_eq(pfile->write_page(1, buffer), 0); + + pfile->allocate_pages(1); + + ck_assert_int_eq(pfile->write_page(1, buffer), 1); + ck_assert_int_eq(pfile->write_page(2, buffer), 0); + + delete pfile; + + auto pfile2 = PagedFile::create(new_file, false); + ck_assert_int_eq(pfile2->get_page_count(), 1); + + char *buffer2 = (char *) aligned_alloc(SECTOR_SIZE, PAGE_SIZE); + ck_assert_int_eq(pfile2->read_page(1, buffer2), 1); + ck_assert_int_eq(123, *((int*) buffer2)); + + free(buffer); + free(buffer2); + delete pfile2; +} +END_TEST + + +START_TEST(t_write_pages) +{ + auto pfile = PagedFile::create(new_file, true); + ck_assert_ptr_nonnull(pfile); + + PageNum page_cnt = 13; + char *buffer = (char *) aligned_alloc(SECTOR_SIZE, page_cnt*PAGE_SIZE); + + for (size_t i=0; i<page_cnt; i++) { + *((int*) (buffer + PAGE_SIZE * i)) = i; + } + + ck_assert_int_eq(pfile->write_pages(1, 13, buffer), 0); + + pfile->allocate_pages(20); + + PageNum start_pg = 3; + ck_assert_int_eq(pfile->write_pages(start_pg, page_cnt, buffer), 1); + + delete pfile; + free(buffer); + + auto pfile2 = PagedFile::create(new_file, false); + buffer = (char *) aligned_alloc(SECTOR_SIZE, page_cnt*PAGE_SIZE); + + ck_assert_int_eq(pfile2->read_pages(start_pg, page_cnt, buffer), 1); + for (size_t i=0; i<page_cnt; i++) { + ck_assert_int_eq(*((int*) (buffer + PAGE_SIZE * i)), i); + } + + free(buffer); + delete pfile2; +} +END_TEST + + +START_TEST(t_remove) +{ + auto pfile = PagedFile::create(new_file, true); + ck_assert_ptr_nonnull(pfile); + + delete pfile; + + pfile = PagedFile::create(new_file, false); + ck_assert_ptr_nonnull(pfile); + + ck_assert_int_eq(pfile->remove_file(), 1); + delete pfile; + + pfile = PagedFile::create(new_file, false); + ck_assert_ptr_null(pfile); +} +END_TEST + + +START_TEST(t_iterator) +{ + size_t pg_cnt = 20; + ck_assert(initialize_test_file(existing_file1, pg_cnt)); + auto pfile = PagedFile::create(existing_file1, false); + ck_assert_ptr_nonnull(pfile); + + auto iter = pfile->start_scan(); + ck_assert_ptr_nonnull(iter); + size_t i=0; + while (iter->next()) { + i++; + ck_assert_int_eq(i, *((int*) iter->get_item())); + } + + ck_assert_int_eq(i, pg_cnt); + + delete iter; + delete pfile; +} +END_TEST + + +START_TEST(t_iterator_page_range) +{ + size_t pg_cnt = 20; + ck_assert(initialize_test_file(existing_file1, pg_cnt)); + auto pfile = PagedFile::create(existing_file1, false); + ck_assert_ptr_nonnull(pfile); + + auto iter = pfile->start_scan(30, 45); + ck_assert_ptr_null(iter); + + iter = pfile->start_scan(5, 13); + + size_t i=4; + while (iter->next()) { + i++; + ck_assert_int_eq(i, *((int*) iter->get_item())); + } + + ck_assert_int_eq(i, 13); + + delete iter; + delete pfile; +} +END_TEST + + +Suite *unit_testing() +{ + Suite *unit = suite_create("PagedFile Unit Testing"); + TCase *initialize = tcase_create("de::PagedFile::create Testing"); + tcase_add_test(initialize, t_create); + tcase_add_test(initialize, t_create_fail); + tcase_add_test(initialize, t_create_open); + suite_add_tcase(unit, initialize); + + TCase *read = tcase_create("de::PagedFile::read_page(s) Testing"); + tcase_add_test(read, t_read_page); + tcase_add_test(read, t_read_pages_sg); + tcase_add_test(read, t_read_pages_seq); + suite_add_tcase(unit, read); + + TCase *allocate = tcase_create("de::PagedFile::allocate_pages Testing"); + tcase_add_test(allocate, t_allocate_pages); + suite_add_tcase(unit, allocate); + + TCase *write = tcase_create("de::PagedFile::write_page(s) Testing"); + tcase_add_test(write, t_write); + tcase_add_test(write, t_write_pages); + suite_add_tcase(unit, write); + + TCase *remove = tcase_create("de::PagedFile::remove_file Testing"); + tcase_add_test(remove, t_remove); + suite_add_tcase(unit, remove); + + TCase *iter = tcase_create("de::PagedFile::start_scan Testing"); + tcase_add_test(iter, t_iterator); + tcase_add_test(iter, t_iterator_page_range); + suite_add_tcase(unit, iter); + + return unit; +} + + +int run_unit_tests() +{ + int failed = 0; + Suite *unit = unit_testing(); + SRunner *unit_runner = srunner_create(unit); + + srunner_run_all(unit_runner, CK_NORMAL); + failed = srunner_ntests_failed(unit_runner); + srunner_free(unit_runner); + + return failed; +} + + +int main() +{ + int unit_failed = run_unit_tests(); + + return (unit_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE; +} + |