Add new api that is easier to use for user to add only one Attribute. (#571)

* Add new api that is easier to use for user to add only one Attribute.

* Rename addAttribute[s] to putAttribute[s] and deprecate the old API.

* Update tests to use putAttributes.

2 files changed, 14 insertions, 0 deletions

diff --git a/api/src/test/java/io/opencensus/trace/BlankSpanTest.java b/api/src/test/java/io/opencensus/trace/BlankSpanTest.java
index 8f7993ef..4e8342e2 100644
--- a/api/src/test/java/io/opencensus/trace/BlankSpanTest.java
+++ b/api/src/test/java/io/opencensus/trace/BlankSpanTest.java
@@ -44,6 +44,8 @@ public class BlankSpanTest {
     multipleAttributes.put("MyBooleanAttributeKey", AttributeValue.booleanAttributeValue(true));
     multipleAttributes.put("MyLongAttributeKey", AttributeValue.longAttributeValue(123));
     // Tests only that all the methods are not crashing/throwing errors.
+    BlankSpan.INSTANCE.putAttribute(
+        "MyStringAttributeKey2", AttributeValue.stringAttributeValue("MyStringAttributeValue2"));
     BlankSpan.INSTANCE.addAttributes(attributes);
     BlankSpan.INSTANCE.addAttributes(multipleAttributes);
     BlankSpan.INSTANCE.addAnnotation("MyAnnotation");
diff --git a/api/src/test/java/io/opencensus/trace/SpanTest.java b/api/src/test/java/io/opencensus/trace/SpanTest.java
index 1a25d6b7..5d36bb14 100644
--- a/api/src/test/java/io/opencensus/trace/SpanTest.java
+++ b/api/src/test/java/io/opencensus/trace/SpanTest.java
@@ -17,9 +17,11 @@
 package io.opencensus.trace;
 
 import static com.google.common.truth.Truth.assertThat;
+import static org.mockito.Matchers.eq;
 import static org.mockito.Matchers.same;
 import static org.mockito.Mockito.verify;
 
+import java.util.Collections;
 import java.util.EnumSet;
 import java.util.Map;
 import java.util.Random;
@@ -82,6 +84,16 @@ public class SpanTest {
   }
 
   @Test
+  public void addAttributeCallsAddAttributesByDefault() {
+    Span span = Mockito.spy(new NoopSpan(spanContext, spanOptions));
+    span.putAttribute("MyKey", AttributeValue.booleanAttributeValue(true));
+    span.end();
+    verify(span)
+        .addAttributes(
+            eq(Collections.singletonMap("MyKey", AttributeValue.booleanAttributeValue(true))));
+  }
+
+  @Test
   public void endCallsEndWithDefaultOptions() {
     Span span = Mockito.spy(new NoopSpan(spanContext, spanOptions));
     span.end();