From 4fa5e2d85c6cb49ba25d0fc8b92d0b962ba91db8 Mon Sep 17 00:00:00 2001
From: Miguel Serra <Miguel.Serra@criticaltechworks.com>
Date: Wed, 25 Dec 2019 14:50:20 +0000
Subject: [PATCH] KT-20357: Add samples for linkedSetOf

---
 .../test/samples/collections/collections.kt   | 22 +++++++++++++++++++
 .../stdlib/src/kotlin/collections/Sets.kt     |  6 ++++-
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/libraries/stdlib/samples/test/samples/collections/collections.kt b/libraries/stdlib/samples/test/samples/collections/collections.kt
index ec056efd8f5..66166138b95 100644
--- a/libraries/stdlib/samples/test/samples/collections/collections.kt
+++ b/libraries/stdlib/samples/test/samples/collections/collections.kt
@@ -333,6 +333,28 @@ class Collections {
             set += listOf(4, 5)
             assertPrints(set, "[1, 2, 4, 5]")
         }
+
+        @Sample
+        fun emptyLinkedHashSet() {
+            val set: LinkedHashSet<Int> = linkedSetOf<Int>()
+
+            set.add(1)
+            set.add(3)
+            set.add(2)
+
+            assertPrints(set, "[1, 3, 2]")
+        }
+
+        @Sample
+        fun linkedHashSet() {
+            val set: LinkedHashSet<Int> = linkedSetOf(1, 3, 2)
+
+            assertPrints(set, "[1, 3, 2]")
+
+            set.remove(3)
+            set += listOf(5, 4)
+            assertPrints(set, "[1, 2, 5, 4]")
+        }
     }
 
     class Transformations {
diff --git a/libraries/stdlib/src/kotlin/collections/Sets.kt b/libraries/stdlib/src/kotlin/collections/Sets.kt
index 391b9ec81d1..b88026c1621 100644
--- a/libraries/stdlib/src/kotlin/collections/Sets.kt
+++ b/libraries/stdlib/src/kotlin/collections/Sets.kt
@@ -73,7 +73,10 @@ public inline fun <T> hashSetOf(): HashSet<T> = HashSet()
 /** Returns a new [HashSet] with the given elements. */
 public fun <T> hashSetOf(vararg elements: T): HashSet<T> = elements.toCollection(HashSet(mapCapacity(elements.size)))
 
-/** Returns an empty new [LinkedHashSet]. */
+/**
+ * Returns an empty new [LinkedHashSet].
+ * @sample samples.collections.Collections.Sets.emptyLinkedHashSet
+ */
 @SinceKotlin("1.1")
 @kotlin.internal.InlineOnly
 public inline fun <T> linkedSetOf(): LinkedHashSet<T> = LinkedHashSet()
@@ -81,6 +84,7 @@ public inline fun <T> linkedSetOf(): LinkedHashSet<T> = LinkedHashSet()
 /**
  * Returns a new [LinkedHashSet] with the given elements.
  * Elements of the set are iterated in the order they were specified.
+ * @sample samples.collections.Collections.Sets.linkedHashSet
  */
 public fun <T> linkedSetOf(vararg elements: T): LinkedHashSet<T> = elements.toCollection(LinkedHashSet(mapCapacity(elements.size)))