1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
|
import os
import sys
from stetl.etl import ETL
from stetl.packet import Packet
from stetl.inputs.fileinput import StringFileInput
from tests.stetl_test_case import StetlTestCase
class StringFileInputTest(StetlTestCase):
"""Unit tests for StringFileInput"""
def setUp(self):
super(StringFileInputTest, self).setUp()
# Initialize Stetl
curr_dir = os.path.dirname(os.path.realpath(__file__))
cfg_dict = {'config_file': os.path.join(curr_dir, 'configs/stringfileinput.cfg')}
self.etl = ETL(cfg_dict)
def test_class(self):
chain = StetlTestCase.get_chain(self.etl)
section = StetlTestCase.get_section(chain)
class_name = self.etl.configdict.get(section, 'class')
self.assertEqual('stetl.inputs.fileinput.StringFileInput', class_name)
def test_instance(self):
chain = StetlTestCase.get_chain(self.etl)
self.assertTrue(isinstance(chain.first_comp, StringFileInput))
def test_execute(self):
# Read content of input file
chain = StetlTestCase.get_chain(self.etl)
section = StetlTestCase.get_section(chain)
fn = self.etl.configdict.get(section, 'file_path')
with open(fn, 'r') as f:
contents = f.read()
# Invoke first component of chain
chain.first_comp.do_init()
packet = Packet()
packet.init()
packet.component = chain.first_comp
chain.first_comp.before_invoke(packet)
packet = chain.first_comp.invoke(packet)
self.assertEqual(packet.data, contents)
def test_format_args(self):
chain = StetlTestCase.get_chain(self.etl, 1)
chain.first_comp.do_init()
packet = Packet()
packet.init()
packet.component = chain.first_comp
chain.first_comp.before_invoke(packet)
packet = chain.first_comp.invoke(packet)
self.assertEqual(packet.data, 'Hello NLExtract!')
|