ObjectNameBuilderTest.java

/*
 * Copyright 2014 Netflix, 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.netflix.servo.jmx;

import com.netflix.servo.tag.BasicTagList;
import com.netflix.servo.tag.Tags;
import org.testng.annotations.Test;

import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;

import static org.testng.Assert.assertEquals;
import static org.testng.Assert.fail;

public class ObjectNameBuilderTest {

  @Test
  public void testInvalidCharactersSanitized() {
    ObjectName name =
        ObjectNameBuilder.forDomain("test*Domain&")
            .addProperty("foo%", "$bar")
            .build();
    assertEquals(name.getDomain(), "test_Domain_");
    assertEquals(name.getKeyPropertyListString(), "foo_=_bar");
  }

  @Test
  public void testAddTagList() {
    ObjectName name =
        ObjectNameBuilder.forDomain("testDomain")
            .addProperties(BasicTagList.of("foo", "bar", "test", "stuff"))
            .build();
    assertEquals(name.getDomain(), "testDomain");
    assertEquals(name.getKeyPropertyListString(), "foo=bar,test=stuff");
  }

  @Test
  public void testTagByTag() {
    // Order will be in the order tags were added to the builder
    ObjectName name =
        ObjectNameBuilder.forDomain("testDomain")
            .addProperty(Tags.newTag("foo", "bar"))
            .addProperty(Tags.newTag("test", "stuff"))
            .build();
    assertEquals(name.getDomain(), "testDomain");
    assertEquals(name.getKeyPropertyListString(), "foo=bar,test=stuff");
  }

  @Test
  public void testBuildWithoutPropertyAdded() {
    try {
      ObjectNameBuilder.forDomain("testDomain").build();
      fail("Should have thrown an exception without keys being added!");
    } catch (RuntimeException expected) {
      assertEquals(expected.getCause().getClass(), MalformedObjectNameException.class);
    }
  }

}