You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@omid.apache.org by fp...@apache.org on 2016/06/17 21:21:08 UTC

[33/51] [partial] incubator-omid-site git commit: [wagon-git] commit to branch asf-site 6/17/16 2:20 PM

http://git-wip-us.apache.org/repos/asf/incubator-omid-site/blob/3a40a657/xref-test/org/apache/omid/tso/TestBatch.html
----------------------------------------------------------------------
diff --git a/xref-test/org/apache/omid/tso/TestBatch.html b/xref-test/org/apache/omid/tso/TestBatch.html
index e9bd07d..e34c8be 100644
--- a/xref-test/org/apache/omid/tso/TestBatch.html
+++ b/xref-test/org/apache/omid/tso/TestBatch.html
@@ -8,164 +8,170 @@
 <body>
 <div id="overview"><a href="../../../../../../target/site/testapidocs/org/apache/omid/tso/TestBatch.html">View Javadoc</a></div><pre>
 
-<a class="jxr_linenumber" name="1" href="#1">1</a>   <em class="jxr_javadoccomment">/**</em>
-<a class="jxr_linenumber" name="2" href="#2">2</a>   <em class="jxr_javadoccomment"> * Licensed to the Apache Software Foundation (ASF) under one</em>
-<a class="jxr_linenumber" name="3" href="#3">3</a>   <em class="jxr_javadoccomment"> * or more contributor license agreements.  See the NOTICE file</em>
-<a class="jxr_linenumber" name="4" href="#4">4</a>   <em class="jxr_javadoccomment"> * distributed with this work for additional information</em>
-<a class="jxr_linenumber" name="5" href="#5">5</a>   <em class="jxr_javadoccomment"> * regarding copyright ownership.  The ASF licenses this file</em>
-<a class="jxr_linenumber" name="6" href="#6">6</a>   <em class="jxr_javadoccomment"> * to you under the Apache License, Version 2.0 (the</em>
-<a class="jxr_linenumber" name="7" href="#7">7</a>   <em class="jxr_javadoccomment"> * "License"); you may not use this file except in compliance</em>
-<a class="jxr_linenumber" name="8" href="#8">8</a>   <em class="jxr_javadoccomment"> * with the License.  You may obtain a copy of the License at</em>
-<a class="jxr_linenumber" name="9" href="#9">9</a>   <em class="jxr_javadoccomment"> *</em>
-<a class="jxr_linenumber" name="10" href="#10">10</a>  <em class="jxr_javadoccomment"> *   <a href="http://www.apache.org/licenses/LICENSE-2.0" target="alexandria_uri">http://www.apache.org/licenses/LICENSE-2.0</a></em>
-<a class="jxr_linenumber" name="11" href="#11">11</a>  <em class="jxr_javadoccomment"> *</em>
-<a class="jxr_linenumber" name="12" href="#12">12</a>  <em class="jxr_javadoccomment"> * Unless required by applicable law or agreed to in writing, software</em>
-<a class="jxr_linenumber" name="13" href="#13">13</a>  <em class="jxr_javadoccomment"> * distributed under the License is distributed on an "AS IS" BASIS,</em>
-<a class="jxr_linenumber" name="14" href="#14">14</a>  <em class="jxr_javadoccomment"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</em>
-<a class="jxr_linenumber" name="15" href="#15">15</a>  <em class="jxr_javadoccomment"> * See the License for the specific language governing permissions and</em>
-<a class="jxr_linenumber" name="16" href="#16">16</a>  <em class="jxr_javadoccomment"> * limitations under the License.</em>
-<a class="jxr_linenumber" name="17" href="#17">17</a>  <em class="jxr_javadoccomment"> */</em>
+<a class="jxr_linenumber" name="1" href="#1">1</a>   <em class="jxr_comment">/*</em>
+<a class="jxr_linenumber" name="2" href="#2">2</a>   <em class="jxr_comment"> * Licensed to the Apache Software Foundation (ASF) under one</em>
+<a class="jxr_linenumber" name="3" href="#3">3</a>   <em class="jxr_comment"> * or more contributor license agreements.  See the NOTICE file</em>
+<a class="jxr_linenumber" name="4" href="#4">4</a>   <em class="jxr_comment"> * distributed with this work for additional information</em>
+<a class="jxr_linenumber" name="5" href="#5">5</a>   <em class="jxr_comment"> * regarding copyright ownership.  The ASF licenses this file</em>
+<a class="jxr_linenumber" name="6" href="#6">6</a>   <em class="jxr_comment"> * to you under the Apache License, Version 2.0 (the</em>
+<a class="jxr_linenumber" name="7" href="#7">7</a>   <em class="jxr_comment"> * "License"); you may not use this file except in compliance</em>
+<a class="jxr_linenumber" name="8" href="#8">8</a>   <em class="jxr_comment"> * with the License.  You may obtain a copy of the License at</em>
+<a class="jxr_linenumber" name="9" href="#9">9</a>   <em class="jxr_comment"> *</em>
+<a class="jxr_linenumber" name="10" href="#10">10</a>  <em class="jxr_comment"> *   <a href="http://www.apache.org/licenses/LICENSE-2.0" target="alexandria_uri">http://www.apache.org/licenses/LICENSE-2.0</a></em>
+<a class="jxr_linenumber" name="11" href="#11">11</a>  <em class="jxr_comment"> *</em>
+<a class="jxr_linenumber" name="12" href="#12">12</a>  <em class="jxr_comment"> * Unless required by applicable law or agreed to in writing, software</em>
+<a class="jxr_linenumber" name="13" href="#13">13</a>  <em class="jxr_comment"> * distributed under the License is distributed on an "AS IS" BASIS,</em>
+<a class="jxr_linenumber" name="14" href="#14">14</a>  <em class="jxr_comment"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</em>
+<a class="jxr_linenumber" name="15" href="#15">15</a>  <em class="jxr_comment"> * See the License for the specific language governing permissions and</em>
+<a class="jxr_linenumber" name="16" href="#16">16</a>  <em class="jxr_comment"> * limitations under the License.</em>
+<a class="jxr_linenumber" name="17" href="#17">17</a>  <em class="jxr_comment"> */</em>
 <a class="jxr_linenumber" name="18" href="#18">18</a>  <strong class="jxr_keyword">package</strong> org.apache.omid.tso;
 <a class="jxr_linenumber" name="19" href="#19">19</a>  
-<a class="jxr_linenumber" name="20" href="#20">20</a>  <strong class="jxr_keyword">import</strong> org.apache.omid.metrics.MetricsRegistry;
-<a class="jxr_linenumber" name="21" href="#21">21</a>  <strong class="jxr_keyword">import</strong> org.apache.omid.metrics.NullMetricsProvider;
-<a class="jxr_linenumber" name="22" href="#22">22</a>  <strong class="jxr_keyword">import</strong> org.jboss.netty.channel.Channel;
-<a class="jxr_linenumber" name="23" href="#23">23</a>  <strong class="jxr_keyword">import</strong> org.mockito.Mock;
-<a class="jxr_linenumber" name="24" href="#24">24</a>  <strong class="jxr_keyword">import</strong> org.mockito.Mockito;
-<a class="jxr_linenumber" name="25" href="#25">25</a>  <strong class="jxr_keyword">import</strong> org.mockito.MockitoAnnotations;
-<a class="jxr_linenumber" name="26" href="#26">26</a>  <strong class="jxr_keyword">import</strong> org.slf4j.Logger;
-<a class="jxr_linenumber" name="27" href="#27">27</a>  <strong class="jxr_keyword">import</strong> org.slf4j.LoggerFactory;
-<a class="jxr_linenumber" name="28" href="#28">28</a>  <strong class="jxr_keyword">import</strong> org.testng.Assert;
-<a class="jxr_linenumber" name="29" href="#29">29</a>  <strong class="jxr_keyword">import</strong> org.testng.AssertJUnit;
-<a class="jxr_linenumber" name="30" href="#30">30</a>  <strong class="jxr_keyword">import</strong> org.testng.annotations.BeforeMethod;
-<a class="jxr_linenumber" name="31" href="#31">31</a>  <strong class="jxr_keyword">import</strong> org.testng.annotations.Test;
-<a class="jxr_linenumber" name="32" href="#32">32</a>  
-<a class="jxr_linenumber" name="33" href="#33">33</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Matchers.any;
-<a class="jxr_linenumber" name="34" href="#34">34</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Matchers.anyLong;
-<a class="jxr_linenumber" name="35" href="#35">35</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Matchers.eq;
-<a class="jxr_linenumber" name="36" href="#36">36</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.timeout;
-<a class="jxr_linenumber" name="37" href="#37">37</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.verify;
-<a class="jxr_linenumber" name="38" href="#38">38</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertEquals;
-<a class="jxr_linenumber" name="39" href="#39">39</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertFalse;
-<a class="jxr_linenumber" name="40" href="#40">40</a>  
-<a class="jxr_linenumber" name="41" href="#41">41</a>  <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">class</strong> <a href="../../../../org/apache/omid/tso/TestBatch.html">TestBatch</a> {
+<a class="jxr_linenumber" name="20" href="#20">20</a>  <strong class="jxr_keyword">import</strong> org.apache.commons.pool2.PooledObject;
+<a class="jxr_linenumber" name="21" href="#21">21</a>  <strong class="jxr_keyword">import</strong> org.jboss.netty.channel.Channel;
+<a class="jxr_linenumber" name="22" href="#22">22</a>  <strong class="jxr_keyword">import</strong> org.mockito.Mock;
+<a class="jxr_linenumber" name="23" href="#23">23</a>  <strong class="jxr_keyword">import</strong> org.slf4j.Logger;
+<a class="jxr_linenumber" name="24" href="#24">24</a>  <strong class="jxr_keyword">import</strong> org.slf4j.LoggerFactory;
+<a class="jxr_linenumber" name="25" href="#25">25</a>  <strong class="jxr_keyword">import</strong> org.testng.annotations.BeforeMethod;
+<a class="jxr_linenumber" name="26" href="#26">26</a>  <strong class="jxr_keyword">import</strong> org.testng.annotations.Test;
+<a class="jxr_linenumber" name="27" href="#27">27</a>  
+<a class="jxr_linenumber" name="28" href="#28">28</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.MockitoAnnotations.initMocks;
+<a class="jxr_linenumber" name="29" href="#29">29</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertEquals;
+<a class="jxr_linenumber" name="30" href="#30">30</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertFalse;
+<a class="jxr_linenumber" name="31" href="#31">31</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertTrue;
+<a class="jxr_linenumber" name="32" href="#32">32</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.fail;
+<a class="jxr_linenumber" name="33" href="#33">33</a>  
+<a class="jxr_linenumber" name="34" href="#34">34</a>  <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">class</strong> <a href="../../../../org/apache/omid/tso/TestBatch.html">TestBatch</a> {
+<a class="jxr_linenumber" name="35" href="#35">35</a>  
+<a class="jxr_linenumber" name="36" href="#36">36</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> Logger LOG = LoggerFactory.getLogger(TestBatch.<strong class="jxr_keyword">class</strong>);
+<a class="jxr_linenumber" name="37" href="#37">37</a>  
+<a class="jxr_linenumber" name="38" href="#38">38</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">int</strong> BATCH_SIZE = 1000;
+<a class="jxr_linenumber" name="39" href="#39">39</a>  
+<a class="jxr_linenumber" name="40" href="#40">40</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> ANY_ST = 1231;
+<a class="jxr_linenumber" name="41" href="#41">41</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> ANY_CT = 2241;
 <a class="jxr_linenumber" name="42" href="#42">42</a>  
-<a class="jxr_linenumber" name="43" href="#43">43</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> Logger LOG = LoggerFactory.getLogger(TestBatch.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="44" href="#44">44</a>  
-<a class="jxr_linenumber" name="45" href="#45">45</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">int</strong> BATCH_SIZE = 1000;
-<a class="jxr_linenumber" name="46" href="#46">46</a>      <strong class="jxr_keyword">private</strong> MetricsRegistry metrics = <strong class="jxr_keyword">new</strong> NullMetricsProvider();
+<a class="jxr_linenumber" name="43" href="#43">43</a>      @Mock
+<a class="jxr_linenumber" name="44" href="#44">44</a>      <strong class="jxr_keyword">private</strong> Channel channel;
+<a class="jxr_linenumber" name="45" href="#45">45</a>      @Mock
+<a class="jxr_linenumber" name="46" href="#46">46</a>      <strong class="jxr_keyword">private</strong> MonitoringContext monCtx;
 <a class="jxr_linenumber" name="47" href="#47">47</a>  
-<a class="jxr_linenumber" name="48" href="#48">48</a>      @Mock
-<a class="jxr_linenumber" name="49" href="#49">49</a>      <strong class="jxr_keyword">private</strong> Channel channel;
-<a class="jxr_linenumber" name="50" href="#50">50</a>      @Mock
-<a class="jxr_linenumber" name="51" href="#51">51</a>      <strong class="jxr_keyword">private</strong> RetryProcessor retryProcessor;
-<a class="jxr_linenumber" name="52" href="#52">52</a>      @Mock
-<a class="jxr_linenumber" name="53" href="#53">53</a>      <strong class="jxr_keyword">private</strong> ReplyProcessor replyProcessor;
-<a class="jxr_linenumber" name="54" href="#54">54</a>  
-<a class="jxr_linenumber" name="55" href="#55">55</a>      <em class="jxr_comment">// The batch element to test</em>
-<a class="jxr_linenumber" name="56" href="#56">56</a>      <strong class="jxr_keyword">private</strong> PersistenceProcessorImpl.Batch batch;
-<a class="jxr_linenumber" name="57" href="#57">57</a>  
-<a class="jxr_linenumber" name="58" href="#58">58</a>      @BeforeMethod(alwaysRun = <strong class="jxr_keyword">true</strong>, timeOut = 30_000)
-<a class="jxr_linenumber" name="59" href="#59">59</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> initMocksAndComponents() {
-<a class="jxr_linenumber" name="60" href="#60">60</a>          MockitoAnnotations.initMocks(<strong class="jxr_keyword">this</strong>);
-<a class="jxr_linenumber" name="61" href="#61">61</a>          batch = <strong class="jxr_keyword">new</strong> PersistenceProcessorImpl.Batch(BATCH_SIZE);
-<a class="jxr_linenumber" name="62" href="#62">62</a>      }
+<a class="jxr_linenumber" name="48" href="#48">48</a>      @BeforeMethod
+<a class="jxr_linenumber" name="49" href="#49">49</a>      <strong class="jxr_keyword">void</strong> setup() {
+<a class="jxr_linenumber" name="50" href="#50">50</a>          initMocks(<strong class="jxr_keyword">this</strong>);
+<a class="jxr_linenumber" name="51" href="#51">51</a>      }
+<a class="jxr_linenumber" name="52" href="#52">52</a>  
+<a class="jxr_linenumber" name="53" href="#53">53</a>      @Test(timeOut = 10_000)
+<a class="jxr_linenumber" name="54" href="#54">54</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testBatchFunctionality() {
+<a class="jxr_linenumber" name="55" href="#55">55</a>  
+<a class="jxr_linenumber" name="56" href="#56">56</a>          <em class="jxr_comment">// Component to test</em>
+<a class="jxr_linenumber" name="57" href="#57">57</a>          Batch batch = <strong class="jxr_keyword">new</strong> Batch(0, BATCH_SIZE);
+<a class="jxr_linenumber" name="58" href="#58">58</a>  
+<a class="jxr_linenumber" name="59" href="#59">59</a>          <em class="jxr_comment">// Test initial state is OK</em>
+<a class="jxr_linenumber" name="60" href="#60">60</a>          assertTrue(batch.isEmpty(), <span class="jxr_string">"Batch should be empty"</span>);
+<a class="jxr_linenumber" name="61" href="#61">61</a>          assertFalse(batch.isFull(), <span class="jxr_string">"Batch shouldn't be full"</span>);
+<a class="jxr_linenumber" name="62" href="#62">62</a>          assertEquals(batch.getNumEvents(), 0, <span class="jxr_string">"Num events should be 0"</span>);
 <a class="jxr_linenumber" name="63" href="#63">63</a>  
-<a class="jxr_linenumber" name="64" href="#64">64</a>      @Test
-<a class="jxr_linenumber" name="65" href="#65">65</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testBatchFunctionality() {
-<a class="jxr_linenumber" name="66" href="#66">66</a>  
-<a class="jxr_linenumber" name="67" href="#67">67</a>          <em class="jxr_comment">// Required mocks</em>
-<a class="jxr_linenumber" name="68" href="#68">68</a>          Channel channel = Mockito.mock(Channel.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="69" href="#69">69</a>          ReplyProcessor replyProcessor = Mockito.mock(ReplyProcessor.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="70" href="#70">70</a>          RetryProcessor retryProcessor = Mockito.mock(RetryProcessor.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="71" href="#71">71</a>  
-<a class="jxr_linenumber" name="72" href="#72">72</a>          <em class="jxr_comment">// The batch element to test</em>
-<a class="jxr_linenumber" name="73" href="#73">73</a>          PersistenceProcessorImpl.Batch batch = <strong class="jxr_keyword">new</strong> PersistenceProcessorImpl.Batch(BATCH_SIZE);
-<a class="jxr_linenumber" name="74" href="#74">74</a>  
-<a class="jxr_linenumber" name="75" href="#75">75</a>          <em class="jxr_comment">// Test initial state is OK</em>
-<a class="jxr_linenumber" name="76" href="#76">76</a>          AssertJUnit.assertFalse(<span class="jxr_string">"Batch shouldn't be full"</span>, batch.isFull());
-<a class="jxr_linenumber" name="77" href="#77">77</a>          AssertJUnit.assertEquals(<span class="jxr_string">"Num events should be 0"</span>, 0, batch.getNumEvents());
-<a class="jxr_linenumber" name="78" href="#78">78</a>  
-<a class="jxr_linenumber" name="79" href="#79">79</a>          <em class="jxr_comment">// Test adding a single commit event is OK</em>
-<a class="jxr_linenumber" name="80" href="#80">80</a>          MonitoringContext monCtx = <strong class="jxr_keyword">new</strong> MonitoringContext(metrics);
-<a class="jxr_linenumber" name="81" href="#81">81</a>          monCtx.timerStart(<span class="jxr_string">"commitPersistProcessor"</span>);
-<a class="jxr_linenumber" name="82" href="#82">82</a>          batch.addCommit(0, 1, channel, monCtx);
-<a class="jxr_linenumber" name="83" href="#83">83</a>          AssertJUnit.assertFalse(<span class="jxr_string">"Batch shouldn't be full"</span>, batch.isFull());
-<a class="jxr_linenumber" name="84" href="#84">84</a>          AssertJUnit.assertEquals(<span class="jxr_string">"Num events should be 1"</span>, 1, batch.getNumEvents());
-<a class="jxr_linenumber" name="85" href="#85">85</a>  
-<a class="jxr_linenumber" name="86" href="#86">86</a>          <em class="jxr_comment">// Test when filling the batch with events, batch is full</em>
-<a class="jxr_linenumber" name="87" href="#87">87</a>          <strong class="jxr_keyword">for</strong> (<strong class="jxr_keyword">int</strong> i = 0; i &lt; (BATCH_SIZE - 1); i++) {
-<a class="jxr_linenumber" name="88" href="#88">88</a>              <strong class="jxr_keyword">if</strong> (i % 2 == 0) {
-<a class="jxr_linenumber" name="89" href="#89">89</a>                  monCtx = <strong class="jxr_keyword">new</strong> MonitoringContext(metrics);
-<a class="jxr_linenumber" name="90" href="#90">90</a>                  monCtx.timerStart(<span class="jxr_string">"timestampPersistProcessor"</span>);
-<a class="jxr_linenumber" name="91" href="#91">91</a>                  batch.addTimestamp(i, channel, monCtx);
-<a class="jxr_linenumber" name="92" href="#92">92</a>              } <strong class="jxr_keyword">else</strong> {
-<a class="jxr_linenumber" name="93" href="#93">93</a>                  monCtx = <strong class="jxr_keyword">new</strong> MonitoringContext(metrics);
-<a class="jxr_linenumber" name="94" href="#94">94</a>                  monCtx.timerStart(<span class="jxr_string">"commitPersistProcessor"</span>);
-<a class="jxr_linenumber" name="95" href="#95">95</a>                  batch.addCommit(i, i + 1, channel, monCtx);
-<a class="jxr_linenumber" name="96" href="#96">96</a>              }
-<a class="jxr_linenumber" name="97" href="#97">97</a>          }
-<a class="jxr_linenumber" name="98" href="#98">98</a>          AssertJUnit.assertTrue(<span class="jxr_string">"Batch should be full"</span>, batch.isFull());
-<a class="jxr_linenumber" name="99" href="#99">99</a>          AssertJUnit.assertEquals(<span class="jxr_string">"Num events should be "</span> + BATCH_SIZE, BATCH_SIZE, batch.getNumEvents());
-<a class="jxr_linenumber" name="100" href="#100">100</a> 
-<a class="jxr_linenumber" name="101" href="#101">101</a>         <em class="jxr_comment">// Test an exception is thrown when batch is full and a new element is going to be added</em>
-<a class="jxr_linenumber" name="102" href="#102">102</a>         <strong class="jxr_keyword">try</strong> {
-<a class="jxr_linenumber" name="103" href="#103">103</a>             monCtx = <strong class="jxr_keyword">new</strong> MonitoringContext(metrics);
-<a class="jxr_linenumber" name="104" href="#104">104</a>             monCtx.timerStart(<span class="jxr_string">"commitPersistProcessor"</span>);
-<a class="jxr_linenumber" name="105" href="#105">105</a>             batch.addCommit(0, 1, channel, <strong class="jxr_keyword">new</strong> MonitoringContext(metrics));
-<a class="jxr_linenumber" name="106" href="#106">106</a>             Assert.fail(<span class="jxr_string">"Should throw an IllegalStateException"</span>);
-<a class="jxr_linenumber" name="107" href="#107">107</a>         } <strong class="jxr_keyword">catch</strong> (IllegalStateException e) {
-<a class="jxr_linenumber" name="108" href="#108">108</a>             AssertJUnit.assertEquals(<span class="jxr_string">"message returned doesn't match"</span>, <span class="jxr_string">"batch full"</span>, e.getMessage());
-<a class="jxr_linenumber" name="109" href="#109">109</a>             LOG.debug(<span class="jxr_string">"IllegalStateException catched properly"</span>);
-<a class="jxr_linenumber" name="110" href="#110">110</a>         }
-<a class="jxr_linenumber" name="111" href="#111">111</a> 
-<a class="jxr_linenumber" name="112" href="#112">112</a>         <em class="jxr_comment">// Test that sending replies empties the batch</em>
-<a class="jxr_linenumber" name="113" href="#113">113</a>         <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">boolean</strong> MASTER_INSTANCE = <strong class="jxr_keyword">true</strong>;
-<a class="jxr_linenumber" name="114" href="#114">114</a>         <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">boolean</strong> SHOULD_MAKE_HEURISTIC_DECISSION = <strong class="jxr_keyword">true</strong>;
-<a class="jxr_linenumber" name="115" href="#115">115</a>         batch.sendRepliesAndReset(replyProcessor, retryProcessor, MASTER_INSTANCE);
-<a class="jxr_linenumber" name="116" href="#116">116</a>         verify(replyProcessor, timeout(100).times(BATCH_SIZE / 2))
-<a class="jxr_linenumber" name="117" href="#117">117</a>                 .timestampResponse(anyLong(), any(Channel.<strong class="jxr_keyword">class</strong>), any(MonitoringContext.<strong class="jxr_keyword">class</strong>));
-<a class="jxr_linenumber" name="118" href="#118">118</a>         verify(replyProcessor, timeout(100).times(BATCH_SIZE / 2))
-<a class="jxr_linenumber" name="119" href="#119">119</a>                 .commitResponse(eq(!SHOULD_MAKE_HEURISTIC_DECISSION), anyLong(), anyLong(),
-<a class="jxr_linenumber" name="120" href="#120">120</a>                         any(Channel.<strong class="jxr_keyword">class</strong>), any(MonitoringContext.<strong class="jxr_keyword">class</strong>));
-<a class="jxr_linenumber" name="121" href="#121">121</a>         AssertJUnit.assertFalse(<span class="jxr_string">"Batch shouldn't be full"</span>, batch.isFull());
-<a class="jxr_linenumber" name="122" href="#122">122</a>         AssertJUnit.assertEquals(<span class="jxr_string">"Num events should be 0"</span>, 0, batch.getNumEvents());
-<a class="jxr_linenumber" name="123" href="#123">123</a> 
-<a class="jxr_linenumber" name="124" href="#124">124</a>     }
-<a class="jxr_linenumber" name="125" href="#125">125</a> 
-<a class="jxr_linenumber" name="126" href="#126">126</a>     @Test
-<a class="jxr_linenumber" name="127" href="#127">127</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testBatchFunctionalityWhenMastershipIsLost() {
-<a class="jxr_linenumber" name="128" href="#128">128</a>         Channel channel = Mockito.mock(Channel.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="129" href="#129">129</a> 
-<a class="jxr_linenumber" name="130" href="#130">130</a>         <em class="jxr_comment">// Fill the batch with events till full</em>
-<a class="jxr_linenumber" name="131" href="#131">131</a>         <strong class="jxr_keyword">for</strong> (<strong class="jxr_keyword">int</strong> i = 0; i &lt; BATCH_SIZE; i++) {
-<a class="jxr_linenumber" name="132" href="#132">132</a>             <strong class="jxr_keyword">if</strong> (i % 2 == 0) {
-<a class="jxr_linenumber" name="133" href="#133">133</a>                 MonitoringContext monCtx = <strong class="jxr_keyword">new</strong> MonitoringContext(metrics);
-<a class="jxr_linenumber" name="134" href="#134">134</a>                 monCtx.timerStart(<span class="jxr_string">"timestampPersistProcessor"</span>);
-<a class="jxr_linenumber" name="135" href="#135">135</a>                 batch.addTimestamp(i, channel, monCtx);
-<a class="jxr_linenumber" name="136" href="#136">136</a>             } <strong class="jxr_keyword">else</strong> {
-<a class="jxr_linenumber" name="137" href="#137">137</a>                 MonitoringContext monCtx = <strong class="jxr_keyword">new</strong> MonitoringContext(metrics);
-<a class="jxr_linenumber" name="138" href="#138">138</a>                 monCtx.timerStart(<span class="jxr_string">"commitPersistProcessor"</span>);
-<a class="jxr_linenumber" name="139" href="#139">139</a>                 batch.addCommit(i, i + 1, channel, monCtx);
-<a class="jxr_linenumber" name="140" href="#140">140</a>             }
-<a class="jxr_linenumber" name="141" href="#141">141</a>         }
+<a class="jxr_linenumber" name="64" href="#64">64</a>          <em class="jxr_comment">// Test getting or setting an element in the batch greater than the current number of events is illegal</em>
+<a class="jxr_linenumber" name="65" href="#65">65</a>          <strong class="jxr_keyword">try</strong> {
+<a class="jxr_linenumber" name="66" href="#66">66</a>              batch.get(1);
+<a class="jxr_linenumber" name="67" href="#67">67</a>              fail();
+<a class="jxr_linenumber" name="68" href="#68">68</a>          } <strong class="jxr_keyword">catch</strong> (IllegalStateException ex) {
+<a class="jxr_linenumber" name="69" href="#69">69</a>              <em class="jxr_comment">// Expected, as we can not access elements in the batch greater than the current number of events</em>
+<a class="jxr_linenumber" name="70" href="#70">70</a>          }
+<a class="jxr_linenumber" name="71" href="#71">71</a>          <strong class="jxr_keyword">try</strong> {
+<a class="jxr_linenumber" name="72" href="#72">72</a>              batch.set(1, <strong class="jxr_keyword">new</strong> PersistEvent());
+<a class="jxr_linenumber" name="73" href="#73">73</a>              fail();
+<a class="jxr_linenumber" name="74" href="#74">74</a>          } <strong class="jxr_keyword">catch</strong> (IllegalStateException ex) {
+<a class="jxr_linenumber" name="75" href="#75">75</a>              <em class="jxr_comment">// Expected, as we can not access elements in the batch greater than the current number of events</em>
+<a class="jxr_linenumber" name="76" href="#76">76</a>          }
+<a class="jxr_linenumber" name="77" href="#77">77</a>  
+<a class="jxr_linenumber" name="78" href="#78">78</a>          <em class="jxr_comment">// Test when filling the batch with different types of events, that becomes full</em>
+<a class="jxr_linenumber" name="79" href="#79">79</a>          <strong class="jxr_keyword">for</strong> (<strong class="jxr_keyword">int</strong> i = 0; i &lt; BATCH_SIZE; i++) {
+<a class="jxr_linenumber" name="80" href="#80">80</a>              <strong class="jxr_keyword">if</strong> (i % 4 == 0) {
+<a class="jxr_linenumber" name="81" href="#81">81</a>                  batch.addTimestamp(ANY_ST, channel, monCtx);
+<a class="jxr_linenumber" name="82" href="#82">82</a>              } <strong class="jxr_keyword">else</strong> <strong class="jxr_keyword">if</strong> (i % 4 == 1) {
+<a class="jxr_linenumber" name="83" href="#83">83</a>                  batch.addCommit(ANY_ST, ANY_CT, channel, monCtx);
+<a class="jxr_linenumber" name="84" href="#84">84</a>              } <strong class="jxr_keyword">else</strong> <strong class="jxr_keyword">if</strong> (i % 4 == 2) {
+<a class="jxr_linenumber" name="85" href="#85">85</a>                  batch.addCommitRetry(ANY_ST, channel, monCtx);
+<a class="jxr_linenumber" name="86" href="#86">86</a>              } <strong class="jxr_keyword">else</strong> {
+<a class="jxr_linenumber" name="87" href="#87">87</a>                  batch.addAbort(ANY_ST, channel, monCtx);
+<a class="jxr_linenumber" name="88" href="#88">88</a>              }
+<a class="jxr_linenumber" name="89" href="#89">89</a>          }
+<a class="jxr_linenumber" name="90" href="#90">90</a>          assertFalse(batch.isEmpty(), <span class="jxr_string">"Batch should contain elements"</span>);
+<a class="jxr_linenumber" name="91" href="#91">91</a>          assertTrue(batch.isFull(), <span class="jxr_string">"Batch should be full"</span>);
+<a class="jxr_linenumber" name="92" href="#92">92</a>          assertEquals(batch.getNumEvents(), BATCH_SIZE, <span class="jxr_string">"Num events should be "</span> + BATCH_SIZE);
+<a class="jxr_linenumber" name="93" href="#93">93</a>  
+<a class="jxr_linenumber" name="94" href="#94">94</a>          <em class="jxr_comment">// Test an exception is thrown when batch is full and a new element is going to be added</em>
+<a class="jxr_linenumber" name="95" href="#95">95</a>          <strong class="jxr_keyword">try</strong> {
+<a class="jxr_linenumber" name="96" href="#96">96</a>              batch.addCommit(ANY_ST, ANY_CT, channel, monCtx);
+<a class="jxr_linenumber" name="97" href="#97">97</a>              fail(<span class="jxr_string">"Should throw an IllegalStateException"</span>);
+<a class="jxr_linenumber" name="98" href="#98">98</a>          } <strong class="jxr_keyword">catch</strong> (IllegalStateException e) {
+<a class="jxr_linenumber" name="99" href="#99">99</a>              assertEquals(e.getMessage(), <span class="jxr_string">"batch is full"</span>, <span class="jxr_string">"message returned doesn't match"</span>);
+<a class="jxr_linenumber" name="100" href="#100">100</a>             LOG.debug(<span class="jxr_string">"IllegalStateException catch properly"</span>);
+<a class="jxr_linenumber" name="101" href="#101">101</a>         }
+<a class="jxr_linenumber" name="102" href="#102">102</a>         assertTrue(batch.isFull(), <span class="jxr_string">"Batch shouldn't be empty"</span>);
+<a class="jxr_linenumber" name="103" href="#103">103</a> 
+<a class="jxr_linenumber" name="104" href="#104">104</a>         <em class="jxr_comment">// Check the first 3 events and the last one correspond to the filling done above</em>
+<a class="jxr_linenumber" name="105" href="#105">105</a>         assertTrue(batch.get(0).getType().equals(PersistEvent.Type.TIMESTAMP));
+<a class="jxr_linenumber" name="106" href="#106">106</a>         assertTrue(batch.get(1).getType().equals(PersistEvent.Type.COMMIT));
+<a class="jxr_linenumber" name="107" href="#107">107</a>         assertTrue(batch.get(2).getType().equals(PersistEvent.Type.COMMIT_RETRY));
+<a class="jxr_linenumber" name="108" href="#108">108</a>         assertTrue(batch.get(3).getType().equals(PersistEvent.Type.ABORT));
+<a class="jxr_linenumber" name="109" href="#109">109</a> 
+<a class="jxr_linenumber" name="110" href="#110">110</a>         <em class="jxr_comment">// Set a new value for last element in Batch and check we obtain the right result</em>
+<a class="jxr_linenumber" name="111" href="#111">111</a>         batch.decreaseNumEvents();
+<a class="jxr_linenumber" name="112" href="#112">112</a>         assertEquals(batch.getNumEvents(), BATCH_SIZE - 1, <span class="jxr_string">"Num events should be "</span> + (BATCH_SIZE - 1));
+<a class="jxr_linenumber" name="113" href="#113">113</a>         <strong class="jxr_keyword">try</strong> {
+<a class="jxr_linenumber" name="114" href="#114">114</a>             batch.get(BATCH_SIZE - 1);
+<a class="jxr_linenumber" name="115" href="#115">115</a>             fail();
+<a class="jxr_linenumber" name="116" href="#116">116</a>         } <strong class="jxr_keyword">catch</strong> (IllegalStateException ex) {
+<a class="jxr_linenumber" name="117" href="#117">117</a>             <em class="jxr_comment">// Expected, as we can not access elements in the batch greater than the current number of events</em>
+<a class="jxr_linenumber" name="118" href="#118">118</a>         }
+<a class="jxr_linenumber" name="119" href="#119">119</a> 
+<a class="jxr_linenumber" name="120" href="#120">120</a>         <em class="jxr_comment">// Re-check that batch is NOT full</em>
+<a class="jxr_linenumber" name="121" href="#121">121</a>         assertFalse(batch.isFull(), <span class="jxr_string">"Batch shouldn't be full"</span>);
+<a class="jxr_linenumber" name="122" href="#122">122</a> 
+<a class="jxr_linenumber" name="123" href="#123">123</a>         <em class="jxr_comment">// Clear the batch and goes back to its initial state</em>
+<a class="jxr_linenumber" name="124" href="#124">124</a>         batch.clear();
+<a class="jxr_linenumber" name="125" href="#125">125</a>         assertTrue(batch.isEmpty(), <span class="jxr_string">"Batch should be empty"</span>);
+<a class="jxr_linenumber" name="126" href="#126">126</a>         assertFalse(batch.isFull(), <span class="jxr_string">"Batch shouldn't be full"</span>);
+<a class="jxr_linenumber" name="127" href="#127">127</a>         assertEquals(batch.getNumEvents(), 0, <span class="jxr_string">"Num events should be 0"</span>);
+<a class="jxr_linenumber" name="128" href="#128">128</a> 
+<a class="jxr_linenumber" name="129" href="#129">129</a>     }
+<a class="jxr_linenumber" name="130" href="#130">130</a> 
+<a class="jxr_linenumber" name="131" href="#131">131</a>     @Test(timeOut = 10_000)
+<a class="jxr_linenumber" name="132" href="#132">132</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testBatchFactoryFunctionality() <strong class="jxr_keyword">throws</strong> Exception {
+<a class="jxr_linenumber" name="133" href="#133">133</a> 
+<a class="jxr_linenumber" name="134" href="#134">134</a>         <em class="jxr_comment">// Component to test</em>
+<a class="jxr_linenumber" name="135" href="#135">135</a>         Batch.BatchFactory factory = <strong class="jxr_keyword">new</strong> Batch.BatchFactory(BATCH_SIZE);
+<a class="jxr_linenumber" name="136" href="#136">136</a> 
+<a class="jxr_linenumber" name="137" href="#137">137</a>         <em class="jxr_comment">// Check the factory creates a new batch properly...</em>
+<a class="jxr_linenumber" name="138" href="#138">138</a>         Batch batch = factory.create();
+<a class="jxr_linenumber" name="139" href="#139">139</a>         assertTrue(batch.isEmpty(), <span class="jxr_string">"Batch should be empty"</span>);
+<a class="jxr_linenumber" name="140" href="#140">140</a>         assertFalse(batch.isFull(), <span class="jxr_string">"Batch shouldn't be full"</span>);
+<a class="jxr_linenumber" name="141" href="#141">141</a>         assertEquals(batch.getNumEvents(), 0, <span class="jxr_string">"Num events should be 0"</span>);
 <a class="jxr_linenumber" name="142" href="#142">142</a> 
-<a class="jxr_linenumber" name="143" href="#143">143</a>         <em class="jxr_comment">// Test that sending replies empties the batch also when the replica</em>
-<a class="jxr_linenumber" name="144" href="#144">144</a>         <em class="jxr_comment">// is NOT master and calls the ambiguousCommitResponse() method on the</em>
-<a class="jxr_linenumber" name="145" href="#145">145</a>         <em class="jxr_comment">// reply processor</em>
-<a class="jxr_linenumber" name="146" href="#146">146</a>         <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">boolean</strong> MASTER_INSTANCE = <strong class="jxr_keyword">true</strong>;
-<a class="jxr_linenumber" name="147" href="#147">147</a>         <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">boolean</strong> SHOULD_MAKE_HEURISTIC_DECISSION = <strong class="jxr_keyword">true</strong>;
-<a class="jxr_linenumber" name="148" href="#148">148</a>         batch.sendRepliesAndReset(replyProcessor, retryProcessor, !MASTER_INSTANCE);
-<a class="jxr_linenumber" name="149" href="#149">149</a>         verify(replyProcessor, timeout(100).times(BATCH_SIZE / 2))
-<a class="jxr_linenumber" name="150" href="#150">150</a>                 .timestampResponse(anyLong(), any(Channel.<strong class="jxr_keyword">class</strong>), any(MonitoringContext.<strong class="jxr_keyword">class</strong>));
-<a class="jxr_linenumber" name="151" href="#151">151</a>         verify(replyProcessor, timeout(100).times(BATCH_SIZE / 2))
-<a class="jxr_linenumber" name="152" href="#152">152</a>                 .commitResponse(eq(SHOULD_MAKE_HEURISTIC_DECISSION), anyLong(), anyLong(), any(Channel.<strong class="jxr_keyword">class</strong>), any(
-<a class="jxr_linenumber" name="153" href="#153">153</a>                         MonitoringContext.<strong class="jxr_keyword">class</strong>));
-<a class="jxr_linenumber" name="154" href="#154">154</a>         assertFalse(batch.isFull(), <span class="jxr_string">"Batch shouldn't be full"</span>);
-<a class="jxr_linenumber" name="155" href="#155">155</a>         assertEquals(batch.getNumEvents(), 0, <span class="jxr_string">"Num events should be 0"</span>);
-<a class="jxr_linenumber" name="156" href="#156">156</a> 
-<a class="jxr_linenumber" name="157" href="#157">157</a>     }
-<a class="jxr_linenumber" name="158" href="#158">158</a> }
+<a class="jxr_linenumber" name="143" href="#143">143</a>         <em class="jxr_comment">// ...and is wrapped in to a pooled object</em>
+<a class="jxr_linenumber" name="144" href="#144">144</a>         PooledObject&lt;Batch&gt; pooledBatch = factory.wrap(batch);
+<a class="jxr_linenumber" name="145" href="#145">145</a>         assertEquals(pooledBatch.getObject(), batch);
+<a class="jxr_linenumber" name="146" href="#146">146</a> 
+<a class="jxr_linenumber" name="147" href="#147">147</a>         <em class="jxr_comment">// Put some elements in the batch...</em>
+<a class="jxr_linenumber" name="148" href="#148">148</a>         batch.addTimestamp(ANY_ST, channel, monCtx);
+<a class="jxr_linenumber" name="149" href="#149">149</a>         batch.addCommit(ANY_ST, ANY_CT, channel, monCtx);
+<a class="jxr_linenumber" name="150" href="#150">150</a>         batch.addCommitRetry(ANY_ST, channel, monCtx);
+<a class="jxr_linenumber" name="151" href="#151">151</a>         batch.addAbort(ANY_ST, channel, monCtx);
+<a class="jxr_linenumber" name="152" href="#152">152</a>         assertFalse(batch.isEmpty(), <span class="jxr_string">"Batch should contain elements"</span>);
+<a class="jxr_linenumber" name="153" href="#153">153</a>         assertFalse(batch.isFull(), <span class="jxr_string">"Batch should NOT be full"</span>);
+<a class="jxr_linenumber" name="154" href="#154">154</a>         assertEquals(batch.getNumEvents(), 4, <span class="jxr_string">"Num events should be 4"</span>);
+<a class="jxr_linenumber" name="155" href="#155">155</a> 
+<a class="jxr_linenumber" name="156" href="#156">156</a>         <em class="jxr_comment">// ... and passivate the object through the factory. It should reset the state of the batch</em>
+<a class="jxr_linenumber" name="157" href="#157">157</a>         factory.passivateObject(pooledBatch);
+<a class="jxr_linenumber" name="158" href="#158">158</a>         assertTrue(batch.isEmpty(), <span class="jxr_string">"Batch should NOT contain elements"</span>);
+<a class="jxr_linenumber" name="159" href="#159">159</a>         assertFalse(batch.isFull(), <span class="jxr_string">"Batch should NOT be full"</span>);
+<a class="jxr_linenumber" name="160" href="#160">160</a>         assertEquals(batch.getNumEvents(), 0, <span class="jxr_string">"Num events should be 0"</span>);
+<a class="jxr_linenumber" name="161" href="#161">161</a> 
+<a class="jxr_linenumber" name="162" href="#162">162</a>     }
+<a class="jxr_linenumber" name="163" href="#163">163</a> 
+<a class="jxr_linenumber" name="164" href="#164">164</a> }
 </pre>
 <hr/><div id="footer">This page was automatically generated by <a href="http://maven.apache.org/">Maven</a></div></body>
 </html>

http://git-wip-us.apache.org/repos/asf/incubator-omid-site/blob/3a40a657/xref-test/org/apache/omid/tso/TestLeaseManager.html
----------------------------------------------------------------------
diff --git a/xref-test/org/apache/omid/tso/TestLeaseManager.html b/xref-test/org/apache/omid/tso/TestLeaseManager.html
index 4dcddbd..abe4747 100644
--- a/xref-test/org/apache/omid/tso/TestLeaseManager.html
+++ b/xref-test/org/apache/omid/tso/TestLeaseManager.html
@@ -8,31 +8,31 @@
 <body>
 <div id="overview"><a href="../../../../../../target/site/testapidocs/org/apache/omid/tso/TestLeaseManager.html">View Javadoc</a></div><pre>
 
-<a class="jxr_linenumber" name="1" href="#1">1</a>   <em class="jxr_javadoccomment">/**</em>
-<a class="jxr_linenumber" name="2" href="#2">2</a>   <em class="jxr_javadoccomment"> * Licensed to the Apache Software Foundation (ASF) under one</em>
-<a class="jxr_linenumber" name="3" href="#3">3</a>   <em class="jxr_javadoccomment"> * or more contributor license agreements.  See the NOTICE file</em>
-<a class="jxr_linenumber" name="4" href="#4">4</a>   <em class="jxr_javadoccomment"> * distributed with this work for additional information</em>
-<a class="jxr_linenumber" name="5" href="#5">5</a>   <em class="jxr_javadoccomment"> * regarding copyright ownership.  The ASF licenses this file</em>
-<a class="jxr_linenumber" name="6" href="#6">6</a>   <em class="jxr_javadoccomment"> * to you under the Apache License, Version 2.0 (the</em>
-<a class="jxr_linenumber" name="7" href="#7">7</a>   <em class="jxr_javadoccomment"> * "License"); you may not use this file except in compliance</em>
-<a class="jxr_linenumber" name="8" href="#8">8</a>   <em class="jxr_javadoccomment"> * with the License.  You may obtain a copy of the License at</em>
-<a class="jxr_linenumber" name="9" href="#9">9</a>   <em class="jxr_javadoccomment"> *</em>
-<a class="jxr_linenumber" name="10" href="#10">10</a>  <em class="jxr_javadoccomment"> *   <a href="http://www.apache.org/licenses/LICENSE-2.0" target="alexandria_uri">http://www.apache.org/licenses/LICENSE-2.0</a></em>
-<a class="jxr_linenumber" name="11" href="#11">11</a>  <em class="jxr_javadoccomment"> *</em>
-<a class="jxr_linenumber" name="12" href="#12">12</a>  <em class="jxr_javadoccomment"> * Unless required by applicable law or agreed to in writing, software</em>
-<a class="jxr_linenumber" name="13" href="#13">13</a>  <em class="jxr_javadoccomment"> * distributed under the License is distributed on an "AS IS" BASIS,</em>
-<a class="jxr_linenumber" name="14" href="#14">14</a>  <em class="jxr_javadoccomment"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</em>
-<a class="jxr_linenumber" name="15" href="#15">15</a>  <em class="jxr_javadoccomment"> * See the License for the specific language governing permissions and</em>
-<a class="jxr_linenumber" name="16" href="#16">16</a>  <em class="jxr_javadoccomment"> * limitations under the License.</em>
-<a class="jxr_linenumber" name="17" href="#17">17</a>  <em class="jxr_javadoccomment"> */</em>
+<a class="jxr_linenumber" name="1" href="#1">1</a>   <em class="jxr_comment">/*</em>
+<a class="jxr_linenumber" name="2" href="#2">2</a>   <em class="jxr_comment"> * Licensed to the Apache Software Foundation (ASF) under one</em>
+<a class="jxr_linenumber" name="3" href="#3">3</a>   <em class="jxr_comment"> * or more contributor license agreements.  See the NOTICE file</em>
+<a class="jxr_linenumber" name="4" href="#4">4</a>   <em class="jxr_comment"> * distributed with this work for additional information</em>
+<a class="jxr_linenumber" name="5" href="#5">5</a>   <em class="jxr_comment"> * regarding copyright ownership.  The ASF licenses this file</em>
+<a class="jxr_linenumber" name="6" href="#6">6</a>   <em class="jxr_comment"> * to you under the Apache License, Version 2.0 (the</em>
+<a class="jxr_linenumber" name="7" href="#7">7</a>   <em class="jxr_comment"> * "License"); you may not use this file except in compliance</em>
+<a class="jxr_linenumber" name="8" href="#8">8</a>   <em class="jxr_comment"> * with the License.  You may obtain a copy of the License at</em>
+<a class="jxr_linenumber" name="9" href="#9">9</a>   <em class="jxr_comment"> *</em>
+<a class="jxr_linenumber" name="10" href="#10">10</a>  <em class="jxr_comment"> *   <a href="http://www.apache.org/licenses/LICENSE-2.0" target="alexandria_uri">http://www.apache.org/licenses/LICENSE-2.0</a></em>
+<a class="jxr_linenumber" name="11" href="#11">11</a>  <em class="jxr_comment"> *</em>
+<a class="jxr_linenumber" name="12" href="#12">12</a>  <em class="jxr_comment"> * Unless required by applicable law or agreed to in writing, software</em>
+<a class="jxr_linenumber" name="13" href="#13">13</a>  <em class="jxr_comment"> * distributed under the License is distributed on an "AS IS" BASIS,</em>
+<a class="jxr_linenumber" name="14" href="#14">14</a>  <em class="jxr_comment"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</em>
+<a class="jxr_linenumber" name="15" href="#15">15</a>  <em class="jxr_comment"> * See the License for the specific language governing permissions and</em>
+<a class="jxr_linenumber" name="16" href="#16">16</a>  <em class="jxr_comment"> * limitations under the License.</em>
+<a class="jxr_linenumber" name="17" href="#17">17</a>  <em class="jxr_comment"> */</em>
 <a class="jxr_linenumber" name="18" href="#18">18</a>  <strong class="jxr_keyword">package</strong> org.apache.omid.tso;
 <a class="jxr_linenumber" name="19" href="#19">19</a>  
 <a class="jxr_linenumber" name="20" href="#20">20</a>  <strong class="jxr_keyword">import</strong> com.google.common.base.Charsets;
-<a class="jxr_linenumber" name="21" href="#21">21</a>  <strong class="jxr_keyword">import</strong> org.apache.omid.TestUtils;
-<a class="jxr_linenumber" name="22" href="#22">22</a>  <strong class="jxr_keyword">import</strong> org.apache.omid.tso.TSOStateManager.TSOState;
-<a class="jxr_linenumber" name="23" href="#23">23</a>  <strong class="jxr_keyword">import</strong> org.apache.curator.framework.CuratorFramework;
-<a class="jxr_linenumber" name="24" href="#24">24</a>  <strong class="jxr_keyword">import</strong> org.apache.curator.test.TestingServer;
-<a class="jxr_linenumber" name="25" href="#25">25</a>  <strong class="jxr_keyword">import</strong> org.apache.curator.utils.CloseableUtils;
+<a class="jxr_linenumber" name="21" href="#21">21</a>  <strong class="jxr_keyword">import</strong> org.apache.curator.framework.CuratorFramework;
+<a class="jxr_linenumber" name="22" href="#22">22</a>  <strong class="jxr_keyword">import</strong> org.apache.curator.test.TestingServer;
+<a class="jxr_linenumber" name="23" href="#23">23</a>  <strong class="jxr_keyword">import</strong> org.apache.curator.utils.CloseableUtils;
+<a class="jxr_linenumber" name="24" href="#24">24</a>  <strong class="jxr_keyword">import</strong> org.apache.omid.TestUtils;
+<a class="jxr_linenumber" name="25" href="#25">25</a>  <strong class="jxr_keyword">import</strong> org.apache.omid.tso.TSOStateManager.TSOState;
 <a class="jxr_linenumber" name="26" href="#26">26</a>  <strong class="jxr_keyword">import</strong> org.mockito.ArgumentCaptor;
 <a class="jxr_linenumber" name="27" href="#27">27</a>  <strong class="jxr_keyword">import</strong> org.mockito.Mock;
 <a class="jxr_linenumber" name="28" href="#28">28</a>  <strong class="jxr_keyword">import</strong> org.slf4j.Logger;
@@ -50,94 +50,94 @@
 <a class="jxr_linenumber" name="40" href="#40">40</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.reset;
 <a class="jxr_linenumber" name="41" href="#41">41</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.spy;
 <a class="jxr_linenumber" name="42" href="#42">42</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.timeout;
-<a class="jxr_linenumber" name="43" href="#43">43</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.verify;
-<a class="jxr_linenumber" name="44" href="#44">44</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.when;
-<a class="jxr_linenumber" name="45" href="#45">45</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertEquals;
-<a class="jxr_linenumber" name="46" href="#46">46</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.AssertJUnit.assertFalse;
-<a class="jxr_linenumber" name="47" href="#47">47</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.AssertJUnit.assertTrue;
-<a class="jxr_linenumber" name="48" href="#48">48</a>  
-<a class="jxr_linenumber" name="49" href="#49">49</a>  <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">class</strong> <a href="../../../../org/apache/omid/tso/TestLeaseManager.html">TestLeaseManager</a> {
-<a class="jxr_linenumber" name="50" href="#50">50</a>  
-<a class="jxr_linenumber" name="51" href="#51">51</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_EPOCH_1 = 1L;
-<a class="jxr_linenumber" name="52" href="#52">52</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_EPOCH_2 = 2L;
-<a class="jxr_linenumber" name="53" href="#53">53</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_EPOCH_3 = 3L;
-<a class="jxr_linenumber" name="54" href="#54">54</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_LOW_WATERMARK_1 = DUMMY_EPOCH_1;
-<a class="jxr_linenumber" name="55" href="#55">55</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_LOW_WATERMARK_2 = DUMMY_EPOCH_2;
-<a class="jxr_linenumber" name="56" href="#56">56</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_LOW_WATERMARK_3 = DUMMY_EPOCH_3;
-<a class="jxr_linenumber" name="57" href="#57">57</a>  
-<a class="jxr_linenumber" name="58" href="#58">58</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> String LEASE_MGR_ID_1 = <span class="jxr_string">"LM1"</span>;
-<a class="jxr_linenumber" name="59" href="#59">59</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> String LEASE_MGR_ID_2 = <span class="jxr_string">"LM2"</span>;
-<a class="jxr_linenumber" name="60" href="#60">60</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> String INSTANCE_ID_1 = <span class="jxr_string">"LM1"</span> + <span class="jxr_string">"#"</span>;
-<a class="jxr_linenumber" name="61" href="#61">61</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> String INSTANCE_ID_2 = <span class="jxr_string">"LM2"</span> + <span class="jxr_string">"#"</span>;
-<a class="jxr_linenumber" name="62" href="#62">62</a>  
-<a class="jxr_linenumber" name="63" href="#63">63</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> Logger LOG = LoggerFactory.getLogger(TestLeaseManager.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="64" href="#64">64</a>  
-<a class="jxr_linenumber" name="65" href="#65">65</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> TEST_LEASE_PERIOD_IN_MS = 1000; <em class="jxr_comment">// 1 second</em>
-<a class="jxr_linenumber" name="66" href="#66">66</a>  
-<a class="jxr_linenumber" name="67" href="#67">67</a>      <strong class="jxr_keyword">private</strong> CuratorFramework zkClient;
-<a class="jxr_linenumber" name="68" href="#68">68</a>      <strong class="jxr_keyword">private</strong> TestingServer zkServer;
-<a class="jxr_linenumber" name="69" href="#69">69</a>  
-<a class="jxr_linenumber" name="70" href="#70">70</a>      @Mock
-<a class="jxr_linenumber" name="71" href="#71">71</a>      <strong class="jxr_keyword">private</strong> Panicker panicker;
-<a class="jxr_linenumber" name="72" href="#72">72</a>  
-<a class="jxr_linenumber" name="73" href="#73">73</a>      <strong class="jxr_keyword">private</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a> leaseManager1;
-<a class="jxr_linenumber" name="74" href="#74">74</a>      <strong class="jxr_keyword">private</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a> leaseManager2;
-<a class="jxr_linenumber" name="75" href="#75">75</a>  
-<a class="jxr_linenumber" name="76" href="#76">76</a>      @BeforeClass
-<a class="jxr_linenumber" name="77" href="#77">77</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> beforeClass() <strong class="jxr_keyword">throws</strong> Exception {
-<a class="jxr_linenumber" name="78" href="#78">78</a>  
-<a class="jxr_linenumber" name="79" href="#79">79</a>          LOG.info(<span class="jxr_string">"Starting ZK Server"</span>);
-<a class="jxr_linenumber" name="80" href="#80">80</a>          zkServer = TestUtils.provideTestingZKServer();
-<a class="jxr_linenumber" name="81" href="#81">81</a>          LOG.info(<span class="jxr_string">"ZK Server Started @ {}"</span>, zkServer.getConnectString());
-<a class="jxr_linenumber" name="82" href="#82">82</a>  
-<a class="jxr_linenumber" name="83" href="#83">83</a>          zkClient = TestUtils.provideConnectedZKClient(DEFAULT_ZK_CLUSTER);
-<a class="jxr_linenumber" name="84" href="#84">84</a>  
-<a class="jxr_linenumber" name="85" href="#85">85</a>      }
-<a class="jxr_linenumber" name="86" href="#86">86</a>  
-<a class="jxr_linenumber" name="87" href="#87">87</a>      @AfterClass
-<a class="jxr_linenumber" name="88" href="#88">88</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> afterClass() <strong class="jxr_keyword">throws</strong> Exception {
-<a class="jxr_linenumber" name="89" href="#89">89</a>  
-<a class="jxr_linenumber" name="90" href="#90">90</a>          zkClient.close();
-<a class="jxr_linenumber" name="91" href="#91">91</a>  
-<a class="jxr_linenumber" name="92" href="#92">92</a>          CloseableUtils.closeQuietly(zkServer);
-<a class="jxr_linenumber" name="93" href="#93">93</a>          zkServer = <strong class="jxr_keyword">null</strong>;
-<a class="jxr_linenumber" name="94" href="#94">94</a>          LOG.info(<span class="jxr_string">"ZK Server Stopped"</span>);
-<a class="jxr_linenumber" name="95" href="#95">95</a>  
-<a class="jxr_linenumber" name="96" href="#96">96</a>      }
-<a class="jxr_linenumber" name="97" href="#97">97</a>  
-<a class="jxr_linenumber" name="98" href="#98">98</a>      @Test(timeOut = 30_000)
-<a class="jxr_linenumber" name="99" href="#99">99</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testErrorInitializingTSOStateExitsTheTSO() <strong class="jxr_keyword">throws</strong> Exception {
-<a class="jxr_linenumber" name="100" href="#100">100</a> 
-<a class="jxr_linenumber" name="101" href="#101">101</a>         <strong class="jxr_keyword">final</strong> String TEST_TSO_LEASE_PATH = <span class="jxr_string">"/test0_tsolease"</span>;
-<a class="jxr_linenumber" name="102" href="#102">102</a>         <strong class="jxr_keyword">final</strong> String TEST_CURRENT_TSO_PATH = <span class="jxr_string">"/test0_currenttso"</span>;
-<a class="jxr_linenumber" name="103" href="#103">103</a> 
-<a class="jxr_linenumber" name="104" href="#104">104</a>         Panicker panicker = spy(<strong class="jxr_keyword">new</strong> MockPanicker());
-<a class="jxr_linenumber" name="105" href="#105">105</a> 
-<a class="jxr_linenumber" name="106" href="#106">106</a>         TSOChannelHandler tsoChannelHandler = mock(TSOChannelHandler.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="107" href="#107">107</a>         TSOStateManager stateManager = mock(TSOStateManager.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="108" href="#108">108</a>         when(stateManager.reset()).thenThrow(<strong class="jxr_keyword">new</strong> IOException());
-<a class="jxr_linenumber" name="109" href="#109">109</a>         leaseManager1 = <strong class="jxr_keyword">new</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a>(LEASE_MGR_ID_1,
-<a class="jxr_linenumber" name="110" href="#110">110</a>                                                  tsoChannelHandler,
-<a class="jxr_linenumber" name="111" href="#111">111</a>                                                  stateManager,
-<a class="jxr_linenumber" name="112" href="#112">112</a>                                                  TEST_LEASE_PERIOD_IN_MS,
-<a class="jxr_linenumber" name="113" href="#113">113</a>                                                  TEST_TSO_LEASE_PATH,
-<a class="jxr_linenumber" name="114" href="#114">114</a>                                                  TEST_CURRENT_TSO_PATH,
-<a class="jxr_linenumber" name="115" href="#115">115</a>                                                  zkClient,
-<a class="jxr_linenumber" name="116" href="#116">116</a>                                                  panicker);
-<a class="jxr_linenumber" name="117" href="#117">117</a>         leaseManager1.startService();
-<a class="jxr_linenumber" name="118" href="#118">118</a> 
-<a class="jxr_linenumber" name="119" href="#119">119</a>         <em class="jxr_comment">// ... let the test run for some time...</em>
-<a class="jxr_linenumber" name="120" href="#120">120</a>         Thread.sleep(TEST_LEASE_PERIOD_IN_MS * 2);
-<a class="jxr_linenumber" name="121" href="#121">121</a> 
-<a class="jxr_linenumber" name="122" href="#122">122</a>         verify(panicker, timeout(2000).atLeastOnce()).panic(anyString(), any(IOException.<strong class="jxr_keyword">class</strong>));
-<a class="jxr_linenumber" name="123" href="#123">123</a> 
-<a class="jxr_linenumber" name="124" href="#124">124</a>         leaseManager1.stopService();
-<a class="jxr_linenumber" name="125" href="#125">125</a> 
-<a class="jxr_linenumber" name="126" href="#126">126</a>     }
-<a class="jxr_linenumber" name="127" href="#127">127</a> 
-<a class="jxr_linenumber" name="128" href="#128">128</a>     @Test(timeOut = 60000)
-<a class="jxr_linenumber" name="129" href="#129">129</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testLeaseHolderDoesNotChangeWhenPausedForALongTimeAndTheresNoOtherInstance()
-<a class="jxr_linenumber" name="130" href="#130">130</a>             <strong class="jxr_keyword">throws</strong> Exception {
+<a class="jxr_linenumber" name="43" href="#43">43</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.times;
+<a class="jxr_linenumber" name="44" href="#44">44</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.verify;
+<a class="jxr_linenumber" name="45" href="#45">45</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.mockito.Mockito.when;
+<a class="jxr_linenumber" name="46" href="#46">46</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertEquals;
+<a class="jxr_linenumber" name="47" href="#47">47</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertFalse;
+<a class="jxr_linenumber" name="48" href="#48">48</a>  <strong class="jxr_keyword">import</strong> <strong class="jxr_keyword">static</strong> org.testng.Assert.assertTrue;
+<a class="jxr_linenumber" name="49" href="#49">49</a>  
+<a class="jxr_linenumber" name="50" href="#50">50</a>  <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">class</strong> <a href="../../../../org/apache/omid/tso/TestLeaseManager.html">TestLeaseManager</a> {
+<a class="jxr_linenumber" name="51" href="#51">51</a>  
+<a class="jxr_linenumber" name="52" href="#52">52</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_EPOCH_1 = 1L;
+<a class="jxr_linenumber" name="53" href="#53">53</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_EPOCH_2 = 2L;
+<a class="jxr_linenumber" name="54" href="#54">54</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_EPOCH_3 = 3L;
+<a class="jxr_linenumber" name="55" href="#55">55</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_LOW_WATERMARK_1 = DUMMY_EPOCH_1;
+<a class="jxr_linenumber" name="56" href="#56">56</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_LOW_WATERMARK_2 = DUMMY_EPOCH_2;
+<a class="jxr_linenumber" name="57" href="#57">57</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> DUMMY_LOW_WATERMARK_3 = DUMMY_EPOCH_3;
+<a class="jxr_linenumber" name="58" href="#58">58</a>  
+<a class="jxr_linenumber" name="59" href="#59">59</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> String LEASE_MGR_ID_1 = <span class="jxr_string">"LM1"</span>;
+<a class="jxr_linenumber" name="60" href="#60">60</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> String LEASE_MGR_ID_2 = <span class="jxr_string">"LM2"</span>;
+<a class="jxr_linenumber" name="61" href="#61">61</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> String INSTANCE_ID_1 = <span class="jxr_string">"LM1"</span> + <span class="jxr_string">"#"</span>;
+<a class="jxr_linenumber" name="62" href="#62">62</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> String INSTANCE_ID_2 = <span class="jxr_string">"LM2"</span> + <span class="jxr_string">"#"</span>;
+<a class="jxr_linenumber" name="63" href="#63">63</a>  
+<a class="jxr_linenumber" name="64" href="#64">64</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> Logger LOG = LoggerFactory.getLogger(TestLeaseManager.<strong class="jxr_keyword">class</strong>);
+<a class="jxr_linenumber" name="65" href="#65">65</a>  
+<a class="jxr_linenumber" name="66" href="#66">66</a>      <strong class="jxr_keyword">private</strong> <strong class="jxr_keyword">static</strong> <strong class="jxr_keyword">final</strong> <strong class="jxr_keyword">long</strong> TEST_LEASE_PERIOD_IN_MS = 5000; <em class="jxr_comment">// 5 seconds</em>
+<a class="jxr_linenumber" name="67" href="#67">67</a>  
+<a class="jxr_linenumber" name="68" href="#68">68</a>      <strong class="jxr_keyword">private</strong> CuratorFramework zkClient;
+<a class="jxr_linenumber" name="69" href="#69">69</a>      <strong class="jxr_keyword">private</strong> TestingServer zkServer;
+<a class="jxr_linenumber" name="70" href="#70">70</a>  
+<a class="jxr_linenumber" name="71" href="#71">71</a>      @Mock
+<a class="jxr_linenumber" name="72" href="#72">72</a>      <strong class="jxr_keyword">private</strong> Panicker panicker;
+<a class="jxr_linenumber" name="73" href="#73">73</a>  
+<a class="jxr_linenumber" name="74" href="#74">74</a>      <strong class="jxr_keyword">private</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a> leaseManager1;
+<a class="jxr_linenumber" name="75" href="#75">75</a>      <strong class="jxr_keyword">private</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a> leaseManager2;
+<a class="jxr_linenumber" name="76" href="#76">76</a>  
+<a class="jxr_linenumber" name="77" href="#77">77</a>      @BeforeClass
+<a class="jxr_linenumber" name="78" href="#78">78</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> beforeClass() <strong class="jxr_keyword">throws</strong> Exception {
+<a class="jxr_linenumber" name="79" href="#79">79</a>  
+<a class="jxr_linenumber" name="80" href="#80">80</a>          LOG.info(<span class="jxr_string">"Starting ZK Server"</span>);
+<a class="jxr_linenumber" name="81" href="#81">81</a>          zkServer = TestUtils.provideTestingZKServer();
+<a class="jxr_linenumber" name="82" href="#82">82</a>          LOG.info(<span class="jxr_string">"ZK Server Started @ {}"</span>, zkServer.getConnectString());
+<a class="jxr_linenumber" name="83" href="#83">83</a>  
+<a class="jxr_linenumber" name="84" href="#84">84</a>          zkClient = TestUtils.provideConnectedZKClient(DEFAULT_ZK_CLUSTER);
+<a class="jxr_linenumber" name="85" href="#85">85</a>  
+<a class="jxr_linenumber" name="86" href="#86">86</a>      }
+<a class="jxr_linenumber" name="87" href="#87">87</a>  
+<a class="jxr_linenumber" name="88" href="#88">88</a>      @AfterClass
+<a class="jxr_linenumber" name="89" href="#89">89</a>      <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> afterClass() <strong class="jxr_keyword">throws</strong> Exception {
+<a class="jxr_linenumber" name="90" href="#90">90</a>  
+<a class="jxr_linenumber" name="91" href="#91">91</a>          zkClient.close();
+<a class="jxr_linenumber" name="92" href="#92">92</a>  
+<a class="jxr_linenumber" name="93" href="#93">93</a>          CloseableUtils.closeQuietly(zkServer);
+<a class="jxr_linenumber" name="94" href="#94">94</a>          zkServer = <strong class="jxr_keyword">null</strong>;
+<a class="jxr_linenumber" name="95" href="#95">95</a>          LOG.info(<span class="jxr_string">"ZK Server Stopped"</span>);
+<a class="jxr_linenumber" name="96" href="#96">96</a>  
+<a class="jxr_linenumber" name="97" href="#97">97</a>      }
+<a class="jxr_linenumber" name="98" href="#98">98</a>  
+<a class="jxr_linenumber" name="99" href="#99">99</a>      @Test(timeOut = 80_000)
+<a class="jxr_linenumber" name="100" href="#100">100</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testErrorInitializingTSOStateExitsTheTSO() <strong class="jxr_keyword">throws</strong> Exception {
+<a class="jxr_linenumber" name="101" href="#101">101</a> 
+<a class="jxr_linenumber" name="102" href="#102">102</a>         <strong class="jxr_keyword">final</strong> String TEST_TSO_LEASE_PATH = <span class="jxr_string">"/test0_tsolease"</span>;
+<a class="jxr_linenumber" name="103" href="#103">103</a>         <strong class="jxr_keyword">final</strong> String TEST_CURRENT_TSO_PATH = <span class="jxr_string">"/test0_currenttso"</span>;
+<a class="jxr_linenumber" name="104" href="#104">104</a> 
+<a class="jxr_linenumber" name="105" href="#105">105</a>         Panicker panicker = spy(<strong class="jxr_keyword">new</strong> MockPanicker());
+<a class="jxr_linenumber" name="106" href="#106">106</a> 
+<a class="jxr_linenumber" name="107" href="#107">107</a>         TSOChannelHandler tsoChannelHandler = mock(TSOChannelHandler.<strong class="jxr_keyword">class</strong>);
+<a class="jxr_linenumber" name="108" href="#108">108</a>         TSOStateManager stateManager = mock(TSOStateManager.<strong class="jxr_keyword">class</strong>);
+<a class="jxr_linenumber" name="109" href="#109">109</a>         when(stateManager.initialize()).thenThrow(<strong class="jxr_keyword">new</strong> IOException());
+<a class="jxr_linenumber" name="110" href="#110">110</a>         leaseManager1 = <strong class="jxr_keyword">new</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a>(LEASE_MGR_ID_1,
+<a class="jxr_linenumber" name="111" href="#111">111</a>                                                  tsoChannelHandler,
+<a class="jxr_linenumber" name="112" href="#112">112</a>                                                  stateManager,
+<a class="jxr_linenumber" name="113" href="#113">113</a>                                                  TEST_LEASE_PERIOD_IN_MS,
+<a class="jxr_linenumber" name="114" href="#114">114</a>                                                  TEST_TSO_LEASE_PATH,
+<a class="jxr_linenumber" name="115" href="#115">115</a>                                                  TEST_CURRENT_TSO_PATH,
+<a class="jxr_linenumber" name="116" href="#116">116</a>                                                  zkClient,
+<a class="jxr_linenumber" name="117" href="#117">117</a>                                                  panicker);
+<a class="jxr_linenumber" name="118" href="#118">118</a>         leaseManager1.startService();
+<a class="jxr_linenumber" name="119" href="#119">119</a> 
+<a class="jxr_linenumber" name="120" href="#120">120</a>         <em class="jxr_comment">// ... let the test run for some time...</em>
+<a class="jxr_linenumber" name="121" href="#121">121</a>         Thread.sleep(TEST_LEASE_PERIOD_IN_MS * 2);
+<a class="jxr_linenumber" name="122" href="#122">122</a> 
+<a class="jxr_linenumber" name="123" href="#123">123</a>         verify(panicker, timeout(2000).atLeastOnce()).panic(anyString(), any(IOException.<strong class="jxr_keyword">class</strong>));
+<a class="jxr_linenumber" name="124" href="#124">124</a> 
+<a class="jxr_linenumber" name="125" href="#125">125</a>         leaseManager1.stopService();
+<a class="jxr_linenumber" name="126" href="#126">126</a> 
+<a class="jxr_linenumber" name="127" href="#127">127</a>     }
+<a class="jxr_linenumber" name="128" href="#128">128</a> 
+<a class="jxr_linenumber" name="129" href="#129">129</a>     @Test(timeOut = 80_000)
+<a class="jxr_linenumber" name="130" href="#130">130</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testLeaseHolderDoesNotChangeWhenPausedForALongTimeAndTheresNoOtherInstance() <strong class="jxr_keyword">throws</strong> Exception {
 <a class="jxr_linenumber" name="131" href="#131">131</a> 
 <a class="jxr_linenumber" name="132" href="#132">132</a>         <strong class="jxr_keyword">final</strong> String TEST_TSO_LEASE_PATH = <span class="jxr_string">"/test1_tsolease"</span>;
 <a class="jxr_linenumber" name="133" href="#133">133</a>         <strong class="jxr_keyword">final</strong> String TEST_CURRENT_TSO_PATH = <span class="jxr_string">"/test1_currenttso"</span>;
@@ -145,7 +145,7 @@
 <a class="jxr_linenumber" name="135" href="#135">135</a>         <em class="jxr_comment">// Launch the instance under test...</em>
 <a class="jxr_linenumber" name="136" href="#136">136</a>         TSOChannelHandler tsoChannelHandler1 = mock(TSOChannelHandler.<strong class="jxr_keyword">class</strong>);
 <a class="jxr_linenumber" name="137" href="#137">137</a>         TSOStateManager stateManager1 = mock(TSOStateManager.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="138" href="#138">138</a>         when(stateManager1.reset()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_1, DUMMY_EPOCH_1));
+<a class="jxr_linenumber" name="138" href="#138">138</a>         when(stateManager1.initialize()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_1, DUMMY_EPOCH_1));
 <a class="jxr_linenumber" name="139" href="#139">139</a>         leaseManager1 = <strong class="jxr_keyword">new</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a>(LEASE_MGR_ID_1,
 <a class="jxr_linenumber" name="140" href="#140">140</a>                                                  tsoChannelHandler1,
 <a class="jxr_linenumber" name="141" href="#141">141</a>                                                  stateManager1,
@@ -183,11 +183,11 @@
 <a class="jxr_linenumber" name="173" href="#173">173</a>         <em class="jxr_comment">// ... and check again that nothing changed</em>
 <a class="jxr_linenumber" name="174" href="#174">174</a>         checkLeaseHolder(TEST_TSO_LEASE_PATH, LEASE_MGR_ID_1);
 <a class="jxr_linenumber" name="175" href="#175">175</a>         checkInstanceId(TEST_CURRENT_TSO_PATH, INSTANCE_ID_1 + <span class="jxr_string">"1"</span>);
-<a class="jxr_linenumber" name="176" href="#176">176</a>         assertTrue(leaseManager1.stillInLeasePeriod());
+<a class="jxr_linenumber" name="176" href="#176">176</a>         assertFalse(leaseManager1.stillInLeasePeriod()); <em class="jxr_comment">// Must not be master as it should have triggered and exception</em>
 <a class="jxr_linenumber" name="177" href="#177">177</a> 
 <a class="jxr_linenumber" name="178" href="#178">178</a>     }
 <a class="jxr_linenumber" name="179" href="#179">179</a> 
-<a class="jxr_linenumber" name="180" href="#180">180</a>     @Test(timeOut = 60_000)
+<a class="jxr_linenumber" name="180" href="#180">180</a>     @Test(timeOut = 80_000)
 <a class="jxr_linenumber" name="181" href="#181">181</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testLeaseHolderDoesNotChangeWhenANewLeaseManagerIsUp() <strong class="jxr_keyword">throws</strong> Exception {
 <a class="jxr_linenumber" name="182" href="#182">182</a> 
 <a class="jxr_linenumber" name="183" href="#183">183</a>         <strong class="jxr_keyword">final</strong> String TEST_TSO_LEASE_PATH = <span class="jxr_string">"/test2_tsolease"</span>;
@@ -196,7 +196,7 @@
 <a class="jxr_linenumber" name="186" href="#186">186</a>         <em class="jxr_comment">// Launch the master instance...</em>
 <a class="jxr_linenumber" name="187" href="#187">187</a>         TSOChannelHandler tsoChannelHandler1 = mock(TSOChannelHandler.<strong class="jxr_keyword">class</strong>);
 <a class="jxr_linenumber" name="188" href="#188">188</a>         TSOStateManager stateManager1 = mock(TSOStateManager.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="189" href="#189">189</a>         when(stateManager1.reset()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_1, DUMMY_EPOCH_1));
+<a class="jxr_linenumber" name="189" href="#189">189</a>         when(stateManager1.initialize()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_1, DUMMY_EPOCH_1));
 <a class="jxr_linenumber" name="190" href="#190">190</a>         leaseManager1 = <strong class="jxr_keyword">new</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a>(LEASE_MGR_ID_1,
 <a class="jxr_linenumber" name="191" href="#191">191</a>                                                  tsoChannelHandler1,
 <a class="jxr_linenumber" name="192" href="#192">192</a>                                                  stateManager1,
@@ -219,7 +219,7 @@
 <a class="jxr_linenumber" name="209" href="#209">209</a>         <em class="jxr_comment">// Then launch another instance...</em>
 <a class="jxr_linenumber" name="210" href="#210">210</a>         TSOChannelHandler tsoChannelHandler2 = mock(TSOChannelHandler.<strong class="jxr_keyword">class</strong>);
 <a class="jxr_linenumber" name="211" href="#211">211</a>         TSOStateManager stateManager2 = mock(TSOStateManager.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="212" href="#212">212</a>         when(stateManager2.reset()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_2, DUMMY_EPOCH_2));
+<a class="jxr_linenumber" name="212" href="#212">212</a>         when(stateManager2.initialize()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_2, DUMMY_EPOCH_2));
 <a class="jxr_linenumber" name="213" href="#213">213</a>         leaseManager2 = <strong class="jxr_keyword">new</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a>(LEASE_MGR_ID_2,
 <a class="jxr_linenumber" name="214" href="#214">214</a>                                                  tsoChannelHandler2,
 <a class="jxr_linenumber" name="215" href="#215">215</a>                                                  stateManager2,
@@ -240,7 +240,7 @@
 <a class="jxr_linenumber" name="230" href="#230">230</a>         assertFalse(leaseManager2.stillInLeasePeriod());
 <a class="jxr_linenumber" name="231" href="#231">231</a>     }
 <a class="jxr_linenumber" name="232" href="#232">232</a> 
-<a class="jxr_linenumber" name="233" href="#233">233</a>     @Test(timeOut = 60_000)
+<a class="jxr_linenumber" name="233" href="#233">233</a>     @Test(timeOut = 80_000)
 <a class="jxr_linenumber" name="234" href="#234">234</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testLeaseHolderChangesWhenActiveLeaseManagerIsPaused() <strong class="jxr_keyword">throws</strong> Exception {
 <a class="jxr_linenumber" name="235" href="#235">235</a> 
 <a class="jxr_linenumber" name="236" href="#236">236</a>         <strong class="jxr_keyword">final</strong> String TEST_TSO_LEASE_PATH = <span class="jxr_string">"/test3_tsolease"</span>;
@@ -249,7 +249,7 @@
 <a class="jxr_linenumber" name="239" href="#239">239</a>         <em class="jxr_comment">// Launch the master instance...</em>
 <a class="jxr_linenumber" name="240" href="#240">240</a>         TSOChannelHandler tsoChannelHandler1 = mock(TSOChannelHandler.<strong class="jxr_keyword">class</strong>);
 <a class="jxr_linenumber" name="241" href="#241">241</a>         TSOStateManager stateManager1 = mock(TSOStateManager.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="242" href="#242">242</a>         when(stateManager1.reset()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_1, DUMMY_EPOCH_1));
+<a class="jxr_linenumber" name="242" href="#242">242</a>         when(stateManager1.initialize()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_1, DUMMY_EPOCH_1));
 <a class="jxr_linenumber" name="243" href="#243">243</a>         leaseManager1 = <strong class="jxr_keyword">new</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a>(LEASE_MGR_ID_1,
 <a class="jxr_linenumber" name="244" href="#244">244</a>                                                  tsoChannelHandler1,
 <a class="jxr_linenumber" name="245" href="#245">245</a>                                                  stateManager1,
@@ -272,7 +272,7 @@
 <a class="jxr_linenumber" name="262" href="#262">262</a>         <em class="jxr_comment">// Then launch another instance...</em>
 <a class="jxr_linenumber" name="263" href="#263">263</a>         TSOChannelHandler tsoChannelHandler2 = mock(TSOChannelHandler.<strong class="jxr_keyword">class</strong>);
 <a class="jxr_linenumber" name="264" href="#264">264</a>         TSOStateManager stateManager2 = mock(TSOStateManager.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="265" href="#265">265</a>         when(stateManager2.reset()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_2, DUMMY_EPOCH_2));
+<a class="jxr_linenumber" name="265" href="#265">265</a>         when(stateManager2.initialize()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_2, DUMMY_EPOCH_2));
 <a class="jxr_linenumber" name="266" href="#266">266</a>         leaseManager2 = <strong class="jxr_keyword">new</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a>(LEASE_MGR_ID_2,
 <a class="jxr_linenumber" name="267" href="#267">267</a>                                                  tsoChannelHandler2,
 <a class="jxr_linenumber" name="268" href="#268">268</a>                                                  stateManager2,
@@ -295,7 +295,7 @@
 <a class="jxr_linenumber" name="285" href="#285">285</a>         assertTrue(leaseManager2.stillInLeasePeriod());
 <a class="jxr_linenumber" name="286" href="#286">286</a> 
 <a class="jxr_linenumber" name="287" href="#287">287</a>         <em class="jxr_comment">// Now, lets resume the first instance...</em>
-<a class="jxr_linenumber" name="288" href="#288">288</a>         when(stateManager1.reset()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_3, DUMMY_EPOCH_3));
+<a class="jxr_linenumber" name="288" href="#288">288</a>         when(stateManager1.initialize()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_3, DUMMY_EPOCH_3));
 <a class="jxr_linenumber" name="289" href="#289">289</a>         leaseManager1.resume();
 <a class="jxr_linenumber" name="290" href="#290">290</a> 
 <a class="jxr_linenumber" name="291" href="#291">291</a>         <em class="jxr_comment">// ... let the test run for some time...</em>
@@ -334,7 +334,7 @@
 <a class="jxr_linenumber" name="324" href="#324">324</a>     }
 <a class="jxr_linenumber" name="325" href="#325">325</a> 
 <a class="jxr_linenumber" name="326" href="#326">326</a> 
-<a class="jxr_linenumber" name="327" href="#327">327</a>     @Test(timeOut = 40_000)
+<a class="jxr_linenumber" name="327" href="#327">327</a>     @Test(timeOut = 80_000)
 <a class="jxr_linenumber" name="328" href="#328">328</a>     <strong class="jxr_keyword">public</strong> <strong class="jxr_keyword">void</strong> testLeaseManagerPanicsWhenUnexpectedInfoIsFoundInCurrentTSOZnode() <strong class="jxr_keyword">throws</strong> Exception {
 <a class="jxr_linenumber" name="329" href="#329">329</a> 
 <a class="jxr_linenumber" name="330" href="#330">330</a>         <strong class="jxr_keyword">final</strong> String TEST_TSO_LEASE_PATH = <span class="jxr_string">"/test_wronginfo_tsolease"</span>;
@@ -344,7 +344,7 @@
 <a class="jxr_linenumber" name="334" href="#334">334</a> 
 <a class="jxr_linenumber" name="335" href="#335">335</a>         <em class="jxr_comment">// Launch the master instance...</em>
 <a class="jxr_linenumber" name="336" href="#336">336</a>         TSOStateManager stateManager1 = mock(TSOStateManager.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="337" href="#337">337</a>         when(stateManager1.reset()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_1, DUMMY_EPOCH_1));
+<a class="jxr_linenumber" name="337" href="#337">337</a>         when(stateManager1.initialize()).thenReturn(<strong class="jxr_keyword">new</strong> TSOState(DUMMY_LOW_WATERMARK_1, DUMMY_EPOCH_1));
 <a class="jxr_linenumber" name="338" href="#338">338</a>         <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a> leaseManager = <strong class="jxr_keyword">new</strong> <a href="../../../../org/apache/omid/tso/PausableLeaseManager.html">PausableLeaseManager</a>(LEASE_MGR_ID_1,
 <a class="jxr_linenumber" name="339" href="#339">339</a>                                                                      mock(TSOChannelHandler.<strong class="jxr_keyword">class</strong>),
 <a class="jxr_linenumber" name="340" href="#340">340</a>                                                                      stateManager1,
@@ -370,7 +370,7 @@
 <a class="jxr_linenumber" name="360" href="#360">360</a>         Thread.sleep(TEST_LEASE_PERIOD_IN_MS * 2);
 <a class="jxr_linenumber" name="361" href="#361">361</a> 
 <a class="jxr_linenumber" name="362" href="#362">362</a>         ArgumentCaptor&lt;IllegalArgumentException&gt; trowableIAE = ArgumentCaptor.forClass(IllegalArgumentException.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="363" href="#363">363</a>         verify(panicker).panic(anyString(), trowableIAE.capture());
+<a class="jxr_linenumber" name="363" href="#363">363</a>         verify(panicker, times(2)).panic(anyString(), trowableIAE.capture());
 <a class="jxr_linenumber" name="364" href="#364">364</a>         assertTrue(trowableIAE.getValue() instanceof IllegalArgumentException);
 <a class="jxr_linenumber" name="365" href="#365">365</a>         assertTrue(trowableIAE.getValue().getMessage().contains(<span class="jxr_string">"Incorrect TSO Info found"</span>));
 <a class="jxr_linenumber" name="366" href="#366">366</a> 
@@ -389,7 +389,7 @@
 <a class="jxr_linenumber" name="379" href="#379">379</a> 
 <a class="jxr_linenumber" name="380" href="#380">380</a>         ArgumentCaptor&lt;LeaseManagement.LeaseManagementException&gt; trowableLME =
 <a class="jxr_linenumber" name="381" href="#381">381</a>                 ArgumentCaptor.forClass(LeaseManagement.LeaseManagementException.<strong class="jxr_keyword">class</strong>);
-<a class="jxr_linenumber" name="382" href="#382">382</a>         verify(panicker).panic(anyString(), trowableLME.capture());
+<a class="jxr_linenumber" name="382" href="#382">382</a>         verify(panicker, times(2)).panic(anyString(), trowableLME.capture());
 <a class="jxr_linenumber" name="383" href="#383">383</a>         assertTrue(trowableLME.getValue() instanceof LeaseManagement.LeaseManagementException);
 <a class="jxr_linenumber" name="384" href="#384">384</a>         assertTrue(trowableLME.getValue().getMessage().contains(<span class="jxr_string">"Another TSO replica was found"</span>));
 <a class="jxr_linenumber" name="385" href="#385">385</a>     }