import os
import unittest

from tests.utils import (
    setup_test_env,
)
setup_test_env()
from softwarecenter.backend.login import get_login_backend
from softwarecenter.backend.login_impl.login_sso import (
    LoginBackendDbusSSO)
from softwarecenter.backend.login_impl.login_fake import (
    LoginBackendDbusSSOFake,
    )


class TestLoginBackend(unittest.TestCase):
    """ tests the login backend stuff """

    def test_fake_and_real_provide_similar_methods(self):
        """ test if the real and fake login provide the same functions """
        login_real = LoginBackendDbusSSO
        login_fake = LoginBackendDbusSSOFake
        # ensure that both fake and real implement the same methods
        self.assertEqual(
            set([x for x in dir(login_real) if not x.startswith("_")]),
            set([x for x in dir(login_fake) if not x.startswith("_")]))

    def test_get_login_backend(self):
        # test that we get the real one
        self.assertEqual(type(get_login_backend(None, None, None)),
                         LoginBackendDbusSSO)
        # test that we get the fake one
        os.environ["SOFTWARE_CENTER_FAKE_REVIEW_API"] = "1"
        self.assertEqual(type(get_login_backend(None, None, None)),
                         LoginBackendDbusSSOFake)
        # clean the environment
        del os.environ["SOFTWARE_CENTER_FAKE_REVIEW_API"]


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