PotentialMatchRepositoryTest.java

package com.rapidftr.repository;

import com.rapidftr.CustomTestRunner;
import com.rapidftr.database.DatabaseSession;
import com.rapidftr.database.ShadowSQLiteHelper;
import com.rapidftr.model.Child;
import com.rapidftr.model.Enquiry;
import com.rapidftr.model.PotentialMatch;
import org.json.JSONException;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.io.IOException;
import java.sql.SQLException;
import java.util.List;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.notNullValue;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.core.IsNull.nullValue;
import static org.junit.Assert.assertNull;

@RunWith(CustomTestRunner.class)
public class PotentialMatchRepositoryTest {

    public DatabaseSession session;
    public PotentialMatchRepository repository;

    @Before
    public void setupSession() throws IOException {
        session = new ShadowSQLiteHelper("test_database").getSession();
        repository = new PotentialMatchRepository("user1", session);
    }

    @Test
    public void shouldReturnFalseWhenAChildWithTheGivenIdDoesNotExistsInTheDatabase() throws JSONException, SQLException {
        assertThat(repository.exists("1234"), is(false));
    }

    @Test
    public void shouldReturnTrueWhenAChildWithTheGivenIdExistsInTheDatabase() throws JSONException, SQLException {
        PotentialMatch match = new PotentialMatch("enquiry", "child", "unique_identifier");
        repository.createOrUpdate(match);
        assertThat(repository.exists("unique_identifier"), is(true));
    }

    @Test
    public void shouldReturnEmptyListIfEnquiryDoesNotHaveInternalId() throws JSONException, SQLException {
        String enquiryJSON = "{\"name\":\"foo bar\", \"nationality\":\"ugandan\"}";
        Enquiry enquiry = new Enquiry(enquiryJSON);
        List<PotentialMatch> matches = repository.getPotentialMatchesFor(enquiry);
        assertThat(matches.size(), is(0));
    }

    @Test
    public void shouldReturnPotentialMatchesByEnquiry() throws JSONException, SQLException {
        String enquiryJSON = "{\"_id\":\"enquiry_id\", \"name\":\"foo bar\", \"nationality\":\"ugandan\"}";
        Enquiry enquiry = new Enquiry(enquiryJSON);
        PotentialMatch potentialMatch = new PotentialMatch("enquiry_id", "child_id", "unique_id_2");
        repository.createOrUpdate(potentialMatch);
        repository.createOrUpdate(new PotentialMatch("not_matching", "child_id", "unique_id_1"));

        List<PotentialMatch> matches = repository.getPotentialMatchesFor(enquiry);

        assertThat(matches.size(), is(1));
        assertThat(matches.get(0).getEnquiryId(), is("enquiry_id"));
    }

    @Test
    public void shouldReturnEmptyListIfChildDoesNotHaveInternalId() throws JSONException, SQLException {
        String childJSON = "{\"name\":\"foo bar\", \"nationality\":\"ugandan\"}";
        Child child = new Child(childJSON);
        List<PotentialMatch> matches = repository.getPotentialMatchesFor(child);
        assertThat(matches.size(), is(0));
    }

    @Test
    public void shouldReturnPotentialMatchesByChild() throws JSONException, SQLException {
        String childJSON = "{\"_id\":\"child_id\", \"name\":\"foo bar\", \"nationality\":\"ugandan\"}";
        Child child = new Child(childJSON);
        PotentialMatch potentialMatch = new PotentialMatch("enquiry_id", "child_id", "unique_id_1");
        repository.createOrUpdate(potentialMatch);
        repository.createOrUpdate(new PotentialMatch("enquiry_id", "not_matching", "unique_id_2"));

        List<PotentialMatch> matches = repository.getPotentialMatchesFor(child);

        assertThat(matches.size(), is(1));
        assertThat(matches.get(0).getChildId(), is("child_id"));
    }

    @Test
    public void shouldUpdateExistingRecords() throws JSONException, SQLException {
        PotentialMatch potentialMatch = new PotentialMatch("enquiry_id", "child_id", "unique_id_1");
        repository.createOrUpdate(potentialMatch);
        repository.createOrUpdateWithoutHistory(new PotentialMatch("enquiry_id", "child_id", "unique_id_1", true));
        PotentialMatch savedPotentialMatch = repository.get("unique_id_1");
        assert (savedPotentialMatch.isConfirmed());
    }

    @Test
    public void shouldGetExistingRecords() throws JSONException, SQLException {
        PotentialMatch potentialMatch = new PotentialMatch("enquiry_id", "child_id", "unique_id_1");
        repository.createOrUpdate(potentialMatch);
        assertThat(repository.get("unique_id_1"), is(notNullValue()));
    }

    @Test
    public void shouldReturnNullIfPotentialMatchDoesNotExist() throws JSONException, SQLException {
        assertThat(repository.get("unique_id_1"), is(nullValue()));
    }

    @Test
    public void shouldDeletePotentialMatch() throws JSONException, SQLException {
        PotentialMatch potentialMatch = new PotentialMatch("enquiry_id", "child_id", "unique_id_1");
        repository.createOrUpdate(potentialMatch);
        assertThat(repository.get("unique_id_1"), is(notNullValue()));

        repository.delete(potentialMatch);
        assertNull(repository.get("unique_id_1"));
    }

    @Test
    public void shouldDeletePotentialMatchMarkedAsDeleted() throws JSONException, SQLException {
        PotentialMatch potentialMatch = new PotentialMatch("enquiry_id", "child_id", "unique_id_1");
        repository.createOrUpdate(potentialMatch);
        assertThat(repository.get("unique_id_1"), is(notNullValue()));

        potentialMatch.put(PotentialMatch.STATUS_FIELD, PotentialMatch.PotentialMatchStatus.DELETED.name());
        repository.createOrUpdateWithoutHistory(potentialMatch);

        assertNull(repository.get("unique_id_1"));
    }
}