Python TriggerPostgreSQL.sql_create_function_insertの例

コード例 #1

    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)