from unittest import TestCase from vitamin.modules.database import PDO, ORM from tests.database.models_for_test import User, Parent, Children pdo = PDO(config="tests.framework.simple.config") orm = ORM(pdo=pdo, config="tests.framework.simple.config") pdo.connect() orm.regiserModel(User) orm.regiserModel(Parent) orm.regiserModel(Children) class TestSQL(TestCase): def test_0create(self): User.Create().go() def test_1insert(self): u = User() u.name = "Karlson" u.address = "roof" User.Insert().instance(u).go() u2 = User() User.Insert().instance(u2).go() def test_2update(self): u = User() u.id = 1
address = CharField(length(100)) class Parent(Model): id = IntegerField(autoinc, primary) name = CharField(length(100), notnull) class Children(Model): id = IntegerField(autoinc, primary) parent = ForeignField(Parent) pdo = PDO() pdo.regiserModel(User) pdo.regiserModel(Parent) pdo.regiserModel(Children) pdo.connect() class TestSQL(TestCase): def test_0create(self): User.Create().go() def test_1insert(self): u = User() u.name = "Karlson" u.address = "roof" User.Insert().instance(u).go()