2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.controller.config.persist.storage.file.xml;
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;
19 import java.nio.file.Files;
20 import java.util.List;
21 import java.util.SortedSet;
22 import java.util.TreeSet;
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;
29 public class FileStorageAdapterTest {
35 public void setUp() throws Exception {
36 file = Files.createTempFile("testFilePersist", ".txt").toFile();
39 com.google.common.io.Files.write("", file, Charsets.UTF_8);
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() {
50 public String getConfigSnapshot() {
51 return createConfig();
55 public SortedSet<String> getCapabilities() {
59 storage.persistConfig(holder);
61 storage.persistConfig(holder);
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());
71 storage = new XmlFileStorageAdapter();
72 storage.setFileStorage(file);
73 storage.setNumberOfBackups(Integer.MAX_VALUE);
75 List<ConfigSnapshotHolder> last = storage.loadLastConfigs();
76 Assert.assertEquals(createCaps(), last.get(0).getCapabilities());
79 private SortedSet<String> createCaps() {
80 SortedSet<String> caps = new TreeSet<>();
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);
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() {
96 public String getConfigSnapshot() {
97 return createConfig();
101 public SortedSet<String> getCapabilities() {
105 storage.persistConfig(holder);
107 storage.persistConfig(holder);
109 assertEquals(16, com.google.common.io.Files.readLines(file, Charsets.UTF_8).size());
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", ""));
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() {
125 public String getConfigSnapshot() {
126 return createConfig();
130 public SortedSet<String> getCapabilities() {
134 storage.persistConfig(holder);
136 storage.persistConfig(holder);
137 storage.persistConfig(holder);
139 List<String> readLines = com.google.common.io.Files.readLines(file, Charsets.UTF_8);
140 assertEquals(27, readLines.size());
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()));
151 public void testNoLastConfig() throws Exception {
152 File file = Files.createTempFile("testFilePersist", ".txt").toFile();
155 XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
156 storage.setFileStorage(file);
158 List<ConfigSnapshotHolder> elementOptional = storage.loadLastConfigs();
159 assertThat(elementOptional.size(), is(0));
162 @Test(expected = NullPointerException.class)
163 public void testNoProperties() throws Exception {
164 XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
165 storage.loadLastConfigs();
168 @Test(expected = NullPointerException.class)
169 public void testNoProperties2() throws Exception {
170 XmlFileStorageAdapter storage = new XmlFileStorageAdapter();
171 storage.persistConfig(new ConfigSnapshotHolder() {
173 public String getConfigSnapshot() {
174 return Mockito.mock(String.class);
178 public SortedSet<String> getCapabilities() {
179 return new TreeSet<>();
184 static String createConfig() {
185 return "<config>" + i++ + "</config>";