def test_load_from_cache(self):
d = {"aa": 1, "bb": 2, "brother": {"name": "jack"}}
factory = CommonModelFactory()
models = factory.from_json(data=d, root_name='data')
cache = models.to_cache()
new_model = factory.from_cache(data=cache, root_name='data')
assert len(models.model.brothers) == 1
def test_max_depth(self):
factory = CommonModelFactory(use_foreign_key=True, column_fmt="UNDERLINE", is_echo=False)
d = {"aaBbb": {"nameNN": "example"}}
model = factory.from_json(data=d, root_name="data", max_depth=0)
model.create_tables_in_db()
model.store(data=d)
## TODO: consider need support max_depth = 0 when data is pressed
model.store(data=d, is_press=True)
def test_type_convert(self):
d = {"aa": 1, "bb": 2, "brother": {"name": "jack"}}
factory = CommonModelFactory(is_echo=False, use_foreign_key=True)
model = factory.from_json(data=d, root_name="data")
model.create_tables_in_db()
new_d = {"aa": '111', "name": "jack", "bb": 2, "namess": [{"other": "jack"}], 'sons': [{"son_name": "aa"}]}
model.store(data=new_d, is_press=True)
assert len(model.search(search_args=[("aa", 111)])) == 1
def test_use_str_as_db_type(self):
factory = CommonModelFactory(use_foreign_key=True, column_fmt="UNDERLINE", is_echo=False,
str2col='VARCHAR(1000)', database='sqlite',
int2col='INTEGER',
float2col='DECIMAL(10,2)')
for d in self.TEST_DATA:
model = factory.from_json(data=d, root_name="data", max_depth=0)
model.create_tables_in_db()
model.store(data=d)
def test_convert(self):
factory = CommonModelFactory(use_foreign_key=True, column_fmt="UNDERLINE", is_echo=False,
str2col='VARCHAR(1000)', database='sqlite',
int2col='INTEGER',
float2col='DECIMAL(10,2)')
for d in self.TEST_DATA:
model = factory.from_json(data=d, root_name="data")
model.create_tables_in_db()
nd = model.convert(data=d)
assert nd == d
def test_alia_table(self):
d = {"aa": 1, "bbb": {":<<bb": 2}, "brother": {"name": "jack", "namess": [{"other": "jack"}]},
'sons': [{"son_name": "aa"}]}
factory = CommonModelFactory(is_echo=False, use_foreign_key=False)
model = factory.from_json(data=d, root_name="data")
model.create_tables_in_db()
time.sleep(1)
new_d = {"aa": 1, "name": "jack", "bb": 2, "namess": [{"other": "jack"}], 'sons': [{"son_name": "aa"}]}
model.store(data=new_d, is_press=True)
assert len(model.search(search_args=[])) == 1
assert len(model.search(search_args=[("aa", 1)])) == 1
assert len(model.search(search_args=[("bbb", 2)])) == 1
def test_add_table_comment(self):
## PS: if you want to run this test please set your default_db_url (sqlite not support comment)
factory = CommonModelFactory(use_foreign_key=True, column_fmt="UNDERLINE", is_echo=False,
str2col='VARCHAR(1000)', database='sqlite',
int2col='INTEGER',
float2col='DECIMAL(10,2)',)
# default_db_url="mysql+mysqldb://root:[email protected]/test?charset=utf8")
d = {"aa": 1.0, "bb": 2, "brother": {"name": "jack"}}
model = factory.from_json(data=d, root_name='jackLove')
models = model.model
models.comment = "Test Comment ADD IN DB"
model.create_tables_in_db()
model.store(data=d)
model.delete_tables_in_db()
def test_alias_model(self):
factory = CommonModelFactory(use_foreign_key=True, column_fmt="UNDERLINE", is_echo=False,
str2col='VARCHAR(1000)', database='sqlite',
int2col='INTEGER',
float2col='DECIMAL(10,2)')
d = {"aa": 1.0, "bb": 2, "brother": {"name": "jack"}}
model = factory.from_json(data=d, root_name='jackLove')
model.create_tables_in_db()
model.store(data=d)
models = model.model
models.alias = "newJack"
brother = models.brothers['brother']
brother.alias = "brother_new_name"
brother.fields['name'].alias = "new_name"
brother.fields['name'].name = "name"
assert len(model.search(search_args=[("brother_new_name.new_name", "jack")])) == 1
def test_issues4_foreign_key(self):
factory = CommonModelFactory(use_foreign_key=True, column_fmt="UNDERLINE", is_echo=False)
model = factory.from_json(data={"aaBbb": {"nameNN": "example"}}, root_name="data")
model.create_tables_in_db()
def test_load_from_json(self):
d = {"aa": 1, "bb": 2}
factory = CommonModelFactory()
models = factory.from_json(data=d, root_name='data')
assert len(models.model.sons) == 0
def test_drop_all_table_with_foreign_key(self):
d = {"aa": 1, "bb": 2, "brother": {"name": "jack"}}
factory = CommonModelFactory(is_echo=False, use_foreign_key=True)
model = factory.from_json(data=d, root_name="data")
model.delete_tables_in_db()
def test_create_all_table(self):
d = {"aa": 1, "bb": 2, "brother": {"name": "jack"}}
factory = CommonModelFactory(is_echo=False)
model = factory.from_json(data=d, root_name="data")
model.create_tables_in_db()
def test_load_describe(self):
d = {"aa": 1, "bb": 2, "brother": {":name": "jack"}}
assert len(CommonModelFactory(ignore_head=':').from_json(data=d, root_name='data').model.brothers) == 0
def test_load_brother(self):
d = {"aa": 1, "bb": 2, "brother": {"name": "jack"}}
assert len(CommonModelFactory().from_json(data=d, root_name='data').model.brothers) == 1
def test_load_son(self):
d = {"aa": 1, "bb": 2, "sons": [{"name": "jack"}]}
assert len(CommonModelFactory().from_json(data=d, root_name='data').model.sons) == 1