1// Copyright 2011 The Kyua Authors. 2// All rights reserved. 3// 4// Redistribution and use in source and binary forms, with or without 5// modification, are permitted provided that the following conditions are 6// met: 7// 8// * Redistributions of source code must retain the above copyright 9// notice, this list of conditions and the following disclaimer. 10// * Redistributions in binary form must reproduce the above copyright 11// notice, this list of conditions and the following disclaimer in the 12// documentation and/or other materials provided with the distribution. 13// * Neither the name of Google Inc. nor the names of its contributors 14// may be used to endorse or promote products derived from this software 15// without specific prior written permission. 16// 17// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 29#include "store/migrate.hpp" 30 31extern "C" { 32#include <sys/stat.h> 33} 34 35#include <atf-c++.hpp> 36 37#include "store/exceptions.hpp" 38#include "utils/env.hpp" 39#include "utils/fs/operations.hpp" 40#include "utils/fs/path.hpp" 41 42namespace fs = utils::fs; 43 44 45ATF_TEST_CASE_WITHOUT_HEAD(detail__backup_database__ok); 46ATF_TEST_CASE_BODY(detail__backup_database__ok) 47{ 48 atf::utils::create_file("test.db", "The DB\n"); 49 store::detail::backup_database(fs::path("test.db"), 13); 50 ATF_REQUIRE(fs::exists(fs::path("test.db"))); 51 ATF_REQUIRE(fs::exists(fs::path("test.db.v13.backup"))); 52 ATF_REQUIRE(atf::utils::compare_file("test.db.v13.backup", "The DB\n")); 53} 54 55 56ATF_TEST_CASE_WITHOUT_HEAD(detail__backup_database__ok_overwrite); 57ATF_TEST_CASE_BODY(detail__backup_database__ok_overwrite) 58{ 59 atf::utils::create_file("test.db", "Original contents"); 60 atf::utils::create_file("test.db.v1.backup", "Overwrite me"); 61 store::detail::backup_database(fs::path("test.db"), 1); 62 ATF_REQUIRE(fs::exists(fs::path("test.db"))); 63 ATF_REQUIRE(fs::exists(fs::path("test.db.v1.backup"))); 64 ATF_REQUIRE(atf::utils::compare_file("test.db.v1.backup", 65 "Original contents")); 66} 67 68 69ATF_TEST_CASE_WITHOUT_HEAD(detail__backup_database__fail_open); 70ATF_TEST_CASE_BODY(detail__backup_database__fail_open) 71{ 72 ATF_REQUIRE_THROW_RE(store::error, "Cannot open.*foo.db", 73 store::detail::backup_database(fs::path("foo.db"), 5)); 74} 75 76 77ATF_TEST_CASE_WITH_CLEANUP(detail__backup_database__fail_create); 78ATF_TEST_CASE_HEAD(detail__backup_database__fail_create) 79{ 80 set_md_var("require.user", "unprivileged"); 81} 82ATF_TEST_CASE_BODY(detail__backup_database__fail_create) 83{ 84 ATF_REQUIRE(::mkdir("dir", 0755) != -1); 85 atf::utils::create_file("dir/test.db", "Does not need to be valid"); 86 ATF_REQUIRE(::chmod("dir", 0111) != -1); 87 ATF_REQUIRE_THROW_RE( 88 store::error, "Cannot create.*dir/test.db.v13.backup", 89 store::detail::backup_database(fs::path("dir/test.db"), 13)); 90} 91ATF_TEST_CASE_CLEANUP(detail__backup_database__fail_create) 92{ 93 if (::chmod("dir", 0755) == -1) { 94 // If we cannot restore the original permissions, we cannot do much 95 // more. However, leaving an unwritable directory behind will cause the 96 // runtime engine to report us as broken. 97 } 98} 99 100 101ATF_TEST_CASE_WITHOUT_HEAD(detail__migration_file__builtin); 102ATF_TEST_CASE_BODY(detail__migration_file__builtin) 103{ 104 utils::unsetenv("KYUA_STOREDIR"); 105 ATF_REQUIRE_EQ(fs::path(KYUA_STOREDIR) / "migrate_v5_v9.sql", 106 store::detail::migration_file(5, 9)); 107} 108 109 110ATF_TEST_CASE_WITHOUT_HEAD(detail__migration_file__overriden); 111ATF_TEST_CASE_BODY(detail__migration_file__overriden) 112{ 113 utils::setenv("KYUA_STOREDIR", "/tmp/test"); 114 ATF_REQUIRE_EQ(fs::path("/tmp/test/migrate_v5_v9.sql"), 115 store::detail::migration_file(5, 9)); 116} 117 118 119ATF_INIT_TEST_CASES(tcs) 120{ 121 ATF_ADD_TEST_CASE(tcs, detail__backup_database__ok); 122 ATF_ADD_TEST_CASE(tcs, detail__backup_database__ok_overwrite); 123 ATF_ADD_TEST_CASE(tcs, detail__backup_database__fail_open); 124 ATF_ADD_TEST_CASE(tcs, detail__backup_database__fail_create); 125 126 ATF_ADD_TEST_CASE(tcs, detail__migration_file__builtin); 127 ATF_ADD_TEST_CASE(tcs, detail__migration_file__overriden); 128 129 // Tests for migrate_schema are in schema_inttest. This is because, for 130 // such tests to be meaningful, they need to be integration tests and don't 131 // really fit the goal of this unit-test module. 132} 133