src/test/java/org/apache/cassandra/sidecar/snapshots/SnapshotDirectoryTest.java

package org.apache.cassandra.sidecar.snapshots;

import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.ValueSource;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;

class SnapshotDirectoryTest
{

    @ParameterizedTest
    @ValueSource(strings = { "not-valid", "/two-levels/not-valid", "three/levels/not-valid", "four/levels/not/valid" })
    void failsOnInvalidLengthDirectory()
    {
        assertThatIllegalArgumentException()
        .isThrownBy(() -> SnapshotDirectory.of("not-valid"))
        .withMessageContaining("Invalid snapshotDirectory. Expected at least 5 parts but found");
    }

    @Test
    void failsOnInvalidDirectory()
    {
        assertThatIllegalArgumentException()
        .isThrownBy(() -> SnapshotDirectory.of("/cassandra/data/ks1/tbl2/sneaky/test-snapshot"))
        .withMessage("Invalid snapshotDirectory. The expected directory structure is " +
                     "'/<data_dir>/<ks>/<table>/snapshots/<snapshot_name>'");
    }

    @Test
    void testValidDirectory1()
    {
        String snapshotDirectory = "/cassandra/data/ks1/tbl2/snapshots/test-snapshot";
        SnapshotDirectory directory = SnapshotDirectory.of(snapshotDirectory);
        assertThat(directory.dataDirectory).isEqualTo("/cassandra/data");
        assertThat(directory.keyspace).isEqualTo("ks1");
        assertThat(directory.tableName).isEqualTo("tbl2");
        assertThat(directory.snapshotName).isEqualTo("test-snapshot");
    }

    @Test
    void testValidDirectory2()
    {
        String snapshotDirectory = "/cassandra/data/ks1/tbl2/SNAPSHOTS/test-snapshot";
        SnapshotDirectory directory = SnapshotDirectory.of(snapshotDirectory);
        assertThat(directory.dataDirectory).isEqualTo("/cassandra/data");
        assertThat(directory.keyspace).isEqualTo("ks1");
        assertThat(directory.tableName).isEqualTo("tbl2");
        assertThat(directory.snapshotName).isEqualTo("test-snapshot");
    }

    @Test
    void testValidDirectory3()
    {
        String snapshotDirectory = "/datadir/inventory/shipping/snapshots/2022-07-23";
        SnapshotDirectory directory = SnapshotDirectory.of(snapshotDirectory);
        assertThat(directory.dataDirectory).isEqualTo("/datadir");
        assertThat(directory.keyspace).isEqualTo("inventory");
        assertThat(directory.tableName).isEqualTo("shipping");
        assertThat(directory.snapshotName).isEqualTo("2022-07-23");
    }

    @Test
    void testValidDirectory4()
    {
        String snapshotDirectory = "/cassandra/disk1/data/inventory/shipping/snapshots/2022-07-23/";
        SnapshotDirectory directory = SnapshotDirectory.of(snapshotDirectory);
        assertThat(directory.dataDirectory).isEqualTo("/cassandra/disk1/data");
        assertThat(directory.keyspace).isEqualTo("inventory");
        assertThat(directory.tableName).isEqualTo("shipping");
        assertThat(directory.snapshotName).isEqualTo("2022-07-23");
    }
}