在python unittest中，保存Peewee对象的实例会引发peewee.IntegrityError：和peewee.OperationalError：

Question

When I save a instance inside the setUp of python unittest I have 2 errors popping up:

sqlite3.IntegrityError: NOT NULL constraint failed: registro_c170.reg_c100_id

and

peewee.OperationalError: Connection already opened. (this one is raised for each test_method inside my testing Class.)

Context:

I'm very new to object-oriented programming and unittest. I'm using python to build a model of a hierarchy of registers. This registers holds information about taxation and accountability.

Model:

from peewee import *

db = SqliteDatabase('pis_cofins.db')

class BaseModel(Model):

    class Meta:
        database = db

class Registro_C100(BaseModel):
    """
    """
    # Atributos
    x = CharField()
    y = CharField()


    @property
    def dados(self):
        # some property

    @dados.setter
    def dados(self, novos_dados):
        #...


    @property
    def formato_linha(self):
        # some method

class Registro_C170(BaseModel):
    """
    """
    # Atributos
    a = CharField()
    b = CharField()

    reg_c100 = ForeignKeyField(Registro_C100, backref='registros_c170')

    @property
    def dados(self):
        # some property

    @dados.setter
    def dados(self, novos_dados):
        # ...

    @property
    def formato_linha(self):
        # some method

Model_Test:

import unittest, os

from peewee import SqliteDatabase

from modelo_pis_cofins import Registro_0140, Registro_C170, Registro_C100

MODELS = [Registro_0140, Registro_C170, Registro_C100]

# use an in-memory SQLite for tests.
test_db = SqliteDatabase(':memory:')

class TestRegistroC170(unittest.TestCase):

    def setUp(self):
        # http://docs.peewee-orm.com/en/latest/peewee/database.html#testing-peewee-applications 
        test_db.bind(MODELS, bind_refs=False, bind_backrefs=False)
        test_db.connect()
        test_db.create_tables(MODELS)

        self.regc170 = Registro_C170()
        self.regc170_1 = Registro_C170(a="a", b="b")
        self.regc170_1.save() # this raises the errors

        self.regc100 = Registro_C100(x="x", y="y")
        self.regc100.save() # this raises the errors

    def tearDown(self):
        test_db.drop_tables(MODELS)
        test_db.close()

    def test_Registro_C170_atributos(self):
        # do some test

    def test_Registro_c170_dados(self): 
        # do some test:

    def test_Registro_C170_formato_linha(self):
        # do some test

    def test_Registro_C170_relacionamentos(self):
        # I will test relationships between "Registro_C170" and "Registro_C100"

        query = Registro_C100.get(Registro_C100.id == 1)
        query.registros_c170

Errors:

======================================================================
ERROR: test_Registro_C170_atributos (modelo_pis_cofins_testes.TestRegistroC170)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 2640, in execute_sql
    cursor.execute(sql, params or ())
sqlite3.IntegrityError: NOT NULL constraint failed: registro_c170.reg_c100_id

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:\Users\vitor.rabelo\Documents\projetos_automatizacao\projeto_efd\processamento_efd\modelo_pis_cofins_testes.py", line 76, in setUp
    self.regc170_1.save()
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 5577, in save
    pk_from_cursor = self.insert(**field_dict).execute()
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 1574, in inner
    return method(self, database, *args, **kwargs)
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 1645, in execute
    return self._execute(database)
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 2288, in _execute
    return super(Insert, self)._execute(database)
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 2063, in _execute
    cursor = database.execute(self)
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 2653, in execute
    return self.execute_sql(sql, params, commit=commit)
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 2647, in execute_sql
    self.commit()
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 2438, in __exit__
    reraise(new_type, new_type(*exc_args), traceback)
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 177, in reraise
    raise value.with_traceback(tb)
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 2640, in execute_sql
    cursor.execute(sql, params or ())
peewee.IntegrityError: NOT NULL constraint failed: registro_c170.reg_c100_id

======================================================================
ERROR: test_Registro_C170_formato_linha (modelo_pis_cofins_testes.TestRegistroC170)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "C:\Users\vitor.rabelo\Documents\projetos_automatizacao\projeto_efd\processamento_efd\modelo_pis_cofins_testes.py", line 68, in setUp
    test_db.connect()
  File "C:\Users\vitor.rabelo\virtual_enviroments\processamento_efd\lib\site-packages\peewee.py", line 2583, in connect
    raise OperationalError('Connection already opened.')
peewee.OperationalError: Connection already opened.

What I tried:

Honestly, the only thing I know I could try is to create each instance inside each test method, but it is exactly what I'm trying to avoid.

I have searched about "peewee testing" and "peewee unittest" and haven't found anython helpful. Peewee documentation only shows you how to write the setUp and tearDown methods: http://docs.peewee-orm.com/en/latest/peewee/database.html#testing-peewee-applications or how to use their test_utils: http://docs.peewee-orm.com/en/latest/peewee/playhouse.html#test-utils

ps: This is my first question, hope it's clear enough :)

Answer 1

Ok, this is embarassing. I realized that I was saving an instance with no value to the ForeignKeyField (that defaults to null=False ). All I need to do is to allow the field to accept null values ( null=True ).

Sorry for any trouble.