from unittest import TestCase

import os

from bitter import utils
from bitter import config as c

class TestUtils(TestCase):

    def setUp(self):
        self.credfile = '/tmp/credentials.txt'
        c.CREDENTIALS = self.credfile
        if os.path.exists(self.credfile):
            os.remove(self.credfile)
        utils.create_credentials(self.credfile)
        

    def test_create_credentials(self):
        assert os.path.exists(self.credfile)
        os.remove(self.credfile)
        utils.create_credentials() # From config
        assert os.path.exists(self.credfile)

    def test_add_credentials(self):
        utils.add_credentials(self.credfile, user="test")
        assert utils.get_credentials(self.credfile)
        assert utils.get_credentials(user="test")
        assert list(utils.get_credentials(user="test"))[0]["user"] == "test"

    def test_get_credentials(self):
        utils.add_credentials(self.credfile, user="test")
        assert utils.get_credentials(user="test")
        assert not utils.get_credentials(user="test", inverse=True)

    def test_add_two_credentials(self):
        utils.add_credentials(self.credfile, user="test")
        utils.add_credentials(self.credfile, user="test2")
        assert utils.get_credentials(user="test")
        assert utils.get_credentials(user="test2")


    def test_delete_credentials(self):
        utils.add_credentials(self.credfile, user="test")
        assert utils.get_credentials(user="test")
        utils.delete_credentials(user="test")
        print(utils.get_credentials())
        assert not utils.get_credentials(user="test")