mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
cfd57005bc
* state * note * add changeset * buttons * clear button * lint * if * clear' * reset state * fix test --------- Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
import gradio as gr
|
|
|
|
|
|
class TestClearButton:
|
|
def test_clear_event_setup_correctly(self):
|
|
with gr.Blocks() as demo:
|
|
chatbot = gr.Chatbot([["Hello", "How are you?"]])
|
|
with gr.Row():
|
|
textbox = gr.Textbox(scale=3, interactive=True)
|
|
gr.ClearButton([textbox, chatbot], scale=1)
|
|
|
|
clear_event_trigger = demo.dependencies.pop()
|
|
assert not clear_event_trigger["backend_fn"]
|
|
assert clear_event_trigger["js"]
|
|
assert clear_event_trigger["outputs"] == [textbox._id, chatbot._id]
|
|
|
|
def test_clear_event_setup_correctly_with_state(self):
|
|
with gr.Blocks() as demo:
|
|
chatbot = gr.Chatbot([["Hello", "How are you?"]])
|
|
state = gr.State("")
|
|
gr.ClearButton([state, chatbot], scale=1)
|
|
|
|
clear_event_trigger_state = demo.dependencies.pop()
|
|
assert clear_event_trigger_state["backend_fn"]
|
|
|
|
|
|
class TestOAuthButtons:
|
|
def test_login_button_warns_when_not_on_spaces(self):
|
|
with pytest.warns(UserWarning):
|
|
with gr.Blocks():
|
|
gr.LoginButton()
|
|
|
|
def test_logout_button_warns_when_not_on_spaces(self):
|
|
with pytest.warns(UserWarning):
|
|
with gr.Blocks():
|
|
gr.LogoutButton()
|
|
|
|
@patch("gradio.oauth.get_space", lambda: "fake_space")
|
|
@patch("gradio.oauth._add_oauth_routes")
|
|
def test_login_button_setup_correctly(self, mock_add_oauth_routes):
|
|
with gr.Blocks() as demo:
|
|
button = gr.LoginButton()
|
|
|
|
login_event = demo.dependencies[0]
|
|
assert login_event["targets"][0][1] == "click"
|
|
assert not login_event["backend_fn"] # No Python code
|
|
assert login_event["js"] # But JS code instead
|
|
assert login_event["inputs"] == [button._id]
|
|
assert login_event["outputs"] == []
|