Advertisement
Guest User

Untitled

a guest
Dec 21st, 2018
136
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 7.78 KB | None | 0 0
  1. import DeliveryCompany.app.enumerate.RegisterStatus;
  2. import DeliveryCompany.app.enumerate.UserType;
  3. import DeliveryCompany.app.functionality.UserFunc;
  4. import DeliveryCompany.database.init.DatabaseInit;
  5. import DeliveryCompany.database.structure.Client;
  6. import DeliveryCompany.database.structure.Courier;
  7. import DeliveryCompany.database.structure.Email;
  8. import DeliveryCompany.database.structure.Storeman;
  9. import DeliveryCompany.database.structure.User;
  10. import java.security.NoSuchAlgorithmException;
  11. import org.hibernate.Query;
  12. import org.hibernate.Session;
  13. import org.hibernate.Transaction;
  14. import org.junit.Test;
  15. import org.junit.Assert;
  16. import org.junit.Before;
  17. import org.junit.BeforeClass;
  18. import org.junit.experimental.categories.Category;
  19. import org.junit.runner.RunWith;
  20. import org.junit.runners.Parameterized;
  21. import org.junit.runners.ParentRunner;
  22. import org.mockito.*;
  23. import static org.mockito.Matchers.any;
  24. import org.mockito.Mockito;
  25. import org.mockito.runners.MockitoJUnitRunner;
  26.  
  27. /**
  28.  *
  29.  * @author Paweł
  30.  */
  31. @Category(UserFunc.class)
  32. @RunWith(MockitoJUnitRunner.class)
  33. public class UserFuncTest {
  34.    
  35.    
  36.     //@Mock
  37.     Session session;
  38.     //@Mock
  39.     DatabaseInit database;
  40.    
  41.    
  42.  
  43.     public UserFuncTest() {
  44.     }
  45.    
  46.     @Before
  47.     public void setUp()
  48.     {
  49.  
  50.         session = Mockito.mock(Session.class);
  51.        
  52.         database = Mockito.spy(new DatabaseInit());
  53.         database.setSession(session);
  54.        
  55.         Mockito.when(session.beginTransaction()).thenReturn(null);
  56.         Mockito.when(session.getTransaction()).thenReturn(Mockito.mock(Transaction.class));
  57.        
  58.     }
  59.     /*
  60.     @Test
  61.     public void loginTestNotNull() throws NoSuchAlgorithmException
  62.     {
  63.         User user = new User();
  64.         user.setUsername("pawel");
  65.         user.setPassword("pass");
  66.         user.setUserType(UserType.Client.toString());
  67.        
  68.         Query q = Mockito.mock(Query.class);
  69.        
  70.         Mockito.when(session.createQuery(any(String.class))).thenReturn(q);
  71.         Mockito.when(q.uniqueResult()).thenReturn(user);
  72.         //Mockito.when(session.beginTransaction()).thenReturn(null);
  73.         //Mockito.when(session.getTransaction()).thenReturn(Mockito.mock(Transaction.class));
  74.        
  75.        
  76.         UserFunc userFunc = new UserFunc();
  77.        
  78.         Assert.assertNotNull(userFunc.Login("pawel", "pass"));
  79.        
  80.     }
  81. */
  82.    
  83.     @Test
  84.     public void loginTestNull() throws NoSuchAlgorithmException
  85.     {
  86.        
  87.         Query q = Mockito.mock(Query.class);
  88.        
  89.         Mockito.when(session.createQuery(any(String.class))).thenReturn(q);
  90.         Mockito.when(q.uniqueResult()).thenReturn(null);
  91.         //Mockito.when(session.beginTransaction()).thenReturn(null);
  92.         //Mockito.when(session.getTransaction()).thenReturn(Mockito.mock(Transaction.class));
  93.        
  94.        
  95.         UserFunc userFunc = new UserFunc();
  96.        
  97.         Assert.assertNull(userFunc.Login("pawel", "pass"));
  98.        
  99.     }
  100.     /*
  101.     @Test
  102.     public void registryClientTestEmailExists()
  103.     {
  104.  
  105.         Email email = new Email();
  106.         email.setEmail("email");
  107.  
  108.         Query queryEmail = Mockito.mock(Query.class);
  109.        
  110.         Mockito.when(session.createQuery("FROM Email WHERE email = :e")).thenReturn(queryEmail);
  111.         Mockito.when(queryEmail.uniqueResult()).thenReturn(email);
  112.        
  113.         Query queryUser = Mockito.mock(Query.class);
  114.        
  115.         Mockito.when(session.createQuery("FROM User WHERE Username = :u")).thenReturn(queryUser);
  116.         Mockito.when(queryUser.uniqueResult()).thenReturn(null);
  117.        
  118.         UserFunc userFunc = new UserFunc();
  119.        
  120.        Assert.assertEquals(userFunc.Registry("pawel", "pass", "email@email.com", UserType.Client), RegisterStatus.EmailExists);
  121.        
  122.     }
  123.    
  124.     @Test
  125.     public void registryClientTestUsernameExists()
  126.     {
  127.  
  128.  
  129.         Query queryEmail = Mockito.mock(Query.class);
  130.        
  131.         Mockito.when(session.createQuery("FROM Email WHERE email = :e")).thenReturn(queryEmail);
  132.         Mockito.when(queryEmail.uniqueResult()).thenReturn(null);
  133.        
  134.        
  135.         User user = new User();
  136.         user.setUsername("pawel");
  137.         user.setPassword("pass");
  138.         user.setUserType(UserType.Client.toString());
  139.        
  140.         Query queryUser = Mockito.mock(Query.class);
  141.        
  142.         Mockito.when(session.createQuery("FROM User WHERE Username = :u")).thenReturn(queryUser);
  143.         Mockito.when(queryUser.uniqueResult()).thenReturn(user);
  144.        
  145.         UserFunc userFunc = new UserFunc();
  146.        
  147.         Assert.assertEquals(userFunc.Registry("pawel", "pass", "email@email.com", UserType.Client), RegisterStatus.UsernameExists);
  148.        
  149.     }
  150.    
  151.     @Test
  152.     public void registryClientTestSuccess()
  153.     {
  154.  
  155.  
  156.         Query queryEmail = Mockito.mock(Query.class);
  157.        
  158.         Mockito.when(session.createQuery("FROM Email WHERE email = :e")).thenReturn(queryEmail);
  159.         Mockito.when(queryEmail.uniqueResult()).thenReturn(null);
  160.        
  161.        
  162.         Query queryUser = Mockito.mock(Query.class);
  163.        
  164.         Mockito.when(session.createQuery("FROM User WHERE Username = :u")).thenReturn(queryUser);
  165.         Mockito.when(queryUser.uniqueResult()).thenReturn(null);
  166.        
  167.         UserFunc userFunc = new UserFunc();
  168.        
  169.         Assert.assertEquals(userFunc.Registry("pawel", "pass", "email@email.com", UserType.Client), RegisterStatus.Success);
  170.        
  171.     }
  172.    
  173.     @Test
  174.     public void getMembershipTestReturnClient()
  175.     {
  176.         Client client = new Client();
  177.        
  178.         Query query = Mockito.mock(Query.class);
  179.        
  180.         Mockito.when(session.createQuery("FROM Client WHERE user = :u")).thenReturn(query);
  181.         Mockito.when(query.uniqueResult()).thenReturn(client);
  182.        
  183.         User user = new User();
  184.         user.setUsername("pawel");
  185.         user.setPassword("pass");
  186.         user.setUserType(UserType.Client.toString());
  187.        
  188.         UserFunc userFunc = new UserFunc();
  189.        
  190.         //Assert.assertEquals(userFunc.getMembership(user), );
  191.         Assert.assertNotNull(userFunc.getMembership(user));
  192.         Assert.assertTrue(userFunc.getMembership(user) instanceof Client);
  193.     }
  194.    
  195.     @Test
  196.     public void getMembershipTestReturnCourier()
  197.     {
  198.         Courier courier = new Courier();
  199.        
  200.         Query query = Mockito.mock(Query.class);
  201.        
  202.         Mockito.when(session.createQuery("FROM Courier WHERE user = :u")).thenReturn(query);
  203.         Mockito.when(query.uniqueResult()).thenReturn(courier);
  204.        
  205.         User user = new User();
  206.         user.setUsername("pawel");
  207.         user.setPassword("pass");
  208.         user.setUserType(UserType.Courier.toString());
  209.        
  210.         UserFunc userFunc = new UserFunc();
  211.        
  212.         //Assert.assertEquals(userFunc.getMembership(user), );
  213.         Assert.assertNotNull(userFunc.getMembership(user));
  214.         Assert.assertTrue(userFunc.getMembership(user) instanceof Courier);
  215.     }
  216.    
  217.     @Test
  218.     public void getMembershipTestReturnStoreman()
  219.     {
  220.         Storeman storeman = new Storeman();
  221.        
  222.         Query query = Mockito.mock(Query.class);
  223.        
  224.         Mockito.when(session.createQuery("FROM Storeman WHERE user = :u")).thenReturn(query);
  225.         Mockito.when(query.uniqueResult()).thenReturn(storeman);
  226.        
  227.         User user = new User();
  228.         user.setUsername("pawel");
  229.         user.setPassword("pass");
  230.         user.setUserType(UserType.Storeman.toString());
  231.        
  232.         UserFunc userFunc = new UserFunc();
  233.        
  234.         //Assert.assertEquals(userFunc.getMembership(user), );
  235.         Assert.assertNotNull(userFunc.getMembership(user));
  236.         Assert.assertTrue(userFunc.getMembership(user) instanceof Storeman);
  237.     }
  238. */
  239. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement