class TestEncryptionStateStore(unittest.TestCase):
    def setUp(self):
        self.conn = sqlite3.connect('test-db', check_same_thread=False)
        self.e = EncryptionState(self.conn)

    def test_create(self):
        self.assertEquals(self.e.is_active('*****@*****.**'), False)

    def test_enable_encryption(self):
        self.e.activate('*****@*****.**')
        self.assertEquals(self.e.is_active('*****@*****.**'), True)

    def test_disable_encryption(self):
        self.e.activate('*****@*****.**')
        self.assertEquals(self.e.is_active('*****@*****.**'), True)
        self.e.deactivate('*****@*****.**')
        self.assertEquals(self.e.is_active('*****@*****.**'), False)

    def tearDown(self):
        os.remove('test-db')
class TestEncryptionStateStore(unittest.TestCase):

    def setUp(self):
        self.conn = sqlite3.connect('test-db', check_same_thread=False)
        self.e = EncryptionState(self.conn)

    def test_create(self):
        self.assertEquals(self.e.is_active('*****@*****.**'), False)

    def test_enable_encryption(self):
        self.e.activate('*****@*****.**')
        self.assertEquals(self.e.is_active('*****@*****.**'), True)

    def test_disable_encryption(self):
        self.e.activate('*****@*****.**')
        self.assertEquals(self.e.is_active('*****@*****.**'), True)
        self.e.deactivate('*****@*****.**')
        self.assertEquals(self.e.is_active('*****@*****.**'), False)

    def tearDown(self):
        os.remove('test-db')
 def setUp(self):
     self.conn = sqlite3.connect('test-db', check_same_thread=False)
     self.e = EncryptionState(self.conn)
 def setUp(self):
     self.conn = sqlite3.connect('test-db', check_same_thread=False)
     self.e = EncryptionState(self.conn)