From 4226100ea92afba457903442590127b732a555c8 Mon Sep 17 00:00:00 2001
From: Nancy Bosecker <nbosecker@sourceintelligence.com>
Date: Wed, 16 Aug 2017 12:10:01 -0700
Subject: [PATCH] moved test file to correct location (#2437)

---
 .../commons/collections4/BagTests.java        | 85 +++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 libraries/src/test/java/com/baeldung/commons/collections4/BagTests.java

diff --git a/libraries/src/test/java/com/baeldung/commons/collections4/BagTests.java b/libraries/src/test/java/com/baeldung/commons/collections4/BagTests.java
new file mode 100644
index 000000000000..1270c5000887
--- /dev/null
+++ b/libraries/src/test/java/com/baeldung/commons/collections4/BagTests.java
@@ -0,0 +1,85 @@
+package com.baeldung.commons.collections4;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.apache.commons.collections4.Bag;
+import org.apache.commons.collections4.bag.CollectionBag;
+import org.apache.commons.collections4.bag.HashBag;
+import org.apache.commons.collections4.bag.TreeBag;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+
+public class BagTests {
+
+	Bag<String> baseBag;
+	TreeBag<String> treeBag;
+
+	@Before
+	public void before() {
+		baseBag = new HashBag<String>();
+		treeBag = new TreeBag<String>();
+		treeBag = new TreeBag<String>();
+	}
+
+	@Test
+	public void whenAdd_thenRemoveFromBaseBag_thenContainsCorrect() {
+		baseBag.add("apple", 2);
+		baseBag.add("lemon", 6);
+		baseBag.add("lime");
+
+		baseBag.remove("lemon");
+		Assert.assertEquals(3, baseBag.size());
+		Assert.assertFalse(baseBag.contains("lemon"));
+
+		Assert.assertTrue(baseBag.uniqueSet().contains("apple"));
+		Assert.assertFalse(baseBag.uniqueSet().contains("lemon"));
+		Assert.assertTrue(baseBag.uniqueSet().contains("lime"));
+
+		List<String> containList = new ArrayList<String>();
+		containList.add("apple");
+		containList.add("lemon");
+		containList.add("lime");
+		Assert.assertFalse(baseBag.containsAll(containList));
+	}
+
+	@Test
+	public void whenAdd_thenRemoveFromBaseCollectionBag_thenContainsCorrect() {
+		baseBag.add("apple", 2);
+		baseBag.add("lemon", 6);
+		baseBag.add("lime");
+
+		CollectionBag<String> baseCollectionBag = new CollectionBag<String>(
+				baseBag);
+
+		baseCollectionBag.remove("lemon");
+		Assert.assertEquals(8, baseCollectionBag.size());
+		Assert.assertTrue(baseCollectionBag.contains("lemon"));
+
+		Assert.assertTrue(baseBag.uniqueSet().contains("apple"));
+		Assert.assertTrue(baseBag.uniqueSet().contains("lemon"));
+		Assert.assertTrue(baseBag.uniqueSet().contains("lime"));
+
+		List<String> containList = new ArrayList<String>();
+		containList.add("apple");
+		containList.add("lemon");
+		containList.add("lime");
+		Assert.assertTrue(baseBag.containsAll(containList));
+	}
+	
+	@Test
+	public void whenAddtoTreeBag_thenRemove_thenContainsCorrect() {
+		treeBag.add("banana", 8);
+		treeBag.add("apple", 2);
+		treeBag.add("lime");
+		
+		Assert.assertEquals(11, treeBag.size());
+		Assert.assertEquals("apple", treeBag.first());
+		Assert.assertEquals("lime", treeBag.last());
+		
+		treeBag.remove("apple");
+		Assert.assertEquals(9, treeBag.size());
+		Assert.assertEquals("banana", treeBag.first());
+	}
+}