add another test for referee

Author: kelvich

tests2/lib/failure_injector.py

@@ -81,3 +81,30 @@ class SkewTime(FailureInjector):
     def __init__(self, node):
         self.node = node
         super().__init__()
+
+class StopNode(FailureInjector):
+
+    def __init__(self, node):
+        self.node = node
+        super().__init__()
+
+    # XXX: Is it really a good idea to call cli.stop inside method called start?
+    def start(self):
+        self.docker_api.stop(self.node)
+
+    def stop(self):
+        return
+
+
+class StartNode(FailureInjector):
+
+    def __init__(self, node):
+        self.node = node
+        super().__init__()
+
+    # XXX: Is it really a good idea to call cli.stop inside method called start?
+    def start(self):
+        return
+
+    def stop(self):
+        self.docker_api.start(self.node)

tests2/test_referee.py

@@ -14,7 +14,7 @@
 from lib.test_helper import *
 
 
-class MajorTest(unittest.TestCase, TestHelper):
+class RefereeTest(unittest.TestCase, TestHelper):
 
     @classmethod
     def setUpClass(cls):
@@ -108,6 +108,34 @@ def test_double_failure_referee(self):
         self.assertIsolation(aggs)
 
 
+    def test_winner_restart(self):
+        print('### test_winner_restart ###')
+
+        aggs_failure, aggs = self.performFailure(StopNode('node1'))
+
+        self.assertNoCommits(aggs_failure[:1])
+        self.assertCommits(aggs_failure[1:])
+        self.assertIsolation(aggs_failure)
+
+        self.assertNoCommits(aggs[:1])
+        self.assertCommits(aggs[1:])
+        self.assertIsolation(aggs)
+
+        aggs_failure, aggs = self.performFailure(RestartNode('node2'))
+
+        self.assertNoCommits(aggs_failure)
+        self.assertIsolation(aggs_failure)
+
+        self.assertNoCommits(aggs[:1])
+        self.assertCommits(aggs[1:])
+        self.assertIsolation(aggs)
+
+        # need to start node1 to perform consequent tests
+        docker_api = docker.Client()
+        docker_api.start('node1')
+        time.sleep(20)
+
+
 if __name__ == '__main__':
     unittest.main()