[Neo-report] r2138 gregory - in /trunk: neo/tests/storage/ tools/

[nobody at svn.erp5.org]

Author: gregory
Date: Mon Jun  7 14:34:08 2010
New Revision: 2138

Log:
Add tests for storage identification handler.

Added:
    trunk/neo/tests/storage/testIdentificationHandler.py
Modified:
    trunk/tools/runner

Added: trunk/neo/tests/storage/testIdentificationHandler.py
==============================================================================
--- trunk/neo/tests/storage/testIdentificationHandler.py (added)
+++ trunk/neo/tests/storage/testIdentificationHandler.py [iso-8859-1] Mon Jun  7 14:34:08 2010
@@ -1,0 +1,94 @@
+#
+# Copyright (C) 2009-2010  Nexedi SA
+#
+# This program 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; either version 2
+# of the License, or (at your option) any later version.
+#
+# 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, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+
+import unittest
+from mock import Mock
+from neo.tests import NeoTestBase
+from neo.protocol import NodeTypes, NotReadyError, BrokenNodeDisallowedError
+from neo.pt import PartitionTable
+from neo.storage.app import Application
+from neo.storage.handlers.identification import IdentificationHandler
+
+class StorageIdentificationHandlerTests(NeoTestBase):
+
+    def setUp(self):
+        config = self.getStorageConfiguration(master_number=1)
+        self.app = Application(config)
+        self.app.name = 'NEO'
+        self.app.ready = True
+        self.app.pt = PartitionTable(4, 1)
+        self.identification = IdentificationHandler(self.app)
+
+    def test_requestIdentification1(self):
+        """ nodes are rejected during election or if unknown storage """
+        self.app.ready = False
+        self.assertRaises(
+                NotReadyError,
+                self.identification.requestIdentification,
+                self.getFakeConnection(),
+                NodeTypes.CLIENT,
+                self.getNewUUID(),
+                None,
+                self.app.name,
+        )
+        self.app.ready = True
+        self.assertRaises(
+                NotReadyError,
+                self.identification.requestIdentification,
+                self.getFakeConnection(),
+                NodeTypes.STORAGE,
+                self.getNewUUID(),
+                None,
+                self.app.name,
+        )
+
+    def test_requestIdentification3(self):
+        """ broken nodes must be rejected """
+        uuid = self.getNewUUID()
+        conn = self.getFakeConnection(uuid=uuid)
+        node = self.app.nm.createClient(uuid=uuid)
+        node.setBroken()
+        self.assertRaises(BrokenNodeDisallowedError,
+                self.identification.requestIdentification,
+                conn,
+                NodeTypes.CLIENT,
+                uuid,
+                None,
+                self.app.name,
+        )
+
+    def test_requestIdentification2(self):
+        """ accepted client must be connected and running """
+        uuid = self.getNewUUID()
+        conn = self.getFakeConnection(uuid=uuid)
+        node = self.app.nm.createClient(uuid=uuid)
+        self.identification.requestIdentification(conn, NodeTypes.CLIENT, uuid,
+                None, self.app.name)
+        self.assertTrue(node.isRunning())
+        self.assertTrue(node.isConnected())
+        self.assertEqual(node.getUUID(), uuid)
+        self.assertTrue(node.getConnection() is conn)
+        self.checkUUIDSet(conn, uuid)
+        args = self.checkAcceptIdentification(conn, decode=True)
+        node_type, address, _np, _nr, _uuid = args
+        self.assertEqual(node_type, NodeTypes.STORAGE)
+        self.assertEqual(address, None)
+        self.assertEqual(_uuid, uuid)
+
+
+if __name__ == "__main__":
+    unittest.main()

Modified: trunk/tools/runner
==============================================================================
--- trunk/tools/runner [iso-8859-1] (original)
+++ trunk/tools/runner [iso-8859-1] Mon Jun  7 14:34:08 2010
@@ -54,6 +54,7 @@
     'neo.tests.storage.testStorageHandler',
     'neo.tests.storage.testStorageMySQLdb',
     'neo.tests.storage.testVerificationHandler',
+    'neo.tests.storage.testIdentificationHandler',
     'neo.tests.storage.testTransactions',
     # client application
     'neo.tests.client.testClientApp',