You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
3.4 KiB
112 lines
3.4 KiB
from os.path import exists
|
|
import unittest
|
|
|
|
from .context import ptaimport
|
|
from ptaimport import data_importers as imp
|
|
|
|
class TestDataImporters( unittest.TestCase ):
|
|
"""
|
|
Tests the data_importers modules functions
|
|
"""
|
|
|
|
def setUp( self ):
|
|
# nothing provided
|
|
self.config_none = {
|
|
}
|
|
|
|
# name provided, but not import
|
|
self.config_noImport = {
|
|
"name": "test importer"
|
|
}
|
|
|
|
# import provided but no name
|
|
self.config_noName = {
|
|
"import": "mock"
|
|
}
|
|
|
|
# wrong import provided
|
|
self.config_wrong = {
|
|
"name": "test importer",
|
|
"import": "unknown"
|
|
}
|
|
|
|
# correct
|
|
self.config_mock = {
|
|
"name": "test importer",
|
|
"import": "mock"
|
|
}
|
|
|
|
|
|
def test_verify( self ):
|
|
"""
|
|
Tests the data_importers.verify() function
|
|
"""
|
|
(ok, err) = imp.verify( self.config_none )
|
|
self.assertFalse(ok)
|
|
self.assertNotEqual( len(err), 0 )
|
|
|
|
(ok, err) = imp.verify( self.config_noImport )
|
|
self.assertFalse(ok)
|
|
self.assertNotEqual( len(err), 0 )
|
|
|
|
(ok, err) = imp.verify( self.config_noName )
|
|
self.assertFalse(ok)
|
|
self.assertNotEqual( len(err), 0 )
|
|
|
|
(ok, err) = imp.verify( self.config_wrong )
|
|
self.assertFalse(ok)
|
|
self.assertNotEqual( len(err), 0 )
|
|
|
|
(ok, err) = imp.verify( self.config_mock )
|
|
self.assertTrue(ok)
|
|
self.assertEqual( len(err), 0 )
|
|
|
|
|
|
def test_factory( self ):
|
|
"""
|
|
Tests the data_importers.Importer() function
|
|
"""
|
|
self.assertRaises( ValueError, imp.Importer, self.config_none )
|
|
self.assertRaises( ValueError, imp.Importer, self.config_noImport )
|
|
self.assertRaises( ValueError, imp.Importer, self.config_noName )
|
|
self.assertRaises( ValueError, imp.Importer, self.config_wrong )
|
|
imp.Importer( self.config_mock ) # should not raise
|
|
|
|
|
|
#-------------------------------------------------------------------------------
|
|
|
|
from ptaimport.data_importers.modo import ModoDataImporter
|
|
|
|
class TestModoDataImporter( unittest.TestCase ):
|
|
|
|
def setUp( self ):
|
|
pass
|
|
|
|
# Note: to activate this test, save a modo invoice (HTML) from your account
|
|
# to the "tests/data/" directory (as "tests/data/modo_invoice.html"), and
|
|
# also create a "modo_invoice_expected.txt" file containing the expected
|
|
# transactions to test against
|
|
@unittest.skipUnless( exists("tests/data/modo_invoice.html")
|
|
and exists("tests/data/modo_invoice_expected.txt"),
|
|
"Missing test data" )
|
|
def test_html_parse( self ):
|
|
modo = ModoDataImporter({
|
|
"name": "test",
|
|
"user_id": 42,
|
|
"password_id": "none"
|
|
})
|
|
|
|
expected_transactions = []
|
|
with open("tests/data/modo_invoice_expected.txt") as f:
|
|
expected_transactions = f.readlines()
|
|
|
|
actual_transactions = []
|
|
with open("tests/data/modo_invoice.html") as f:
|
|
actual_transactions = modo._parse_invoice( f.read() )
|
|
|
|
self.assertEqual( len(expected_transactions), len(actual_transactions) )
|
|
for i in range(0, len(expected_transactions)):
|
|
self.assertEqual(
|
|
expected_transactions[i].strip(),
|
|
actual_transactions[i].strip()
|
|
)
|