@Category(UserFunc.class)
@RunWith(MockitoJUnitRunner.class)
public class UserFuncTest {


    //@Mock
    Session session;
    //@Mock
    DatabaseInit database;



    public UserFuncTest() {
    }

    @Before
    public void setUp()
    {

        session = Mockito.mock(Session.class);

        database = Mockito.spy(new DatabaseInit());
        database.setSession(session);

        Mockito.when(session.beginTransaction()).thenReturn(null);
        Mockito.when(session.getTransaction()).thenReturn(Mockito.mock(Transaction.class));

    }
    /*
    @Test
    public void loginTestNotNull() throws NoSuchAlgorithmException
    {
        User user = new User();
        user.setUsername("pawel");
        user.setPassword("pass");
        user.setUserType(UserType.Client.toString());

        Query q = Mockito.mock(Query.class);

        Mockito.when(session.createQuery(any(String.class))).thenReturn(q);
        Mockito.when(q.uniqueResult()).thenReturn(user);
        //Mockito.when(session.beginTransaction()).thenReturn(null);
        //Mockito.when(session.getTransaction()).thenReturn(Mockito.mock(Transaction.class));


        UserFunc userFunc = new UserFunc();

        Assert.assertNotNull(userFunc.Login("pawel", "pass"));

    }
*/

    @Test
    public void loginTestNull() throws NoSuchAlgorithmException
    {

        Query q = Mockito.mock(Query.class);

        Mockito.when(session.createQuery(any(String.class))).thenReturn(q);
        Mockito.when(q.uniqueResult()).thenReturn(null);
        //Mockito.when(session.beginTransaction()).thenReturn(null);
        //Mockito.when(session.getTransaction()).thenReturn(Mockito.mock(Transaction.class));


        UserFunc userFunc = new UserFunc();

        Assert.assertNull(userFunc.Login("pawel", "pass"));

    }
    /*
    @Test
    public void registryClientTestEmailExists()
    {

        Email email = new Email();
        email.setEmail("email");

        Query queryEmail = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM Email WHERE email = :e")).thenReturn(queryEmail);
        Mockito.when(queryEmail.uniqueResult()).thenReturn(email);

        Query queryUser = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM User WHERE Username = :u")).thenReturn(queryUser);
        Mockito.when(queryUser.uniqueResult()).thenReturn(null);

        UserFunc userFunc = new UserFunc();

       Assert.assertEquals(userFunc.Registry("pawel", "pass", "email@email.com", UserType.Client), RegisterStatus.EmailExists);

    }

    @Test
    public void registryClientTestUsernameExists()
    {


        Query queryEmail = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM Email WHERE email = :e")).thenReturn(queryEmail);
        Mockito.when(queryEmail.uniqueResult()).thenReturn(null);


        User user = new User();
        user.setUsername("pawel");
        user.setPassword("pass");
        user.setUserType(UserType.Client.toString());

        Query queryUser = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM User WHERE Username = :u")).thenReturn(queryUser);
        Mockito.when(queryUser.uniqueResult()).thenReturn(user);

        UserFunc userFunc = new UserFunc();

        Assert.assertEquals(userFunc.Registry("pawel", "pass", "email@email.com", UserType.Client), RegisterStatus.UsernameExists);

    }

    @Test
    public void registryClientTestSuccess()
    {


        Query queryEmail = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM Email WHERE email = :e")).thenReturn(queryEmail);
        Mockito.when(queryEmail.uniqueResult()).thenReturn(null);


        Query queryUser = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM User WHERE Username = :u")).thenReturn(queryUser);
        Mockito.when(queryUser.uniqueResult()).thenReturn(null);

        UserFunc userFunc = new UserFunc();

        Assert.assertEquals(userFunc.Registry("pawel", "pass", "email@email.com", UserType.Client), RegisterStatus.Success);

    }

    @Test
    public void getMembershipTestReturnClient()
    {
        Client client = new Client();

        Query query = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM Client WHERE user = :u")).thenReturn(query);
        Mockito.when(query.uniqueResult()).thenReturn(client);

        User user = new User();
        user.setUsername("pawel");
        user.setPassword("pass");
        user.setUserType(UserType.Client.toString());

        UserFunc userFunc = new UserFunc();

        //Assert.assertEquals(userFunc.getMembership(user), );
        Assert.assertNotNull(userFunc.getMembership(user));
        Assert.assertTrue(userFunc.getMembership(user) instanceof Client);
    }

    @Test
    public void getMembershipTestReturnCourier()
    {
        Courier courier = new Courier();

        Query query = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM Courier WHERE user = :u")).thenReturn(query);
        Mockito.when(query.uniqueResult()).thenReturn(courier);

        User user = new User();
        user.setUsername("pawel");
        user.setPassword("pass");
        user.setUserType(UserType.Courier.toString());

        UserFunc userFunc = new UserFunc();

        //Assert.assertEquals(userFunc.getMembership(user), );
        Assert.assertNotNull(userFunc.getMembership(user));
        Assert.assertTrue(userFunc.getMembership(user) instanceof Courier);
    }

    @Test
    public void getMembershipTestReturnStoreman()
    {
        Storeman storeman = new Storeman();

        Query query = Mockito.mock(Query.class);

        Mockito.when(session.createQuery("FROM Storeman WHERE user = :u")).thenReturn(query);
        Mockito.when(query.uniqueResult()).thenReturn(storeman);

        User user = new User();
        user.setUsername("pawel");
        user.setPassword("pass");
        user.setUserType(UserType.Storeman.toString());

        UserFunc userFunc = new UserFunc();

        //Assert.assertEquals(userFunc.getMembership(user), );
        Assert.assertNotNull(userFunc.getMembership(user));
        Assert.assertTrue(userFunc.getMembership(user) instanceof Storeman);
    }
*/
}