From 8ce201d0207a93d26c19fd136df4040f11ebcef1 Mon Sep 17 00:00:00 2001 From: open-license-manager <rillf@maildrop.cc> Date: 周二, 14 10月 2014 06:58:34 +0800 Subject: [PATCH] windows ok --- test/functional/volid_test.cpp | 45 +++++++++++++++++++++++++++++++-------------- 1 files changed, 31 insertions(+), 14 deletions(-) diff --git a/test/functional/volid_test.cpp b/test/functional/volid_test.cpp index 6aa87d1..642f361 100644 --- a/test/functional/volid_test.cpp +++ b/test/functional/volid_test.cpp @@ -1,17 +1,17 @@ #define BOOST_TEST_MODULE standard_license_test //#define BOOST_TEST_MAIN -//#define BOOST_TEST_DYN_LINK +//#undef BOOST_TEST_DYN_LINK #include <boost/test/unit_test.hpp> #include <fstream> +#include <stdio.h> #include <cstring> -#include "../../../src/license-generator/license-generator.h" +#include "../../../src/tools/license-generator/license-generator.h" #include "../../../src/library/api/license++.h" #include <build_properties.h> #include <boost/filesystem.hpp> #include "../../src/library/ini/SimpleIni.h" #include "generate-license.h" #include "../../src/library/pc-identifiers.h" -#include <dbus-1.0/dbus/dbus.h> namespace fs = boost::filesystem; using namespace license; @@ -22,6 +22,7 @@ PcSignature identifier_out; IDENTIFICATION_STRATEGY strategy = IDENTIFICATION_STRATEGY::ETHERNET; + BOOST_CHECKPOINT("Before generate"); FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out, strategy); BOOST_ASSERT(generate_ok == FUNCTION_RETURN::FUNC_RET_OK); @@ -39,7 +40,7 @@ BOOST_CHECK_EQUAL(license.has_expiry, false); BOOST_CHECK_EQUAL(license.linked_to_pc, true); } - + static void generate_reference_file(const string& idfileLocation, IDENTIFICATION_STRATEGY strategies[], int num_strategies) { ofstream idfile(idfileLocation); @@ -47,7 +48,12 @@ for (int i = 0; i < num_strategies; i++) { FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out, strategies[i]); - BOOST_ASSERT(generate_ok == FUNCTION_RETURN::FUNC_RET_OK); + if (generate_ok != FUNC_RET_OK){ + idfile.close(); + remove(idfileLocation.c_str()); + BOOST_ERROR("Generating identifier for strategy " << strategies[i] << "failed with: " << generate_ok); + } + BOOST_ASSERT(generate_ok == FUNC_RET_OK); idfile << identifier_out << endl; } idfile.close(); @@ -56,19 +62,30 @@ BOOST_AUTO_TEST_CASE(generated_identifiers_stability) { const string idfileLocation(PROJECT_TEST_TEMP_DIR "/identifiers_file"); IDENTIFICATION_STRATEGY strategies[] = - { IDENTIFICATION_STRATEGY::DEFAULT, - IDENTIFICATION_STRATEGY::DISK_LABEL, - IDENTIFICATION_STRATEGY::DISK_NUM, - IDENTIFICATION_STRATEGY::ETHERNET }; + { DEFAULT, + DISK_LABEL, + DISK_NUM, + ETHERNET }; const int num_strategies = sizeof(strategies) / sizeof(strategies[0]); std::ifstream test_idfile_exist(idfileLocation); if (!test_idfile_exist.good()) { generate_reference_file(idfileLocation, strategies, num_strategies); } - std::ifstream is(idfileLocation); - std::istream_iterator<string> start(is), end; - std::vector<string> reference_signatures(start, end); - BOOST_ASSERT(reference_signatures.size() == num_strategies); + //try to locate a "good" reference file. + int tries = 2; + std::vector<string> reference_signatures; + do{ + std::ifstream is(idfileLocation); + std::istream_iterator<string> start(is), end; + reference_signatures = vector<string>(start, end); + if (reference_signatures.size() != num_strategies){ + is.close(); + remove(idfileLocation.c_str()); + generate_reference_file(idfileLocation, strategies, num_strategies); + } + } while (reference_signatures.size() != num_strategies && tries-- > 0); + BOOST_ASSERT(tries > 0); + PcSignature generated_identifier; BOOST_CHECKPOINT("Generating current signatures and comparing with past"); for (int i = 0; i < num_strategies; i++) { @@ -77,7 +94,7 @@ BOOST_ASSERT(generate_ok == FUNCTION_RETURN::FUNC_RET_OK); if (reference_signatures[i] != generated_identifier) { string message = string("pc signature compare fail: strategy:") - + to_string(strategies[i]) + " generated: [" + + to_string((long double) strategies[i]) + " generated: [" + generated_identifier + "] reference: [" + reference_signatures[i] + "]"; BOOST_FAIL(message); -- Gitblit v1.9.1