#354 Moved Tests to the correct area

feature-toggle/src/test/java/com/iluwatar/featuretoggle/pattern/tieredversion/TieredFeatureToggleVersionTest.java

@@ -0,0 +1,36 @@
+package com.iluwatar.featuretoggle.pattern.tieredversion;
+
+import com.iluwatar.featuretoggle.pattern.Service;
+import com.iluwatar.featuretoggle.user.User;
+import com.iluwatar.featuretoggle.user.UserGroup;
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+public class TieredFeatureToggleVersionTest {
+
+  final User paidUser = new User("Jamie Coder");
+  final User freeUser = new User("Alan Defect");
+  final Service service = new TieredFeatureToggleVersion();
+
+  @Before
+  public void setUp() throws Exception {
+    UserGroup.addUserToPaidGroup(paidUser);
+    UserGroup.addUserToFreeGroup(freeUser);
+  }
+
+  @Test
+  public void testGetWelcomeMessageForPaidUser() throws Exception {
+    final String welcomeMessage = service.getWelcomeMessage(paidUser);
+    final String expected = "You're amazing Jamie Coder. Thanks for paying for this awesome software.";
+    assertEquals(expected, welcomeMessage);
+  }
+
+  @Test
+  public void testGetWelcomeMessageForFreeUser() throws Exception {
+    final String welcomeMessage = service.getWelcomeMessage(freeUser);
+    final String expected = "I suppose you can use this software.";
+    assertEquals(expected, welcomeMessage);
+  }
+}

feature-toggle/src/test/java/com/iluwatar/featuretoggle/user/UserGroupTest.java

@@ -0,0 +1,37 @@
+package com.iluwatar.featuretoggle.user;
+
+import org.junit.Test;
+
+import static junit.framework.TestCase.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+public class UserGroupTest {
+
+  @Test
+  public void testAddUserToFreeGroup() throws Exception {
+    User user = new User("Free User");
+    UserGroup.addUserToFreeGroup(user);
+    assertFalse(UserGroup.isPaid(user));
+  }
+
+  @Test
+  public void testAddUserToPaidGroup() throws Exception {
+    User user = new User("Paid User");
+    UserGroup.addUserToPaidGroup(user);
+    assertTrue(UserGroup.isPaid(user));
+  }
+
+  @Test(expected = IllegalArgumentException.class)
+  public void testAddUserToPaidWhenOnFree() throws Exception {
+    User user = new User("Paid User");
+    UserGroup.addUserToFreeGroup(user);
+    UserGroup.addUserToPaidGroup(user);
+  }
+
+  @Test(expected = IllegalArgumentException.class)
+  public void testAddUserToFreeWhenOnPaid() throws Exception {
+    User user = new User("Free User");
+    UserGroup.addUserToPaidGroup(user);
+    UserGroup.addUserToFreeGroup(user);
+  }
+}