import unittest
import tempfile
from nullptr.store import Store
from nullptr.models import Base

class Dummy(Base):
  def define(self):
    self.count: int = 0
    self.data: str = ""
  
  def update(self, d):
    self.seta('count', d)
    
  @classmethod
  def ext(self):
    return 'dum'
  
  def f(self, detail=1):
    r = super().f(detail)
    if detail >2:
      r += f' c:{self.count}'
    return r
    
class TestStore(unittest.TestCase):
  def setUp(self):
    self.store_file = tempfile.NamedTemporaryFile()
    self.s = Store(self.store_file.name, False)
  
  def tearDown(self):
    self.s.close()
    self.store_file.close()
    
  def reopen(self):
    self.s.flush()
    self.s.close()
    self.s = Store(self.store_file.name, False)
    
  def test_single(self):
    dum = self.s.get(Dummy, "5", create=True)
    dum.count = 1337
    dum.data = "A" * 1000
    self.reopen()
    
    dum = self.s.get(Dummy, "5")
    self.assertEqual(1337, dum.count)
  
  def test_grow(self):
    dum = self.s.get(Dummy, "5", create=True)
    dum.data = "A"
    dum2 = self.s.get(Dummy, "7",create=True)
    self.reopen()
    dum = self.s.get(Dummy, "5")
    dum.data = "A" * 1000
    dum.count = 1337
    self.reopen()
    dum = self.s.get(Dummy, "5")
    self.assertEqual(1337, dum.count)
  
  def test_purge(self):
    dum = self.s.get(Dummy, "5", create=True)
    dum.data = "A"
    dum2 = self.s.get(Dummy, "7",create=True)
    dum2.count = 1337
    self.s.flush()
    self.s.purge(dum)
    self.reopen()
    dum2 = self.s.get(Dummy, "7")
    self.assertEqual(1337, dum2.count)
  
  def test_grow_last(self):
    dum = self.s.get(Dummy, "5", create=True)
    dum.data = "A"
    dum2 = self.s.get(Dummy, "7",create=True)
    self.reopen()
    dum2 = self.s.get(Dummy, "7")
    dum2.data = "A" * 1000
    dum2.count = 1337
    dum3 = self.s.get(Dummy, "9",create=True)
    dum3.count = 1338
    self.reopen()
    dum2 = self.s.get(Dummy, "7")
    self.assertEqual(1337, dum2.count)
    dum3 = self.s.get(Dummy, "9")
    self.assertEqual(1338, dum3.count)
    
  def test_purge_last(self):
    dum = self.s.get(Dummy, "5", create=True)
    dum.data = "A"
    dum2 = self.s.get(Dummy, "7",create=True)
    self.reopen()
    dum2 = self.s.get(Dummy, "7")
    self.s.purge(dum2)
    dum3 = self.s.get(Dummy, "9",create=True)
    dum3.count = 1338
    self.reopen()
    dum2 = self.s.get(Dummy, "7")
    self.assertIsNone(dum2)
    dum3 = self.s.get(Dummy, "9")
    self.assertEqual(1338, dum3.count)