X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Ftestsuite.cpp;h=59e6102e7bcc6ea09d849dbadeeb827c8e5d4f04;hb=34b2dde776b332116d04cea356843b324ab099fb;hp=c61e106b66f5dbbea06bc7e980dd686b60adfcd2;hpb=cd712c40e1b352c05e7ae0f72e0a5e84cdf64323;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/testsuite.cpp b/src/testsuite.cpp
index c61e106b6..59e6102e7 100644
--- a/src/testsuite.cpp
+++ b/src/testsuite.cpp
@@ -1,16 +1,24 @@
-/*	   +------------------------------------+
- *	   | Inspire Internet Relay Chat Daemon |
- *	   +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
  *
- *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
- * See: http://wiki.inspircd.org/Credits
+ *   Copyright (C) 2009 Robin Burchell <robin+git@viroteck.net>
+ *   Copyright (C) 2008 Craig Edwards <craigedwards@brainbox.cc>
+ *   Copyright (C) 2008 Dennis Friis <peavey@inspircd.org>
  *
- * This program is free but copyrighted software; see
- *		  the file COPYING for details.
+ * This file is part of InspIRCd.  InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
  *
- * ---------------------------------------------------
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+
 /* $Core */
 
 #include "inspircd.h"
@@ -57,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";
 
@@ -97,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;
@@ -319,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";