mdsalutil/mdsalutil-testutils/src/test/java/org/opendaylight/genius/datastoreutils/testutils/tests/DataBrokerFailuresTest.java

   1 /*
   2  * Copyright (c) 2017 Red Hat, Inc. and others. All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.genius.datastoreutils.testutils.tests;
   9
  10 import static org.junit.Assert.fail;
  11 import static org.junit.runners.MethodSorters.NAME_ASCENDING;
  12 import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
  13
  14 import javax.inject.Inject;
  15 import org.junit.FixMethodOrder;
  16 import org.junit.Rule;
  17 import org.junit.Test;
  18 import org.junit.rules.MethodRule;
  19 import org.mockito.Mockito;
  20 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  21 import org.opendaylight.controller.md.sal.common.api.data.OptimisticLockFailedException;
  22 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  23 import org.opendaylight.genius.datastoreutils.testutils.DataBrokerFailures;
  24 import org.opendaylight.genius.datastoreutils.testutils.DataBrokerFailuresModule;
  25 import org.opendaylight.infrautils.inject.guice.testutils.AnnotationsModule;
  26 import org.opendaylight.infrautils.inject.guice.testutils.GuiceRule;
  27
  28 /**
  29  * Unit test for DataBrokerFailuresImpl.
  30  *
  31  * @author Michael Vorburger.ch
  32  */
  33 @FixMethodOrder(NAME_ASCENDING)
  34 public class DataBrokerFailuresTest {
  35
  36     private final DataBroker mockDataBroker = Mockito.mock(DataBroker.class, RETURNS_DEEP_STUBS);
  37
  38     public @Rule MethodRule guice = new GuiceRule(
  39             new DataBrokerFailuresModule(mockDataBroker), new AnnotationsModule());
  40
  41     @Inject DataBrokerFailures dbFailures;
  42     @Inject DataBroker dataBroker;
  43
  44     @Test
  45     public void testFailReadWriteTransactionSubmit() throws TransactionCommitFailedException {
  46         dbFailures.failSubmits(new OptimisticLockFailedException("bada boum bam!"));
  47         checkSubmitFails();
  48         // Now make sure that it still fails, and not just once:
  49         checkSubmitFails();
  50         // and still:
  51         checkSubmitFails();
  52     }
  53
  54     private void checkSubmitFails() {
  55         try {
  56             dataBroker.newReadWriteTransaction().submit().checkedGet();
  57             fail("This should have lead to a TransactionCommitFailedException!");
  58         } catch (TransactionCommitFailedException e) {
  59             // as expected!
  60         }
  61     }
  62
  63     @Test
  64     public void testFailReadWriteTransactionSubmitNext() throws TransactionCommitFailedException {
  65         // This must pass (the failSubmits from previous test cannot affect this)
  66         // (It's a completely new instance of DataBroker & DataBrokerFailures anyways, but just to be to sure.)
  67         dataBroker.newReadWriteTransaction().submit().checkedGet();
  68     }
  69
  70     @Test
  71     public void testFailTwoReadWriteTransactionSubmit() throws TransactionCommitFailedException {
  72         dbFailures.failSubmits(2, new OptimisticLockFailedException("bada boum bam!"));
  73         checkSubmitFails();
  74         // Now make sure that it still fails again a 2nd time, and not just once:
  75         checkSubmitFails();
  76         // But now it should pass.. because we specified howManyTimes = 2 above
  77         dataBroker.newReadWriteTransaction().submit().checkedGet();
  78         dataBroker.newWriteOnlyTransaction().submit().checkedGet();
  79         dataBroker.newReadWriteTransaction().submit().checkedGet();
  80     }
  81
  82     @Test(expected = OptimisticLockFailedException.class)
  83     public void testFailWriteTransactionSubmit() throws TransactionCommitFailedException {
  84         dbFailures.failSubmits(new OptimisticLockFailedException("bada boum bam!"));
  85         dataBroker.newWriteOnlyTransaction().submit().checkedGet();
  86     }
  87
  88     @Test
  89     public void testUnfailSubmits() throws TransactionCommitFailedException {
  90         dbFailures.failSubmits(new OptimisticLockFailedException("bada boum bam!"));
  91         checkSubmitFails();
  92         dbFailures.unfailSubmits();
  93         dataBroker.newReadWriteTransaction().submit().checkedGet();
  94         dataBroker.newWriteOnlyTransaction().submit().checkedGet();
  95         dataBroker.newReadWriteTransaction().submit().checkedGet();
  96     }
  97
  98     @Test
  99     public void testFailButSubmitsAnywaysReadWriteTransaction() {
 100         dbFailures.failButSubmitsAnyways();
 101         checkSubmitFails();
 102     }
 103
 104     // TODO make this work for TransactionChain as well ...
 105
 106 }