# py-scopes: tests.test_storage

""" Tests for the (PostgreSQL) storage-related functionalities"""

from scopes import config
config.settings = config.Settings(_env_file='tests/test.env')

from datetime import datetime
import pytest

from scopes.api import main
from scopes.storage import concept, folder, message, topic, tracking
from scopes.util import keywords as k

# predefined keywords used as domain, action, class head values in messages
k('session system')

@pytest.fixture(scope='module')
def storage():
    storage = config.StorageFactory(config.settings)(config.settings.dbschema)
    storage.dropTable('tracks')
    storage.dropTable('folders')
    storage.dropTable('types')
    storage.dropTable('preds')
    storage.dropTable('rels')
    storage.dropTable('topics')
    storage.dropTable('messages')
    storage.commit()
    return storage

def test_tracking(storage):
    tracks = storage.create(tracking.Container)
    tr01 = tracking.Track('t01', 'john')
    assert tr01.head == {'taskId': 't01', 'userName': 'john'}
    assert tr01.taskId == 't01'
    assert tr01.userName == 'john'
    tr01.update(dict(activity='testing'))
    trid01 = tracks.save(tr01)
    assert trid01 > 0

    tr01a = tracks[f'{trid01:07}']
    assert tr01a.head == tr01.head
    assert tr01a.trackId == trid01
    assert tr01a.data.get('activity') == 'testing'
    tr01a.update(dict(text='Set up unit tests.'))
    tr01a.timeStamp = None
    assert tracks.save(tr01a) > 0

    tr01b = tracks.queryLast(taskId='t01')
    assert tr01b.head == tr01.head
    assert tr01b.trackId != trid01
    assert tr01b.data.get('activity') == 'testing'
    assert tr01b.data.get('text') == 'Set up unit tests.'

    tr02 = tracking.Track('t02', 'jim', trackId=31, timeStamp=datetime(2023, 11, 30),
                        data=dict(activity='concept'))
    trid02 = tracks.upsert(tr02)
    assert trid02 == 31
    assert tr02.uid == 'rec-31'
    tr02.trackId = trid01
    trid021 = tracks.upsert(tr02)
    assert trid021 == trid01
    assert tr02.uid == 'rec-' + str(trid01)

    tr03 = storage.getItem('rec-31')
    assert tr03.trackId == 31
    n = tracks.remove(tr03)
    assert n == 1
    assert tracks.get(31) == None

    storage.commit()

def test_folder(storage):
    root = folder.Root(storage)
    assert list(root.keys()) == []
    root['top'] = folder.Folder()
    assert list(root.keys()) == ['top']
    top = root['top']
    top['child1'] = folder.Folder(data=dict(title='First Child'))
    assert list(top.keys()) == ['child1']
    ch1 = top['child1']
    assert ch1.parent == top.rid
    assert list(top.keys()) == ['child1']
    ch1.set('name', 'level2-item1')
    ch1.storeTrack()
    storage.commit()

def test_type(storage):
    concept.setupCoreTypes(storage)
    types = storage.getContainer(concept.Type)
    tps = list(types.query())
    assert len(tps) == 7
    tfolder = types.queryLast(name='folder')
    fldrs = list(tfolder.values())
    assert len(fldrs) == 2
    assert fldrs[0].name == 'top'
    storage.commit()

def test_topic(storage):
    rels = storage.getContainer(concept.Triple)
    topics = storage.getContainer(topic.Topic)
    types = storage.getContainer(concept.Type)
    concept.storePredicate(storage, concept.defaultPredicate)
    root = folder.Root(storage)
    root['top']['topics'] = ftopics = folder.Folder()
    ttopic = types.queryLast(name='topic')
    assert ttopic.name == 'topic'
    ftopics.setTarget(ttopic)
    assert ftopics.getTarget().name == 'topic'
    tp_itc = topic.Topic('itc', data=dict(
        title='ITC', description='Information and Communication Technology'))
    topics.save(tp_itc)
    tp_proglang = topic.Topic('prog_lang', data=dict(
        title='Programming Languages', 
        description='Programming Languages'))
    topics.save(tp_proglang)
    tp_itc.addChild(tp_proglang)
    c = list(tp_itc.children())
    assert c[0].name == 'prog_lang'
    storage.commit()

def test_message(storage):
    messages = storage.create(message.Messages)
    m01 = message.Message('system', 'data', 'session', 'V1_317784226621611853')
    m01.update(dict(userid='tst9'))
    mid01 = messages.save(m01)
    msgs = list(messages.query())
    assert msgs[0].domain is k.system
    assert len(msgs) == 1
    storage.commit()