christabor / flask_jsondash

def test_no_demo_mode(monkeypatch, client):

    _, test = client

    monkeypatch.setattr(charts_builder, 'auth', auth_valid)

    data = dict(name='newname', modules=[])

    test.post(url_for('jsondash.create'), data=data, follow_redirects=True)

    view_id = read()[0]['id']

    url = url_for('jsondash.view', c_id=view_id)

    res = test.get(url)

    dom = pq(res.data)

    assert dom.find('.chart-header small > .btn')

    assert dom.find('.chart-header small > .btn').text().strip() == 'Back'

    assert dom.find('.chart-header .dropdown-toggle')

def test_demo_mode(monkeypatch, client):

    # Test that certain UI elements are removed when in demo mode.

    _, test = client

    monkeypatch.setattr(charts_builder, 'auth', auth_valid)

    assert res.status_code == 200

    assert len(read()) == 1

    view_id = read()[0]['id']

    dash_data.update(mode='grid')

    res = test.post(url_for('jsondash.update', c_id=view_id),

                    data=dash_data,

                    follow_redirects=True)

    dom = pq(res.data)

    flash_msg = ('Cannot use grid layout without specifying row(s)! '

                 'Edit JSON manually to override this.')

    assert flash_msg in dom.find('.alert-danger').text()

    assert len(read()) == 1

def test_no_demo_mode(monkeypatch, client):

    _, test = client

    monkeypatch.setattr(charts_builder, 'auth', auth_valid)

    data = dict(name='newname', modules=[])