eionet.webq.dao.KnownHostsTest.java Source code

Introduction

Here is the source code for eionet.webq.dao.KnownHostsTest.java

Source

/*
 * The contents of this file are subject to the Mozilla Public
 * License Version 1.1 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of
 * the License at http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS
 * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * rights and limitations under the License.
 *
 * The Original Code is Web Questionnaires 2
 *
 * The Initial Owner of the Original Code is European Environment
 * Agency. Portions created by TripleDev are Copyright
 * (C) European Environment Agency.  All Rights Reserved.
 *
 * Contributor(s):
 *        Anton Dmitrijev
 */
package eionet.webq.dao;

import configuration.ApplicationTestContextWithMockSession;
import eionet.webq.dao.orm.KnownHost;
import eionet.webq.dto.KnownHostAuthenticationMethod;
import org.hibernate.FlushMode;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.transaction.annotation.Transactional;

import java.util.Collection;

import static org.hamcrest.core.IsEqual.equalTo;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;

/**
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = { ApplicationTestContextWithMockSession.class })
@Transactional
public class KnownHostsTest {

    @Autowired
    private KnownHosts knownHosts;
    @Autowired
    private SessionFactory sessionFactory;

    @Before
    public void before() throws Exception {
        sessionFactory.getCurrentSession().setFlushMode(FlushMode.ALWAYS);
    }

    @Test
    public void shouldSaveKnownHost() throws Exception {
        KnownHost knownHost = createKnownHost();

        int id = knownHosts.save(knownHost);
        assertTrue(id > 0);
    }

    @Test
    public void shouldReadSavedDataAfterSave() throws Exception {
        KnownHost knownHost = createKnownHost();
        int id = save(knownHost);

        KnownHost knownHostFromStorage = knownHosts.findById(id);

        assertEquals(knownHost, knownHostFromStorage);
    }

    @Test
    public void shouldFetchAllKnownHosts() throws Exception {
        KnownHost host1 = createKnownHost();
        KnownHost host2 = createKnownHost();
        host2.setHostURL("http://other.host.url");

        knownHosts.save(host1);
        knownHosts.save(host2);

        Collection<KnownHost> hosts = knownHosts.findAll();

        assertThat(hosts.size(), equalTo(2));
    }

    @Test
    public void allowsToUpdateKnownHost() throws Exception {
        KnownHost example = createKnownHost();
        save(example);

        example.setTicket("new_" + example.getTicket());
        example.setKey("new_" + example.getKey());
        example.setAuthenticationMethod(KnownHostAuthenticationMethod.BASIC);
        example.setHostURL(example.getHostURL() + "/new");
        example.setHostName("new_" + example.getHostName());

        update(example);

        KnownHost updated = knownHosts.findById(example.getId());
        assertEquals(example, updated);
    }

    @Test
    public void allowsKnownHostRemovalById() throws Exception {
        KnownHost knownHost = createKnownHost();
        save(knownHost);
        assertThat(knownHosts.findAll().size(), equalTo(1));

        knownHosts.remove(knownHost.getId());

        assertThat(knownHosts.findAll().size(), equalTo(0));
    }

    private void assertEquals(KnownHost knownHost, KnownHost knownHostFromStorage) {
        assertThat(knownHostFromStorage.getAuthenticationMethod(), equalTo(knownHost.getAuthenticationMethod()));
        assertThat(knownHostFromStorage.getHostName(), equalTo(knownHost.getHostName()));
        assertThat(knownHostFromStorage.getHostURL(), equalTo(knownHost.getHostURL()));
        assertThat(knownHostFromStorage.getKey(), equalTo(knownHost.getKey()));
        assertThat(knownHostFromStorage.getTicket(), equalTo(knownHost.getTicket()));
    }

    private void update(KnownHost example) {
        knownHosts.update(example);
        evict(example);
    }

    private int save(KnownHost knownHost) {
        int id = knownHosts.save(knownHost);
        evict(knownHost);
        return id;
    }

    private void evict(KnownHost knownHost) {
        Session currentSession = sessionFactory.getCurrentSession();
        currentSession.flush();
        currentSession.evict(knownHost);
    }

    private KnownHost createKnownHost() {
        KnownHost knownHost = new KnownHost();
        knownHost.setHostURL("http://host.url");
        knownHost.setAuthenticationMethod(KnownHostAuthenticationMethod.REQUEST_PARAMETER);
        knownHost.setHostName("Host name");
        knownHost.setKey("api-key");
        knownHost.setTicket("api-ticket");
        return knownHost;
    }
}