Mercurial > hg > release > thermostat-0.15
view agent/cli/src/test/java/com/redhat/thermostat/agent/cli/impl/db/MongoProcessRunnerTest.java @ 1233:4279c0994a38
Add support for mongodb v2.4.
Reviewed-by: omajid
Review-thread: http://icedtea.classpath.org/pipermail/thermostat/2013-August/008039.html
author | Severin Gehwolf <sgehwolf@redhat.com> |
---|---|
date | Mon, 26 Aug 2013 18:33:02 +0200 |
parents | d1f55aa2b083 |
children |
line wrap: on
line source
package com.redhat.thermostat.agent.cli.impl.db; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNull; import static org.junit.Assert.fail; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; import java.io.BufferedReader; import java.io.ByteArrayInputStream; import java.io.File; import java.io.IOException; import java.util.List; import org.junit.After; import org.junit.Before; import org.junit.Test; import com.redhat.thermostat.agent.cli.impl.db.MongoProcessRunner; import com.redhat.thermostat.shared.config.InvalidConfigurationException; public class MongoProcessRunnerTest { private MongoProcessRunner runner; private DBStartupConfiguration config; private static final String NO_JOURNAL_MONGODB_VERSION = "2.0.0"; private static final String JOURNAL_MONGODB_VERSION = "1.8.0"; private static final String BIND_IP = "127.0.0.1"; private static final long PORT = 12456; @Before public void setUp() { File dbPath = new File("/path/to/db"); File logPath = new File("/path/to/log"); File pidFile = new File("/path/to/pid"); config = mock(DBStartupConfiguration.class); when(config.getBindIP()).thenReturn(BIND_IP); when(config.getPort()).thenReturn(PORT); when(config.getDBPath()).thenReturn(dbPath); when(config.getLogFile()).thenReturn(logPath); when(config.getPidFile()).thenReturn(pidFile); runner = new MongoProcessRunner(config, false); } @After public void tearDown() { runner = null; config = null; } @Test public void testCommandArgumentsWithJournalVersion() throws Exception { String[] expected = { "mongod", "--nojournal", "--quiet", "--fork", "--auth", "--nohttpinterface", "--bind_ip", config.getBindIP(), "--dbpath", config.getDBPath().getCanonicalPath(), "--logpath", config.getLogFile().getCanonicalPath(), "--pidfilepath", config.getPidFile().getCanonicalPath(), "--port", Long.toString(config.getPort()) }; List<String> cmds = runner.getStartupCommand(NO_JOURNAL_MONGODB_VERSION); String[] actual = cmds.toArray(new String[0]); verifyEquals(expected, actual); } @Test public void testCommandArgumentsWithNoJournalVersion() throws Exception { String[] expected = { "mongod", "--quiet", "--fork", "--auth", "--nohttpinterface", "--bind_ip", config.getBindIP(), "--dbpath", config.getDBPath().getCanonicalPath(), "--logpath", config.getLogFile().getCanonicalPath(), "--pidfilepath", config.getPidFile().getCanonicalPath(), "--port", Long.toString(config.getPort()) }; List<String> cmds = runner.getStartupCommand(JOURNAL_MONGODB_VERSION); String[] actual = cmds.toArray(new String[0]); verifyEquals(expected, actual); } @Test public void testCommandArgumentsWithSSLEnabled() throws Exception { when(config.isSslEnabled()).thenReturn(true); File pemFile = new File("/path/to/cert_and_key.pem"); when(config.getSslPemFile()).thenReturn(pemFile); when(config.getSslKeyPassphrase()).thenReturn("non-null"); String[] expected = { "mongod", "--quiet", "--fork", "--auth", "--nohttpinterface", "--bind_ip", config.getBindIP(), "--dbpath", config.getDBPath().getCanonicalPath(), "--logpath", config.getLogFile().getCanonicalPath(), "--pidfilepath", config.getPidFile().getCanonicalPath(), "--port", Long.toString(config.getPort()), "--sslOnNormalPorts", "--sslPEMKeyFile", config.getSslPemFile().getCanonicalPath(), "--sslPEMKeyPassword", config.getSslKeyPassphrase() }; List<String> cmds = runner.getStartupCommand(JOURNAL_MONGODB_VERSION); String[] actual = cmds.toArray(new String[0]); verifyEquals(expected, actual); } @Test public void testCommandArgumentsWithSSLEnabledThrowsInvalidConfigException() throws IOException { when(config.isSslEnabled()).thenReturn(true); // PEM file can't be null when SSL == true when(config.getSslPemFile()).thenReturn(null); try { runner.getStartupCommand(JOURNAL_MONGODB_VERSION); fail("Should have thrown exception!"); } catch (InvalidConfigurationException e) { assertEquals("No SSL PEM file specified!", e.getMessage()); } // Key password can't be null when SSL == true and keyfile present File pemFile = new File("/path/to/ssl.pem"); when(config.getSslPemFile()).thenReturn(pemFile); when(config.getSslKeyPassphrase()).thenReturn(null); try { runner.getStartupCommand(JOURNAL_MONGODB_VERSION); fail("Should have thrown exception!"); } catch (InvalidConfigurationException e) { assertEquals("No SSL key passphrase set!", e.getMessage()); } } @Test public void testCheckPidNull() throws IOException { BufferedReader reader = mock(BufferedReader.class); when(reader.readLine()).thenReturn(null); Integer pid = runner.doGetPid(reader); assertNull(pid); } @Test public void canGetVersionFromVersionCmdOutputV22() throws IOException { String versionOutput = "db version v2.2.4, pdfile version 4.5\n" + "Mon Aug 26 17:13:45 git version: nogitversion"; ByteArrayInputStream in = new ByteArrayInputStream(versionOutput.getBytes()); String version = runner.doGetDBVersion(in); assertEquals("2.2.4", version); } @Test public void canGetVersionFromVersionCmdOutputV24() throws IOException { String versionOutput = "db version v2.4.5\n" + "Mon Aug 26 18:01:28.404 git version: nogitversion"; ByteArrayInputStream in = new ByteArrayInputStream(versionOutput.getBytes()); String version = runner.doGetDBVersion(in); assertEquals("2.4.5", version); } @Test public void canProceedIfGetDbVersionThrowsException() throws IOException { String versionOutput = "foo\n" + "Mon Aug 26 18:01:28.404 git version: nogitversion"; ByteArrayInputStream in = new ByteArrayInputStream(versionOutput.getBytes()); String version = runner.doGetDBVersion(in); assertEquals(MongoProcessRunner.NO_JOURNAL_FIRST_VERSION, version); } private void verifyEquals(String[] expected, String[] actual) { assertEquals(expected.length, actual.length); for (int i=0; i < expected.length; i++) { assertEquals(expected[i], actual[i]); } } }