RepositoriesIntegrationTests.java

package org.springframework.data.aerospike.repository;

import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;

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

public class RepositoriesIntegrationTests extends BaseRepositoriesIntegrationTests {

    @Autowired
    CompositeObjectRepository repository;

    @Test
    public void shouldSaveObjectWithPersistenceConstructorThatHasAllFields() throws Exception {
        CompositeObject expected = CompositeObject.builder()
                .id("composite-object-1")
                .simpleObject(SimpleObject.builder().property1("prop1").property2(555).build())
                .build();
        repository.save(expected);

        CompositeObject actual = repository.findOne(expected.getId());

        assertThat(actual).isEqualTo(expected);
    }

    @Test
    public void shouldDeleteObjectWithPersistenceConstructor() throws Exception {
        String id = nextId();
        CompositeObject expected = CompositeObject.builder()
                .id(id)
                .build();
        repository.save(expected);
        assertThat(repository.findOne(id)).isNotNull();

        repository.delete(expected);

        assertThat(repository.findOne(id)).isNull();
    }
}