tests.py

import db

TEST_TABLE_NAME = "addresses"


def test_all_table_info():
	tables = db.all_table_info()
	assert len(tables) > 0 and len(tables[0]) > 0


def test_table_record_count():
	r = db.table_record_count(TEST_TABLE_NAME)
	assert r >= 0


def test_get_all_tables_name():
	table_names = db.all_table_name()
	assert len(table_names) > 0


def test_desc_of_table():
	table_name = "addresses"
	desc = db.table_desc(table_name)
	assert len(desc) > 0


def test_table_field_desc():
	table_name = 'event_codes'
	field_name = 'c_event_name'
	field_desc = 'event_name(事件名稱-英文)'
	assert db.table_field_desc(table_name, field_name)['DumpFldNm'] == field_desc


def test_all_table_desc():
	assert db.all_table_desc()


def test_drop_table():
	try:
		r = db.conn.execute("delete from addresses where 1=1;").fetchall()
	except Exception as e:
		assert e


def test_offce_type_tree():
	r = db.offce_type_tree()
	assert r