def test_lookup_foreign_key(self):
expected_fk = 1
actual_fk = dbwrappers.lookup_foreign_key(self.c,
self.foreign_key_table,
self.fk_col_b,
'csv_text_field',
'foo')
self.assertEqual(expected_fk, actual_fk)
def test_lookup_foreign_key_duplicate(self):
"""Table has 2 rows with text_field == 'bar'"""
with self.assertRaises(dbwrappers.ForeignKeyException):
dbwrappers.lookup_foreign_key(self.c, self.foreign_key_table,
self.fk_col_b,
'csv_text_field', 'bar')
def test_lookup_foreign_key_missing(self):
"""No row with text_field == 'baz'"""
with self.assertRaises(dbwrappers.ForeignKeyException):
dbwrappers.lookup_foreign_key(self.c, self.foreign_key_table,
self.fk_col_b,
'csv_text_field','baz')
def test_lookup_foreign_key_wrong_field(self):
"""'foo' is in text_field, not in date_field"""
with self.assertRaises(dbwrappers.ForeignKeyException):
dbwrappers.lookup_foreign_key(self.c, self.foreign_key_table,
self.fk_col_b,
'csv_date_field', 'foo')
