from pytest import raises
from bokeh.models import CustomJS, Slider


def test_js_callback():

    slider = Slider()

    cb = CustomJS(code="foo();", args={'x': slider})
    assert cb.lang == 'javascript'
    assert 'foo()' in cb.code
    assert cb.args['x'] is slider

    with raises(ValueError):  # not a plot object
        CustomJS(code="foo();", args={'x': 3})

    with raises(AttributeError):  # kwargs not supported
        CustomJS(code="foo();", x=slider)


def test_py_callback():

    slider = Slider()
    foo = None  # fool pyflakes

    def cb(x=slider):
        foo()
    cb = CustomJS.from_py_func(cb)
    assert cb.lang == 'javascript'
    assert 'foo()' in cb.code
    assert cb.args['x'] is slider

    with raises(ValueError):  # not a plot object
        def cb(x=4):
            foo()
        CustomJS.from_py_func(cb)
