#!/usr/bin/env python3

import os
import unittest

import bootstrap
import persistent_state as ps

import unittest_utils as uu


class TestPersistentState(unittest.TestCase):
    def test_it_all_baby(self):
        filename = '/tmp/.persistent_state_test.bin'
        os.system(f'/bin/rm {filename}')
        s = ps.PersistentState(filename)
        self.assertEqual(0, len(s))
        s['test'] = 123
        self.assertEqual(123, s['test'])
        self.assertEqual(1, len(s))
        s['ing'] = 234
        self.assertEqual(2, len(s))
        del s['ing']
        self.assertEqual(1, len(s))
        s.save()
        t = ps.PersistentState(filename)
        self.assertEqual(1, len(t))
        self.assertEqual(123, t['test'])


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