78 lines
No EOL
2.2 KiB
Python
78 lines
No EOL
2.2 KiB
Python
from settings import config
|
|
from paramecio.cromosoma.webmodel import WebModel
|
|
from paramecio.cromosoma import corefields
|
|
import unittest
|
|
|
|
|
|
# Create TestWebModelMethods
|
|
|
|
class ExampleModel(WebModel):
|
|
|
|
def create_fields(self):
|
|
|
|
# I can change other fields here, how the name.
|
|
|
|
self.register(corefields.CharField('title'))
|
|
self.register(corefields.CharField('content'))
|
|
|
|
model=ExampleModel()
|
|
|
|
class TestWebModelMethods(unittest.TestCase):
|
|
|
|
def test_test_table(self):
|
|
|
|
sql=model.create_table()
|
|
|
|
self.assertTrue(WebModel.query(WebModel, sql))
|
|
|
|
post={'title': 'Example title', 'content': 'New content'}
|
|
|
|
model.set_valid_fields()
|
|
|
|
self.assertTrue(model.insert(post))
|
|
|
|
self.assertEqual(model.insert_id(), 1)
|
|
|
|
post={'title': 'Example title Updated', 'content': 'New content Updated'}
|
|
|
|
model.conditions=['WHERE id=%s', [1]]
|
|
|
|
self.assertTrue(model.update(post))
|
|
|
|
model.yes_reset_conditions=False
|
|
|
|
model.conditions=['WHERE id=%s', [1]]
|
|
|
|
self.assertEqual(model.select_count(), 1)
|
|
|
|
self.assertEqual(model.select_a_row(1, ['title']), {'title': 'Example title Updated'})
|
|
|
|
self.assertEqual(model.select_a_row_where(['title']), {'title': 'Example title Updated'})
|
|
|
|
self.assertEqual(model.select_to_array(['title', 'content']), {1: {'title': 'Example title Updated', 'content': 'New content Updated'}})
|
|
|
|
model.yes_reset_conditions=True
|
|
|
|
model.reset_conditions()
|
|
|
|
self.assertEqual(model.conditions, ['WHERE 1=1', []])
|
|
|
|
cur=model.select()
|
|
|
|
row=model.fetch(cur)
|
|
|
|
self.assertEqual(row, {'id': 1, 'title': 'Example title Updated', 'content': 'New content Updated'})
|
|
|
|
self.assertTrue(model.element_exists(1))
|
|
|
|
model.conditions=['WHERE id=%s', [2]]
|
|
|
|
self.assertFalse(model.delete())
|
|
|
|
self.assertTrue(model.delete())
|
|
|
|
self.assertTrue(model.drop())
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|