class TestReference(unittest.TestCase):
def setUp(self):
self.ref = Reference()
def tearDown(self):
pass
def test_set_and_get_field(self):
self.ref.set_field('random_field', 'random_value')
self.failUnless(self.ref.get_field('random_field').value == 'random_value')
def test_get_fields(self):
self.ref.set_field('rf01', 'rv01')
self.ref.set_field('rf02', 'rv02')
self.ref.set_field('rf03', 'rv04')
self.failUnless(len(self.ref.get_fields()) == 3)
self.failUnless(self.ref.get_fields() == ['rf01', 'rf02', 'rf03'])
def test_set_field_to_none(self):
self.ref.set_field('some_field', None)
field = self.ref.get_field('some_field')
self.failUnless(field.valid == False)
def test_set_and_get_entry(self):
self.ref.set_entry('This is an entry')
self.failUnless(self.ref.get_entry() == 'This is an entry')
def test_validate_reference_fields(self):
ref = Reference()
ref.set_field('title', 'Some article title')
ref.set_field('year', '32')
raw_text = "Some article title and something else"
self.iec._validate_reference_fields(ref, raw_text)
self.failUnless(ref.get_field('title').valid == True)
self.failUnless(ref.get_field('year').valid == False)
def test_validate_reference_fields(self):
ref = Reference()
ref.set_field('title', 'Some article title')
ref.set_field('year', '32')
raw_text = "Some article title and something else"
self.iec._validate_reference_fields(ref, raw_text)
self.failUnless(ref.get_field('title').valid == True)
self.failUnless(ref.get_field('year').valid == False)
