Add mass objects

This commit is contained in:
Ege Emir Özkan
2020-08-23 00:46:40 +03:00
parent 7efa8c8f4e
commit c8e707e2f0
4 changed files with 89 additions and 3 deletions

View File

@@ -2,11 +2,10 @@ import unittest
from datalite import datalite
from datalite.constraints import Unique, ConstraintFailedError
from datalite.fetch import fetch_if, fetch_all, fetch_range, fetch_from, fetch_equals, fetch_where
from datalite.mass_actions import create_many_entries
from sqlite3 import connect
from dataclasses import dataclass, asdict
from math import floor
from os import remove
from datalite.migrations import basic_migrate, _drop_table
@@ -51,6 +50,12 @@ class ConstraintedClass:
unique_str: Unique[str]
@datalite(db_path='test.db')
@dataclass
class MassCommit:
str_: str
def getValFromDB(obj_id = 1):
with connect('test.db') as db:
cur = db.cursor()
@@ -190,5 +195,16 @@ class DatabaseConstraints(unittest.TestCase):
self.obj.remove_entry()
class DatabaseMassInsert(unittest.TestCase):
def setUp(self) -> None:
self.objs = [MassCommit('cat') for _ in range(30)]
def testMassCreate(self):
create_many_entries(self.objs)
def tearDown(self) -> None:
[obj.remove_entry() for obj in self.objs]
if __name__ == '__main__':
unittest.main()