EvictExpirableRecordsPersistenceTest.java

/*
 * Copyright 2016 Victor Albertos
 *
 * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.rx_cache2.internal.cache;

import io.reactivex.observers.TestObserver;
import io.rx_cache2.internal.Locale;
import io.rx_cache2.internal.Memory;
import io.rx_cache2.internal.Mock;
import io.rx_cache2.internal.Record;
import io.rx_cache2.internal.cache.memory.ReferenceMapMemory;
import io.rx_cache2.internal.common.BaseTest;
import java.util.ArrayList;
import java.util.List;
import org.junit.Test;
import org.junit.experimental.theories.DataPoint;
import org.junit.experimental.theories.Theories;
import org.junit.experimental.theories.Theory;
import org.junit.runner.RunWith;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

@RunWith(Theories.class)
public class EvictExpirableRecordsPersistenceTest extends BaseTest {
    private io.rx_cache2.internal.cache.EvictExpirableRecordsPersistence
        evictExpirableRecordsPersistenceUT;
    private Memory memory;

    @Override public void setUp() {
        super.setUp();
        memory = new ReferenceMapMemory();
    }

    @Test public void When_Not_Reached_Memory_Threshold_Not_Emit() {
        evictExpirableRecordsPersistenceUT = new io.rx_cache2.internal.cache.EvictExpirableRecordsPersistence(memory, disk, 10, null);

        populate(true);
        assertThat(disk.allKeys().size(), is(100));

        TestObserver testObserver = evictExpirableRecordsPersistenceUT.startTaskIfNeeded(false).test();
        testObserver.awaitTerminalEvent();
        testObserver
            .assertNoErrors()
            .assertNoValues();
    }

    @DataPoint public static Integer _3_MB = 3;
    @DataPoint public static Integer _5_MB = 5;
    @DataPoint public static Integer _7_MB = 7;
    @Theory @Test public void When_Reached_Memory_Threshold_Perform_Task(int maxMgPersistenceCache) {
        evictExpirableRecordsPersistenceUT = new io.rx_cache2.internal.cache.EvictExpirableRecordsPersistence(memory, disk, maxMgPersistenceCache, null);

        populate(true);
        assertThat(disk.allKeys().size(), is(mocksCount()));

        TestObserver testObserver = evictExpirableRecordsPersistenceUT.startTaskIfNeeded(false).test();
        testObserver.awaitTerminalEvent();
        testObserver.assertNoErrors();

        int expectedStoredMB = (int) (maxMgPersistenceCache * io.rx_cache2.internal.cache.EvictExpirableRecordsPersistence.PERCENTAGE_MEMORY_STORED_TO_STOP);
        assertThat(expectedStoredMB, is(disk.storedMB()));
    }

    @Test public void When_Reached_Memory_Threshold_But_Not_Expirable_Records_Do_Not_Evict() {
        int maxMgPersistenceCache = 5;

        evictExpirableRecordsPersistenceUT = new io.rx_cache2.internal.cache.EvictExpirableRecordsPersistence(memory, disk, maxMgPersistenceCache, null);

        populate(false);
        assertThat(disk.allKeys().size(), is(mocksCount()));

        TestObserver testObserver = evictExpirableRecordsPersistenceUT.startTaskIfNeeded(false).test();
        testObserver.awaitTerminalEvent();
        testObserver.assertNoErrors();
        //testSubscriber.assertNoValues();

        assertThat(sizeMbDataPopulated(), is(disk.storedMB()));

        //after first time does not start process again, just return warning message
        testObserver = new TestObserver();
        evictExpirableRecordsPersistenceUT.startTaskIfNeeded(false).subscribe(testObserver);
        testObserver.awaitTerminalEvent();
        testObserver.assertNoErrors();
        testObserver.assertValue(Locale.RECORD_CAN_NOT_BE_EVICTED_BECAUSE_NO_ONE_IS_EXPIRABLE);
    }

    //7 mb
    private void populate(boolean expirable) {
        for (int i = 0; i < mocksCount(); i++) {
            List<Mock> mocks = new ArrayList(mocksCount());

            for (int z = 0; z < mocksCount(); z++) {
                Mock mock = new Mock("Contrary to popular belief, Lorem Ipsum is not simply random text. It has roots in a piece of classical Latin literature from 45 BC," +
                        "making it over 2000 years old.Contrary to popular belief, Lorem Ipsum is not simply random text. It has roots in a piece of classical Latin literature from 45 BC, " +
                        "making it over 2000 years old. Contrary to popular belief, Lorem Ipsum is not simply random text. It has roots in a piece of classical Latin literature from 45 BC, " +
                        "making it over 2000 years old. Contrary to popular belief, Lorem Ipsum is not simply random text. It has roots in a piece of classical Latin literature from 45 BC.");
                mocks.add(mock);
            }

            Record<List<Mock>> record = new Record<>(mocks, expirable, 1l);
            disk.saveRecord(String.valueOf(i), record, false, null);
        }
    }

    private int mocksCount() {
        return 100;
    }

    private int sizeMbDataPopulated() {
        return 7;
    }
}