Example #1
0
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
Example #2
0
    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()