summaryrefslogtreecommitdiff
path: root/src/testsuite.cpp
diff options
context:
space:
mode:
authorRobin Burchell <viroteck@viroteck.net>2012-05-28 04:23:29 -0700
committerRobin Burchell <viroteck@viroteck.net>2012-05-28 04:23:29 -0700
commit20a91b6bc458a9f5c531a93dbe6b8d5083a3fd0d (patch)
tree301df4396157da0479864bf08f1ec6d48f222b55 /src/testsuite.cpp
parenta441c1b1be087a9974dad211aa3706e1909d6ab3 (diff)
parente3e7cb89e112bbeed2b3e43798a16fe557a3992a (diff)
Merge pull request #150 from attilamolnar/insp20+incrementuidfix
[2.0] Fix generating invalid UIDs after current_uid is 000Z99999
Diffstat (limited to 'src/testsuite.cpp')
-rw-r--r--src/testsuite.cpp77
1 files changed, 77 insertions, 0 deletions
diff --git a/src/testsuite.cpp b/src/testsuite.cpp
index 064724392..59e6102e7 100644
--- a/src/testsuite.cpp
+++ b/src/testsuite.cpp
@@ -65,6 +65,7 @@ TestSuite::TestSuite()
cout << "(5) Wildcard and CIDR tests\n";
cout << "(6) Comma sepstream tests\n";
cout << "(7) Space sepstream tests\n";
+ cout << "(8) UID generation tests\n";
cout << endl << "(X) Exit test suite\n";
@@ -105,6 +106,9 @@ TestSuite::TestSuite()
case '7':
cout << (DoSpaceSepStreamTests() ? "\nSUCCESS!\n" : "\nFAILURE\n");
break;
+ case '8':
+ cout << (DoGenerateUIDTests() ? "\nSUCCESS!\n" : "\nFAILURE\n");
+ break;
case 'X':
return;
break;
@@ -327,6 +331,79 @@ bool TestSuite::DoThreadTests()
return true;
}
+bool TestSuite::DoGenerateUIDTests()
+{
+ bool success = RealGenerateUIDTests();
+
+ // Reset the UID generation state so running the tests multiple times won't mess things up
+ for (unsigned int i = 0; i < 3; i++)
+ ServerInstance->current_uid[i] = ServerInstance->Config->sid[i];
+ for (unsigned int i = 3; i < UUID_LENGTH-1; i++)
+ ServerInstance->current_uid[i] = '9';
+
+ ServerInstance->current_uid[UUID_LENGTH-1] = '\0';
+
+ return success;
+}
+
+bool TestSuite::RealGenerateUIDTests()
+{
+ std::string first_uid = ServerInstance->GetUID();
+ if (first_uid.length() != UUID_LENGTH-1)
+ {
+ cout << "GENERATEUID: Generated UID is " << first_uid.length() << " characters long instead of " << UUID_LENGTH-1 << endl;
+ return false;
+ }
+
+ if (ServerInstance->current_uid[UUID_LENGTH-1] != '\0')
+ {
+ cout << "GENERATEUID: The null terminator is missing from the end of current_uid" << endl;
+ return false;
+ }
+
+ // The correct UID when generating one for the first time is ...AAAAAA
+ std::string correct_uid = ServerInstance->Config->sid + std::string(UUID_LENGTH - 4, 'A');
+ if (first_uid != correct_uid)
+ {
+ cout << "GENERATEUID: Generated an invalid first UID: " << first_uid << " instead of " << correct_uid << endl;
+ return false;
+ }
+
+ // Set current_uid to be ...Z99999
+ ServerInstance->current_uid[3] = 'Z';
+ for (unsigned int i = 4; i < UUID_LENGTH-1; i++)
+ ServerInstance->current_uid[i] = '9';
+
+ // Store the UID we'll be incrementing so we can display what's wrong later if necessary
+ std::string before_increment(ServerInstance->current_uid);
+ std::string generated_uid = ServerInstance->GetUID();
+
+ // Correct UID after incrementing ...Z99999 is ...0AAAAA
+ correct_uid = ServerInstance->Config->sid + "0" + std::string(UUID_LENGTH - 5, 'A');
+
+ if (generated_uid != correct_uid)
+ {
+ cout << "GENERATEUID: Generated an invalid UID after incrementing " << before_increment << ": " << generated_uid << " instead of " << correct_uid << endl;
+ return false;
+ }
+
+ // Set current_uid to be ...999999 to see if it rolls over correctly
+ for (unsigned int i = 3; i < UUID_LENGTH-1; i++)
+ ServerInstance->current_uid[i] = '9';
+
+ before_increment.assign(ServerInstance->current_uid);
+ generated_uid = ServerInstance->GetUID();
+
+ // Correct UID after rolling over is the first UID we've generated (...AAAAAA)
+ if (generated_uid != first_uid)
+ {
+ cout << "GENERATEUID: Generated an invalid UID after incrementing " << before_increment << ": " << generated_uid << " instead of " << first_uid << endl;
+ return false;
+ }
+
+ return true;
+}
+
TestSuite::~TestSuite()
{
cout << "\n\n*** END OF TEST SUITE ***\n";