def test_Credentials_check_activated_false(self):
"""Test Credentials.check_activated() not activated."""
c = Credentials.get_current()
c.activation = _Activation("some_email", False)
with patch("streamlit.credentials._exit") as p:
c._check_activated(auto_resolve=False)
p.assert_called_once_with("Activation email not valid.")
def test_Credentials_check_activated_already_loaded(self):
"""Test Credentials.check_activated() already loaded."""
c = Credentials.get_current()
c.activation = _Activation("some_email", True)
with patch("streamlit.credentials._exit") as p:
c._check_activated(auto_resolve=False)
p.assert_not_called()
def test_Credentials_check_activated_error(self):
"""Test Credentials.check_activated() has an error."""
c = Credentials.get_current()
c.activation = _Activation("some_email", True)
with patch.object(c, "load", side_effect=Exception(
"Some error")), patch("streamlit.credentials._exit") as p:
c._check_activated(auto_resolve=False)
p.assert_called_once_with("Some error")
def test_Credentials_load_twice(self):
"""Test Credentials.load() called twice."""
c = Credentials.get_current()
c.activation = _Activation("some_email", True)
with patch("streamlit.credentials.LOGGER") as p:
c.load()
p.error.assert_called_once_with(
"Credentials already loaded. Not rereading file.")
def test_Credentials_activate_already_activated(self):
"""Test Credentials.activate() already activated."""
c = Credentials.get_current()
c.activation = _Activation("some_email", True)
with patch("streamlit.credentials.LOGGER") as p:
with pytest.raises(SystemExit):
c.activate()
self.assertEqual(p.error.call_count, 2)
self.assertEqual(p.error.call_args_list[1], call("Already activated"))
def test_Credentials_activate_already_activated_not_valid(self):
"""Test Credentials.activate() already activated but not valid."""
c = Credentials.get_current()
c.activation = _Activation("some_email", False)
with patch("streamlit.credentials.LOGGER") as p:
with pytest.raises(SystemExit):
c.activate()
self.assertEqual(p.error.call_count, 2)
self.assertEqual(
str(p.error.call_args_list[1])[0:27],
"call('Activation not valid.")
def test_Credentials_save(self):
"""Test Credentials.save()."""
c = Credentials.get_current()
c.activation = _Activation("some_email", True)
truth = textwrap.dedent("""
[general]
email = "some_email"
""").lstrip()
streamlit_root_path = os.path.join("/mock/home/folder",
file_util.CONFIG_FOLDER_NAME)
# patch streamlit.*.os.makedirs instead of os.makedirs for py35 compat
with patch("streamlit.credentials.open", mock_open(),
create=True) as open, patch(
"streamlit.credentials.os.makedirs") as make_dirs:
c.save()
make_dirs.assert_called_once_with(streamlit_root_path,
exist_ok=True)
open.return_value.write.assert_called_once_with(truth)
