/*
 * Copyright 2012 Splunk, Inc.
 *
 * Licensed 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 com.splunk;

import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

public class ConfigurationTest extends SDKTestCase {
    private Service appService;
    private String applicationName;

    @Before
    @Override
    public void setUp() throws Exception {
        super.setUp();
        
        // We cannot straightforwardly delete confs,
        // so instead we create a temporary application,
        // reconnect using its namespace, then delete it
        // when we're done.
        applicationName = createTemporaryName();
        service.getApplications().create(applicationName);

        // We cannot easily change the namespace of service,
        // so rather than fight with it, we create a new
        // service with the right namespace and work from it
        // instead. Then we'll return to service in tearDown
        // to clean up.

        // NOTE: If you ever have to put a restart into
        // the configuration tests, then you will need
        // to explicitly re-login appService, since it will
        // not be re-logged in by the call to restartSplunk.
        Args applicationArgs = new Args(command.opts);
        applicationArgs.put("sharing", "app");
        applicationArgs.put("owner", "nobody");
        applicationArgs.put("app", applicationName);
        appService = Service.connect(applicationArgs);
    }

    @After
    @Override
    public void tearDown() throws Exception {
        appService.logout();
        
        final EntityCollection<Application> apps = service.getApplications();
        apps.remove(applicationName);
        assertEventuallyTrue(new EventuallyTrueBehavior() {
            @Override
            public boolean predicate() {
                return !apps.refresh().containsKey(applicationName);
            }
        });
        clearRestartMessage();
        
        super.tearDown();
    }

    @Test
    public void testStandardFilesExist() {
        ConfCollection confs = appService.getConfs();
        Assert.assertTrue(confs.containsKey("eventtypes"));
        Assert.assertTrue(confs.containsKey("indexes"));
        Assert.assertTrue(confs.containsKey("inputs"));
        Assert.assertTrue(confs.containsKey("props"));
        Assert.assertTrue(confs.containsKey("transforms"));
        Assert.assertTrue(confs.containsKey("savedsearches"));

        for (Entity stanza : confs.get("indexes").values()) {
            Assert.assertNotNull(stanza);
            Assert.assertNotNull(stanza.getName());
            Assert.assertNotNull(stanza.getPath());
        }
    }

    @Test
    public void testCreateConf() {
        ConfCollection confs = appService.getConfs();
        String confName = createTemporaryName();

        Assert.assertFalse("New configuration name already used.", confs.containsKey(confName));
        EntityCollection<Entity> conf = confs.create(confName);
        Assert.assertTrue("New configuration doesn't show up after creation.", confs.containsKey(confName));
        Assert.assertEquals("New configuration is not empty.", 0, conf.size());
    }

    @Test
    public void testCreateAndDeleteStanza() {
        ConfCollection confs = appService.getConfs();
        String confName = createTemporaryName();
        EntityCollection<Entity> conf = confs.create(confName);

        String stanzaName = createTemporaryName();
        Assert.assertFalse("Stanza already exists.", conf.containsKey(stanzaName));
        Entity stanza = conf.create(stanzaName);

        Assert.assertTrue("Stanza doesn't show up after creation.", conf.containsKey(stanzaName));
        Assert.assertEquals("Stanza contains something besides eai: and disabled attributes when first created.", 5, stanza.size());

        stanza.remove();
        conf.refresh();
        Assert.assertFalse("Stanza not deleted by remove()", conf.containsKey(stanzaName));
    }

    @Test
    public void testWriteToStanza() {
        ConfCollection confs = appService.getConfs();
        String confName = createTemporaryName();
        EntityCollection<Entity> conf = confs.create(confName);
        String stanzaName = createTemporaryName();
        Entity stanza = conf.create(stanzaName);

        Assert.assertEquals("Stanza begins with only eai:* and disabled in it.", 5, stanza.size());
        Args args = new Args();
        args.put("test-key", "42");
        stanza.update(args);
        Assert.assertEquals("Stanza has right size after write.", 6, stanza.size());
        Assert.assertEquals("Read from key gives right value.", "42", stanza.get("test-key"));
    }
}