From 60ae840790232c52ea3f456881771edee02d35ac Mon Sep 17 00:00:00 2001 From: Maximilien Siavelis <maximilien.siavelis@simulease.com> Date: 周五, 05 1月 2018 09:10:28 +0800 Subject: [PATCH] fix some compilation issues --- test/functional/volid_test.cpp | 180 ++++++++++++++++++++++++++++++++--------------------------- 1 files changed, 98 insertions(+), 82 deletions(-) diff --git a/test/functional/volid_test.cpp b/test/functional/volid_test.cpp index 4e1e847..e9bf9ae 100644 --- a/test/functional/volid_test.cpp +++ b/test/functional/volid_test.cpp @@ -1,103 +1,119 @@ #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/library/api/license++.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; using namespace std; - BOOST_AUTO_TEST_CASE( default_volid_lic_file ) { - const string licLocation(PROJECT_TEST_TEMP_DIR "/volid_license.lic"); - PcSignature identifier_out; +BOOST_AUTO_TEST_CASE( default_volid_lic_file ) { + const string licLocation(PROJECT_TEST_TEMP_DIR "/volid_license.lic"); + PcSignature identifier_out; - IDENTIFICATION_STRATEGY strategy = IDENTIFICATION_STRATEGY::ETHERNET; - FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out, - strategy); - BOOST_ASSERT(generate_ok == FUNCTION_RETURN::OK); - cout << "Identifier:" << identifier_out << endl; - vector<string> extraArgs = { "-s", identifier_out }; - generate_license(licLocation, extraArgs); + IDENTIFICATION_STRATEGY strategy = IDENTIFICATION_STRATEGY::ETHERNET; + BOOST_TEST_CHECKPOINT("Before generate"); + FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out, + strategy); + BOOST_ASSERT(generate_ok == FUNCTION_RETURN::FUNC_RET_OK); + cout << "Identifier:" << identifier_out << endl; + vector<string> extraArgs = { "-s", identifier_out }; + generate_license(licLocation, extraArgs); - LicenseInfo license; - LicenseLocation licenseLocation; - licenseLocation.openFileNearModule = false; - licenseLocation.licenseFileLocation = licLocation.c_str(); - licenseLocation.environmentVariableName = ""; - EVENT_TYPE result = acquire_license("TEST", licenseLocation, &license); - BOOST_CHECK_EQUAL(result, LICENSE_OK); - BOOST_CHECK_EQUAL(license.has_expiry, false); - BOOST_CHECK_EQUAL(license.linked_to_pc, true); - } + LicenseInfo license; + LicenseLocation licenseLocation; + licenseLocation.openFileNearModule = false; + licenseLocation.licenseFileLocation = licLocation.c_str(); + licenseLocation.environmentVariableName = ""; + EVENT_TYPE result = acquire_license("TEST", licenseLocation, &license); + BOOST_CHECK_EQUAL(result, LICENSE_OK); + 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); + PcSignature identifier_out; + for (int i = 0; i < num_strategies; i++) { + FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out, + strategies[i]); + 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(); +} - static void generate_reference_file(const string& idfileLocation, - IDENTIFICATION_STRATEGY strategies[], int num_strategies) { - ofstream idfile(idfileLocation); - PcSignature identifier_out; - 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::OK); - idfile << identifier_out << endl; - } - idfile.close(); - } +BOOST_AUTO_TEST_CASE(generated_identifiers_stability) { + const string idfileLocation(PROJECT_TEST_TEMP_DIR "/identifiers_file"); + IDENTIFICATION_STRATEGY strategies[] = + { 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); + } + //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); - 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 - }; - 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); - PcSignature generated_identifier; - BOOST_CHECKPOINT("Generating current signatures and comparing with past"); - for (int i = 0; i < num_strategies; i++) { - FUNCTION_RETURN generate_ok = generate_user_pc_signature( - generated_identifier, strategies[i]); - BOOST_ASSERT(generate_ok == FUNCTION_RETURN::OK); - if (reference_signatures[i] != generated_identifier) { - string message = string("pc signature compare fail: strategy:") - + to_string(strategies[i]) + " generated: [" - + generated_identifier + "] reference: [" - + reference_signatures[i] + "]"; - BOOST_FAIL(message); - } - } + PcSignature generated_identifier; + BOOST_TEST_CHECKPOINT("Generating current signatures and comparing with past"); + for (int i = 0; i < num_strategies; i++) { + FUNCTION_RETURN generate_ok = generate_user_pc_signature( + generated_identifier, strategies[i]); + 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((long double) strategies[i]) + " generated: [" + + generated_identifier + "] reference: [" + + reference_signatures[i] + "]"; + BOOST_FAIL(message); + } + } - BOOST_CHECKPOINT("Verifying signatures"); - for (int j = 0; j < 100; j++) { - for (unsigned int i = 0; i < reference_signatures.size(); i++) { - PcSignature pcsig; - strncpy(pcsig, reference_signatures[i].c_str(), sizeof(PcSignature)); - EVENT_TYPE val_result = validate_pc_signature(pcsig); - string message = string("pc signature verification strategy:") - + to_string(i) + " generated: [" + generated_identifier - + "] reference: [" + reference_signatures[i] + "]"; - BOOST_CHECKPOINT("Verifying signature: "); - BOOST_CHECK_EQUAL(val_result, LICENSE_OK); - } - } - } - + BOOST_TEST_CHECKPOINT("Verifying signatures"); + for (int j = 0; j < 100; j++) { + for (unsigned int i = 0; i < reference_signatures.size(); i++) { + PcSignature pcsig; + strncpy(pcsig, reference_signatures[i].c_str(), + sizeof(PcSignature)); + EVENT_TYPE val_result = validate_pc_signature(pcsig); + string message = string("pc signature verification strategy:") + + to_string(i) + " generated: [" + generated_identifier + + "] reference: [" + reference_signatures[i] + "]"; + BOOST_TEST_CHECKPOINT("Verifying signature: "); + BOOST_CHECK_EQUAL(val_result, LICENSE_OK); + } + } +} -- Gitblit v1.9.1