juneau-core/juneau-core-utest/src/test/java/org/apache/juneau/config/store/ConfigFileStoreTest.java

// ***************************************************************************************************************************
// * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements.  See the NOTICE file *
// * distributed with this work for additional information regarding copyright ownership.  The ASF licenses this file        *
// * to you under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance            *
// * with the License.  You may obtain a copy of the License at                                                              *
// *                                                                                                                         *
// *  http://www.apache.org/licenses/LICENSE-2.0                                                                             *
// *                                                                                                                         *
// * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an  *
// * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the License for the        *
// * specific language governing permissions and limitations under the License.                                              *
// ***************************************************************************************************************************
package org.apache.juneau.config.store;

import static org.junit.Assert.*;

import java.io.*;
import java.util.concurrent.*;

import org.apache.juneau.internal.*;
import org.junit.*;

public class ConfigFileStoreTest {

	private static final File DIR = new File("./config");

	@After
	public void cleanUp() {
		FileUtils.delete(DIR);
	}

	@Test
	public void testNoFile() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).build();
		assertEquals("", fs.read("X.cfg"));
		assertFileNotExists("X.cfg");
	}

	@Test
	public void testDifferentExtension() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).build();
		assertEquals("", fs.read("X.xxx"));
		assertFileNotExists("X.xxx");
	}

	@Test
	public void testSimpleCreateAndDelete() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).build();
		assertNull(fs.write("X.cfg", null, "foo"));
		assertEquals("foo", fs.read("X.cfg"));
		assertFileExists("X.cfg");
		fs.write("X.cfg", "foo", null);
		assertFileNotExists("X.cfg");
	}

	@Test
	public void testSimpleCreateAndDeleteWithNoExtension() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).build();
		assertNull(fs.write("X", null, "foo"));
		assertEquals("foo", fs.read("X"));
		assertFileExists("X.cfg");
		fs.write("X", "foo", null);
		assertFileNotExists("X.cfg");
	}

	@Test
	public void testSimpleCreateAndDeleteWithNonStandardExtension() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).extensions("xxx").build();
		assertNull(fs.write("X", null, "foo"));
		assertEquals("foo", fs.read("X"));
		assertFileExists("X.xxx");
		fs.write("X", "foo", null);
		assertFileNotExists("X.xxx");
	}

	@Test
	public void testSimpleCreateAndDeleteWithMultipleSpecialExtension() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).extensions("foo1,foo2").build();
		assertNull(fs.write("X", null, "foo"));
		assertEquals("foo", fs.read("X"));
		assertFileExists("X.foo1");
		fs.write("X", "foo", null);
		assertFileNotExists("X.foo1");
	}

	@Test
	public void testFailOnMismatch() throws Exception {
		assertFileNotExists("X.cfg");
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).build();
		assertNotNull(fs.write("X.cfg", "xxx", "foo"));
		assertFileNotExists("X.cfg");
		assertEquals("", fs.read("X.cfg"));
		assertFileNotExists("X.cfg");
		assertNull(fs.write("X.cfg", null, "foo"));
		assertEquals("foo", fs.read("X.cfg"));
		assertNotNull(fs.write("X.cfg", "xxx", "foo"));
		assertEquals("foo", fs.read("X.cfg"));
		assertNull(fs.write("X.cfg", "foo", "bar"));
		assertEquals("bar", fs.read("X.cfg"));
	}

	@Test
	public void testFailOnMismatchNoExtension() throws Exception {
		assertFileNotExists("X.cfg");
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).build();
		assertNotNull(fs.write("X", "xxx", "foo"));
		assertFileNotExists("X.cfg");
		assertEquals("", fs.read("X"));
		assertEquals("", fs.read("X.cfg"));
		assertFileNotExists("X.cfg");
		assertNull(fs.write("X", null, "foo"));
		assertEquals("foo", fs.read("X"));
		assertEquals("foo", fs.read("X.cfg"));
		assertNotNull(fs.write("X", "xxx", "foo"));
		assertEquals("foo", fs.read("X"));
		assertEquals("foo", fs.read("X.cfg"));
		assertNull(fs.write("X", "foo", "bar"));
		assertEquals("bar", fs.read("X"));
		assertEquals("bar", fs.read("X.cfg"));
	}

	@Test
	public void testCharset() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).charset("UTF-8").build();
		assertNull(fs.write("X.cfg", null, "foo"));
		assertEquals("foo", fs.read("X.cfg"));
		assertEquals("foo", fs.read("X"));
	}

	@Test
	public void testCharsetNoExtension() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).charset("UTF-8").build();
		assertNull(fs.write("X", null, "foo"));
		assertEquals("foo", fs.read("X"));
		assertEquals("foo", fs.read("X.cfg"));
	}

	@Test
	public void testWatcher_LONGRUNNING() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).useWatcher().watcherSensitivity(WatcherSensitivity.HIGH).build();

		final CountDownLatch latch = new CountDownLatch(4);
		fs.register("X.cfg", new ConfigStoreListener() {
			@Override
			public void onChange(String contents) {
				if ("xxx".equals(contents))
					latch.countDown();
			}
		});
		fs.register("X", new ConfigStoreListener() {
			@Override
			public void onChange(String contents) {
				if ("xxx".equals(contents))
					latch.countDown();
			}
		});
		fs.register("Y.cfg", new ConfigStoreListener() {
			@Override
			public void onChange(String contents) {
				if ("yyy".equals(contents))
					latch.countDown();
			}
		});
		fs.register("Y", new ConfigStoreListener() {
			@Override
			public void onChange(String contents) {
				if ("yyy".equals(contents))
					latch.countDown();
			}
		});
		IOUtils.write(new File(DIR, "Z.ini"), new StringReader("zzz"));
		IOUtils.write(new File(DIR, "X.cfg"), new StringReader("xxx"));
		IOUtils.write(new File(DIR, "Y.cfg"), new StringReader("yyy"));
		if (! latch.await(10, TimeUnit.SECONDS))
			throw new Exception("CountDownLatch never reached zero.");
	}

	@Test
	public void testUpdate() throws Exception {
		ConfigFileStore fs = ConfigFileStore.create().directory(DIR).build();

		final CountDownLatch latch = new CountDownLatch(4);
		fs.register("X.cfg", new ConfigStoreListener() {
			@Override
			public void onChange(String contents) {
				if ("xxx".equals(contents))
					latch.countDown();
			}
		});
		fs.register("X", new ConfigStoreListener() {
			@Override
			public void onChange(String contents) {
				if ("xxx".equals(contents))
					latch.countDown();
			}
		});
		fs.register("Y.cfg", new ConfigStoreListener() {
			@Override
			public void onChange(String contents) {
				if ("yyy".equals(contents))
					latch.countDown();
			}
		});
		fs.register("Y", new ConfigStoreListener() {
			@Override
			public void onChange(String contents) {
				if ("yyy".equals(contents))
					latch.countDown();
			}
		});

		fs.update("X.cfg", "xxx");
		fs.update("Y.cfg", "yyy");
		if (! latch.await(10, TimeUnit.SECONDS))
			throw new Exception("CountDownLatch never reached zero.");
	}

	@Test
	public void testExists() throws IOException {
		ConfigFileStore cs = ConfigFileStore.DEFAULT;
		assertTrue(cs.exists("test.cfg"));
		assertTrue(cs.exists("test"));
		assertFalse(cs.exists("test2.cfg"));

		assertFalse(cs.exists("foo.cfg"));
		cs.write("foo.cfg", null, "foo");
		assertTrue(cs.exists("foo.cfg"));
		assertTrue(cs.exists("foo"));
		cs.write("foo.cfg", "foo", null);
		assertFalse(cs.exists("foo.cfg"));
		assertFalse(cs.exists("foo"));

		IOUtils.write(new File("Foox.cfg"), new StringReader("xxx"));
		assertTrue(cs.exists("Foox.cfg"));
		assertTrue(cs.exists("Foox"));
		new File("Foox.cfg").delete();
		assertFalse(cs.exists("Foox.cfg"));
		assertFalse(cs.exists("Foox"));
	}

	private void assertFileExists(String name) {
		assertTrue(new File(DIR, name).exists());
	}

	private void assertFileNotExists(String name) {
		assertTrue(! new File(DIR, name).exists());
	}
}