<reponame>pmattd/big-data-generator
from data_generator.data_line_generator import EnumeratedFieldGenerator, DataLineGenerator, DateTimeGenerator


def test_generate():
    generator1 = EnumeratedFieldGenerator("greeting", ["hello"])
    generator2 = EnumeratedFieldGenerator("greeting", ["goodbye"])
    line_generator = DataLineGenerator([generator1, generator2], ",")
    assert line_generator.generate() == "hello,goodbye"


def test_enumerated_field_generator():
    generator = EnumeratedFieldGenerator("greeting", ["hello"])
    assert generator.get_val() == "hello"


def test_timestamp_generator():
    generator = DateTimeGenerator("timestamp", "%Y-%m-%dT%H:%M:%SZ", -3000, 3000)
    print(generator.get_val())
