MessageTest.java

/*
 * Copyright 2011 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.Assert;
import org.junit.Test;

public class MessageTest extends SDKTestCase {
    @Test
    public void testMessage() throws Exception {
        MessageCollection messageCollection = service.getMessages();

        if (messageCollection.containsKey("sdk-test-message1"))
            messageCollection.remove("sdk-test-message1");
        Assert.assertFalse(messageCollection.containsKey("sdk-test-message1"));

        if (messageCollection.containsKey("sdk-test-message2"))
            messageCollection.remove("sdk-test-message2");
        Assert.assertFalse(messageCollection.containsKey("sdk-test-message2"));

        // Create & Check
        {
            messageCollection.create("sdk-test-message1", "hello.");
            Assert.assertTrue(messageCollection.containsKey("sdk-test-message1"));
            Message message = messageCollection.get("sdk-test-message1");
            Assert.assertEquals("sdk-test-message1", message.getKey());
            Assert.assertEquals("hello.", message.getValue());
    
            messageCollection.create("sdk-test-message2", new Args("value", "world."));
            Assert.assertTrue(messageCollection.containsKey("sdk-test-message2"));
            message = messageCollection.get("sdk-test-message2");
            Assert.assertEquals("sdk-test-message2", message.getKey());
            Assert.assertEquals("world.", message.getValue());
        }

        messageCollection.remove("sdk-test-message1");
        Assert.assertFalse(messageCollection.containsKey("sdk-test-message1"));

        messageCollection.remove("sdk-test-message2");
        Assert.assertFalse(messageCollection.containsKey("sdk-test-message2"));
    }
}