summaryrefslogtreecommitdiff
path: root/sqlplain/tests/test_truncate.py
blob: fbfb390c8f764fcccb1b4949fbf2aaa0a7e5a901 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from __future__ import with_statement
import os
from sqlplain import connect, util
from test_million import makedb, makedatafile, clock, create_price_table

databases = 'postgres_test sqlite_test'.split()


def test_truncate():
    for uri in databases:
        db = util.create_db(uri, force=True)
        create_price_table(db)
        fname = makedatafile(100, 100)
        try:
            util.load_file(db, fname, 'price')
            with clock:
                util.truncate_table(db, 'price')
            yield lambda *a: None, uri, 'truncate'
            
            util.load_file(db, fname, 'price')
            with clock:
                db.execute('delete from price')
            yield lambda *a: None, uri, 'delete'
        finally:
            os.remove(fname)