All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.trino.testing.ProcedureTester Maven / Gradle / Ivy

/*
 * 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.trino.testing;

import java.util.List;

import static com.google.common.base.Preconditions.checkState;
import static java.util.Arrays.asList;
import static java.util.Collections.unmodifiableList;
import static java.util.Objects.requireNonNull;

public class ProcedureTester
{
    private boolean called;
    private String name;
    private List arguments;

    public synchronized void reset()
    {
        called = false;
    }

    public void recordCalled(String name, Object... arguments)
    {
        recordCalled(name, asList(arguments));
    }

    public synchronized void recordCalled(String name, List arguments)
    {
        checkState(!called, "already called");
        called = true;
        this.name = requireNonNull(name, "name is null");
        this.arguments = unmodifiableList(requireNonNull(arguments, "arguments is null"));
    }

    public synchronized boolean wasCalled()
    {
        return called;
    }

    public synchronized String getCalledName()
    {
        checkState(called, "not called");
        return name;
    }

    public synchronized List getCalledArguments()
    {
        checkState(called, "not called");
        return arguments;
    }
}