m_callerid Use std::find() where possible

[user/henk/code/inspircd.git] / src / testsuite.cpp

diff --git a/src/testsuite.cpp b/src/testsuite.cpp
index 064724392eb8629dd1f2c9c32666d3a4f98a2387..59e6102e7bcc6ea09d849dbadeeb827c8e5d4f04 100644 (file)
--- 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";