opendaylight/config/config-persister-file-xml-adapter/src/test/java/org/opendaylight/controller/config/persist/storage/file/xml/FileStorageAdapterTest.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, 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
   9 package org.opendaylight.controller.config.persist.storage.file.xml;
  10
  11 import com.google.common.base.Charsets;
  12 import junit.framework.Assert;
  13 import org.junit.Before;
  14 import org.junit.Test;
  15 import org.mockito.Mockito;
  16 import org.opendaylight.controller.config.persist.api.ConfigSnapshotHolder;
  17
  18 import java.io.File;
  19 import java.nio.file.Files;
  20 import java.util.List;
  21 import java.util.SortedSet;
  22 import java.util.TreeSet;
  23
  24 import static junit.framework.Assert.assertFalse;
  25 import static org.hamcrest.CoreMatchers.is;
  26 import static org.junit.Assert.assertEquals;
  27 import static org.junit.Assert.assertThat;
  28
  29 public class FileStorageAdapterTest {
  30
  31     private static int i;
  32     private File file;
  33
  34     @Before
  35     public void setUp() throws Exception {
  36         file = Files.createTempFile("testFilePersist", ".txt").toFile();
  37         if (!file.exists())
  38             return;
  39         com.google.common.io.Files.write("", file, Charsets.UTF_8);
  40         i = 1;
  41     }
  42
  43     @Test
  44     public void testFileAdapter() throws Exception {
  45         XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
  46         storage.setFileStorage(file);
  47         storage.setNumberOfBackups(Integer.MAX_VALUE);
  48         final ConfigSnapshotHolder holder = new ConfigSnapshotHolder() {
  49             @Override
  50             public String getConfigSnapshot() {
  51                 return createConfig();
  52             }
  53
  54             @Override
  55             public SortedSet<String> getCapabilities() {
  56                 return createCaps();
  57             }
  58         };
  59         storage.persistConfig(holder);
  60
  61         storage.persistConfig(holder);
  62
  63         assertEquals(27, com.google.common.io.Files.readLines(file, Charsets.UTF_8).size());
  64         List<ConfigSnapshotHolder> lastConf = storage.loadLastConfigs();
  65         assertEquals(1, lastConf.size());
  66         ConfigSnapshotHolder configSnapshotHolder = lastConf.get(0);
  67         assertEquals("<config>2</config>",
  68                 configSnapshotHolder.getConfigSnapshot().replaceAll("\\s", ""));
  69         assertEquals(createCaps(), configSnapshotHolder.getCapabilities());
  70
  71         storage = new XmlFileStorageAdapter();
  72         storage.setFileStorage(file);
  73         storage.setNumberOfBackups(Integer.MAX_VALUE);
  74
  75         List<ConfigSnapshotHolder> last = storage.loadLastConfigs();
  76         Assert.assertEquals(createCaps(), last.get(0).getCapabilities());
  77     }
  78
  79     private SortedSet<String> createCaps() {
  80         SortedSet<String> caps = new TreeSet<>();
  81
  82         caps.add("cap1" + i);
  83         caps.add("cap2" + i);
  84         caps.add("urn:opendaylight:params:xml:ns:yang:controller:netty?module=netty&revision=2013-11-19" + i);
  85         caps.add("capaaaa as dasfasdf s2" + i);
  86         return caps;
  87     }
  88
  89     @Test
  90     public void testFileAdapterOneBackup() throws Exception {
  91         XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
  92         storage.setFileStorage(file);
  93         storage.setNumberOfBackups(1);
  94         final ConfigSnapshotHolder holder = new ConfigSnapshotHolder() {
  95             @Override
  96             public String getConfigSnapshot() {
  97                 return createConfig();
  98             }
  99
 100             @Override
 101             public SortedSet<String> getCapabilities() {
 102                 return createCaps();
 103             }
 104         };
 105         storage.persistConfig(holder);
 106
 107         storage.persistConfig(holder);
 108
 109         assertEquals(16, com.google.common.io.Files.readLines(file, Charsets.UTF_8).size());
 110
 111         List<ConfigSnapshotHolder> lastConf = storage.loadLastConfigs();
 112         assertEquals(1, lastConf.size());
 113         ConfigSnapshotHolder configSnapshotHolder = lastConf.get(0);
 114         assertEquals("<config>2</config>",
 115                 configSnapshotHolder.getConfigSnapshot().replaceAll("\\s", ""));
 116     }
 117
 118     @Test
 119     public void testFileAdapterOnlyTwoBackups() throws Exception {
 120         XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
 121         storage.setFileStorage(file);
 122         storage.setNumberOfBackups(2);
 123         final ConfigSnapshotHolder holder = new ConfigSnapshotHolder() {
 124             @Override
 125             public String getConfigSnapshot() {
 126                 return createConfig();
 127             }
 128
 129             @Override
 130             public SortedSet<String> getCapabilities() {
 131                 return createCaps();
 132             }
 133         };
 134         storage.persistConfig(holder);
 135
 136         storage.persistConfig(holder);
 137         storage.persistConfig(holder);
 138
 139         List<String> readLines = com.google.common.io.Files.readLines(file, Charsets.UTF_8);
 140         assertEquals(27, readLines.size());
 141
 142         List<ConfigSnapshotHolder> lastConf = storage.loadLastConfigs();
 143         assertEquals(1, lastConf.size());
 144         ConfigSnapshotHolder configSnapshotHolder = lastConf.get(0);
 145         assertEquals("<config>3</config>",
 146                 configSnapshotHolder.getConfigSnapshot().replaceAll("\\s", ""));
 147         assertFalse(readLines.contains(holder.getConfigSnapshot()));
 148     }
 149
 150     @Test
 151     public void testNoLastConfig() throws Exception {
 152         File file = Files.createTempFile("testFilePersist", ".txt").toFile();
 153         if (!file.exists())
 154             return;
 155         XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
 156         storage.setFileStorage(file);
 157
 158         List<ConfigSnapshotHolder> elementOptional = storage.loadLastConfigs();
 159         assertThat(elementOptional.size(), is(0));
 160     }
 161
 162     @Test(expected = NullPointerException.class)
 163     public void testNoProperties() throws Exception {
 164         XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
 165         storage.loadLastConfigs();
 166     }
 167
 168     @Test(expected = NullPointerException.class)
 169     public void testNoProperties2() throws Exception {
 170         XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
 171         storage.persistConfig(new ConfigSnapshotHolder() {
 172             @Override
 173             public String getConfigSnapshot() {
 174                 return Mockito.mock(String.class);
 175             }
 176
 177             @Override
 178             public SortedSet<String> getCapabilities() {
 179                 return new TreeSet<>();
 180             }
 181         } );
 182     }
 183
 184     static String createConfig() {
 185         return "<config>" + i++ + "</config>";
 186     }
 187
 188 }