PrivateRowsQueryExecutorTest.java

package it.inaf.ia2.vollt;

import java.sql.Statement;
import java.util.Arrays;
import org.junit.jupiter.api.Test;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;

public class PrivateRowsQueryExecutorTest {

    @Test
    public void executeQueryTest() throws Exception {

        PrivateRowsQueryExecutor queryExecutor = new PrivateRowsQueryExecutor();

        CustomJobOwner user = new CustomJobOwner("user", Arrays.asList("group1", "group2"));

        Statement mockedStatement = mock(Statement.class);
        queryExecutor.executeQuery(mockedStatement, "", user);

        verify(mockedStatement).execute(eq("SET my.tap_groups = \"{group1,group2}\""));

        user = new CustomJobOwner("anonymous", Arrays.asList());
        queryExecutor.executeQuery(mockedStatement, "", user);

        verify(mockedStatement).execute(eq("SET my.tap_groups = \"{}\""));
    }
}