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 | 52 ++++++++++++++++++++++++++++++++++++----------------
1 files changed, 36 insertions(+), 16 deletions(-)
diff --git a/test/functional/volid_test.cpp b/test/functional/volid_test.cpp
index 996fbb6..642f361 100644
--- a/test/functional/volid_test.cpp
+++ b/test/functional/volid_test.cpp
@@ -1,10 +1,11 @@
#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>
@@ -21,9 +22,10 @@
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::OK);
+ BOOST_ASSERT(generate_ok == FUNCTION_RETURN::FUNC_RET_OK);
cout << "Identifier:" << identifier_out << endl;
vector<string> extraArgs = { "-s", identifier_out };
generate_license(licLocation, extraArgs);
@@ -38,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);
@@ -46,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::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();
@@ -55,28 +62,39 @@
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++) {
FUNCTION_RETURN generate_ok = generate_user_pc_signature(
generated_identifier, strategies[i]);
- BOOST_ASSERT(generate_ok == FUNCTION_RETURN::OK);
+ 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);
@@ -87,7 +105,8 @@
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));
+ 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
@@ -97,3 +116,4 @@
}
}
}
+
--
Gitblit v1.9.1