Source code for tests.test_basic

from .context import lds
import unittest
import json


class BasicTestSuite(unittest.TestCase):
    def test_basic(self):
        db = lds.DocStore()
        first = '{"year":"2017", "months": [{"m":"january"}, {"m":"february"}]}'
        second = '{"year":"2017", "months": [{"m":"march"}, {"m":"april"}]}'
        db.add(first)
        db.add(second)

        assert db.search('{"m":"march"}') == [json.loads(second)]
        assert db.search('{"m":"may"}') == []

        third = '{"year":"2017", "months": [{"m":"may"}, {"m":"april"}]}'
        db.update('{"m":"march"}', '{"months": [{"m":"may"}, {"m":"april"}]}')

        assert db.search('{"m":"march"}') == []
        assert db.search('{"m":"may"}') == [json.loads(third)]

    def test_basic_update(self):
        db = lds.DocStore()
        first = '{"year":"2017", "months": [{"m":"january"}, {"m":"february"}]}'
        second = '{"year":"2017", "months": [{"m":"march"}, {"m":"april"}]}'
        db.add(first)
        db.add(second)

        assert db.search('{"m":"march"}') == [json.loads(second)]
        assert db.search('{"m":"may"}') == []

        third = '{"year":"2017", "months": [{"m":"may"}, {"m":"april"}]}'
        db.update('{"m":"march"}', '{"m":"may"}', exact=True)
        assert db.search('{"m":"march"}') == []
        assert db.search('{"m":"may"}') == [json.loads(third)]


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