org.neo4j.ogm.persistence.authentication.AuthenticatingDriverTest.java Source code

Introduction

Here is the source code for org.neo4j.ogm.persistence.authentication.AuthenticatingDriverTest.java

Source

/*
 * Copyright (c) 2002-2018 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This product is licensed to you under the Apache License, Version 2.0 (the "License").
 * You may not use this product except in compliance with the License.
 *
 * This product may include a number of subcomponents with
 * separate copyright notices and license terms. Your use of the source
 * code for these subcomponents is subject to the terms and
 *  conditions of the subcomponent's license, as noted in the LICENSE file.
 */

package org.neo4j.ogm.persistence.authentication;

import static org.assertj.core.api.Assertions.*;
import static org.junit.Assume.*;

import org.apache.http.client.HttpResponseException;
import org.junit.Before;
import org.junit.Test;
import org.neo4j.ogm.drivers.http.driver.HttpDriver;
import org.neo4j.ogm.session.Session;
import org.neo4j.ogm.session.SessionFactory;
import org.neo4j.ogm.testutil.MultiDriverTestClass;
import org.neo4j.ogm.transaction.Transaction;

/**
 * @author Vince Bickers
 * @author Luanne Misquitta
 */

public class AuthenticatingDriverTest extends MultiDriverTestClass {

    private Session session;

    @Before
    public void beforeMethod() {
        assumeTrue(getBaseConfiguration().build().getDriverClassName().equals(HttpDriver.class.getName()));
    }

    @Test
    public void testUnauthorizedDriver() {

        session = new SessionFactory(getBaseConfiguration().credentials("", "").build(), "dummy").openSession();

        try (Transaction tx = session.beginTransaction()) {
            tx.commit();
            fail("Driver should not have authenticated");
        } catch (Exception rpe) {
            Throwable cause = rpe.getCause();
            while (!(cause instanceof HttpResponseException)) {
                cause = cause.getCause();
            }
            assertThat(cause.getMessage().startsWith("Invalid username or password")).isTrue();
        }
    }

    @Test
    public void testAuthorizedDriver() {

        session = new SessionFactory(driver, "dummy").openSession();

        try (Transaction ignored = session.beginTransaction()) {
            assertThat(ignored).isNotNull();
        } catch (Exception rpe) {
            fail("'" + rpe.getLocalizedMessage() + "' was not expected here");
        }
    }

    /**
     * @see issue #35
     */
    @Test
    public void testInvalidCredentials() {

        session = new SessionFactory(getBaseConfiguration().credentials("neo4j", "invalid_password").build(),
                "dummy").openSession();

        try (Transaction tx = session.beginTransaction()) {
            fail("Driver should not have authenticated");
        } catch (Exception rpe) {
            Throwable cause = rpe.getCause();
            while (!(cause instanceof HttpResponseException)) {
                cause = cause.getCause();
            }
            assertThat(cause.getMessage()).isEqualTo("Invalid username or password.");
        }
    }
}