import unittest
import db_controller
import logging
import db_controller as dbc

from rhizi_server import Config

class TestDBController(unittest.TestCase):

    db_ctl = None
    log = None

    n_map = { 'Skill': [{'name': 'kung fu', 'id': 'skill_00' },
                            {'name': 'judo', 'id': 'skill_01' }
                           ],
                  'Person': [{'name': 'Bob', 'id': 'person_00' },
                             {'name': 'Alice', 'id': 'person_01' }]
            }

    @classmethod
    def setUpClass(self):
        cfg = Config.init_from_file('res/etc/rhizi-backend.conf')
        self.log = logging.getLogger('rhizi')
        self.db_ctl = dbc.DB_Controller(cfg)

    def setUp(self):
        self.flush_db()

    def test_node_lifecycle(self):
        """
        test node commit & load
        """

        self.db_ctl.exec_op(dbc.DBO_add_node_set(self.n_map))
        n_set = self.db_ctl.exec_op(dbc.DBO_load_node_set_by_id_attribute(['skill_00', 'person_01']))
        self.assertEqual(len(n_set), 2, 'incorrect result size')

    def tearDown(self): pass

if __name__ == "__main__":
    unittest.main()