From caadaa0b3539d10f2e8d1fa9fc7ec5c26010dc4b Mon Sep 17 00:00:00 2001 From: Mike Terzo Date: Mon, 23 Jan 2017 19:38:48 -0500 Subject: [PATCH] Test index with division by zero Signed-off-by: Mike Terzo --- test/test_app.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/test/test_app.py b/test/test_app.py index ddba7a7..d5bc9a2 100644 --- a/test/test_app.py +++ b/test/test_app.py @@ -164,6 +164,32 @@ def test_index_all(client, mocker, assert rv.status_code == 200 +def test_index_division_by_zero(client, mocker): + mock_puppetdb_environments(mocker) + mock_puppetdb_default_nodes(mocker) + + query_data = { + 'nodes': [[{'count': 0}]], + 'resources': [[{'count': 40}]], + } + + dbquery = MockDbQuery(query_data) + + mocker.patch.object(app.puppetdb, '_query', side_effect=dbquery.get) + + rv = client.get('/') + + assert rv.status_code == 200 + + soup = BeautifulSoup(rv.data, 'html.parser') + assert soup.title.contents[0] == 'Puppetboard' + + vals = soup.find_all('h1', + {"class": "ui header darkblue no-margin-bottom"}) + assert len(vals) == 3 + assert vals[2].string == '0' + + def test_offline_mode(client, mocker): app.app.config['OFFLINE_MODE'] = True