def test_sql_create_function_insert(self): trig = TriggerPostgreSQL() res = trig.sql_create_function_insert('book_nbpage_insert()', 'book', 'nbpage', 'book_nbpage') attend = u"""CREATE OR REPLACE FUNCTION book_nbpage_insert() RETURNS TRIGGER AS $BODY$ BEGIN IF (SELECT nbpage FROM book_nbpage WHERE nbpage=NEW.nbpage) IS NOT NULL THEN UPDATE book_nbpage SET agg_count=agg_count+1 WHERE nbpage=NEW.nbpage; ELSE INSERT INTO book_nbpage VALUES (NEW.nbpage, 1); END IF; RETURN NEW; END; $BODY$ LANGUAGE plpgsql;""" self.assertEqual(res, attend)
def test_init(self): trig = TriggerPostgreSQL() self.assertEqual('PostgreSQL', trig.name)
def setUp(self): self.agg = TriggerPostgreSQL() self.maxDiff = None