package com.orchard;

import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;

import java.util.stream.Stream;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import static org.junit.jupiter.params.provider.Arguments.arguments;


public class SplitStringTests {

    final SplitStringUdtf splitString = new SplitStringUdtf();

    static Stream<Arguments> stringAndStringArrayProvider() {
        return Stream.of(
                arguments("apple\nbanana\n", new String[]{"apple", "banana"}),
                arguments("nonewlines", new String[]{"nonewlines"}),
                arguments("apple\\nbanana", new String[]{"apple\\nbanana"})
        );
    }

    @ParameterizedTest
    @MethodSource("stringAndStringArrayProvider")
    public void shouldSplitString(String source, String[] expectedResult) {
        final String[] actualResult = splitString.split_string(source).toArray(new String[0]);
        assertArrayEquals(expectedResult, actualResult);
    }

    @Test
    public void shouldSplitStringWithCustomDelimiter() {
        final String source = "hello;world;";
        final String delimiter = ";";
        final String[] expectedResult = {"hello", "world"};
        final String[] actualResult = splitString.split_string(source, delimiter).toArray(new String[0]);
        assertArrayEquals(expectedResult, actualResult);
    }
}