slapos.networkbench: Refactor and include tests
Use json as configuration format Accept remote configuration file (url to a json) Introduce API to assert reponse codes and result for http and dns Add Unit tests.
Showing
This diff is collapsed.
slapos/networkbench/http.py
0 → 100644
slapos/networkbench/ping.py
0 → 100644
| ############################################################################## | ||
| # | ||
| # Copyright (c) 2015 Vifib SARL and Contributors. All Rights Reserved. | ||
| # | ||
| # WARNING: This program as such is intended to be used by professional | ||
| # programmers who take the whole responsibility of assessing all potential | ||
| # consequences resulting from its eventual inadequacies and bugs | ||
| # End users who are looking for a ready-to-use solution with commercial | ||
| # guarantees and support are strongly adviced to contract a Free Software | ||
| # Service Company | ||
| # | ||
| # This program is Free Software; you can redistribute it and/or | ||
| # modify it under the terms of the GNU General Public License | ||
| # as published by the Free Software Foundation; either version 3 | ||
| # of the License, or (at your option) any later version. | ||
| # | ||
| # This program is distributed in the hope that it will be useful, | ||
| # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| # GNU General Public License for more details. | ||
| # | ||
| # You should have received a copy of the GNU General Public License | ||
| # along with this program; if not, write to the Free Software | ||
| # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | ||
| # | ||
| ############################################################################## | ||
| import unittest | ||
| import os.path | ||
| from slapos.networkbench import dnsbench | ||
| from slapos.networkbench.ping import ping, ping6 | ||
| from slapos.networkbench.http import request | ||
| DNS_EXPECTED_LIST = ["85.118.38.162", "176.31.129.213"] | ||
| class TestDNSBench(unittest.TestCase): | ||
| def test_dnsbench_ok(self): | ||
| """ Test dns resolution, this test may fail if | ||
| not ran from Europe. | ||
| # Update domain later. | ||
| """ | ||
| info = dnsbench.resolve( | ||
| "www.erp5.com", DNS_EXPECTED_LIST) | ||
| self.assertEquals(info[0], 'DNS') | ||
| self.assertEquals(info[1], 'www.erp5.com') | ||
| self.assertEquals(info[2], 200) | ||
| self.assertTrue(info[3] < 1) | ||
|
||
| self.assertEquals(info[4], 'OK') | ||
| self.assertEquals(set(info[5]), set([u'85.118.38.162', u'176.31.129.213'])) | ||
| def test_dnsbench_fail(self): | ||
| """ Test dns failure resolution | ||
| """ | ||
| info = dnsbench.resolve( | ||
| "thisdomaindontexist.erp5.com") | ||
| self.assertEquals(info[0], 'DNS') | ||
| self.assertEquals(info[1], 'thisdomaindontexist.erp5.com') | ||
| self.assertEquals(info[2], 600) | ||
| self.assertTrue(info[3] < 1) | ||
| self.assertEquals(info[4], 'Cannot resolve the hostname') | ||
| self.assertEquals(info[5], []) | ||
| def test_dnsbench_unexpected(self): | ||
| """ Test dns unexpected resolution | ||
| """ | ||
| info = dnsbench.resolve( | ||
| "www.erp5.com", [DNS_EXPECTED_LIST[0]]) | ||
| self.assertEquals(info[0], 'DNS') | ||
| self.assertEquals(info[1], 'www.erp5.com') | ||
| self.assertEquals(info[2], 200) | ||
| self.assertTrue(info[3] < 1) | ||
| self.assertEquals(info[4], 'UNEXPECTED') | ||
| self.assertTrue(info[5].startswith("['85.118.38.162'] (expected) != ")) | ||
| class TestPing(unittest.TestCase): | ||
| def test_ping_ok(self): | ||
| info = ping("localhost") | ||
| self.assertEquals(info[0], 'PING') | ||
| self.assertEquals(info[1], 'localhost') | ||
| self.assertEquals(info[2], 200) | ||
| self.assertTrue(float(info[3]) < 0.2) | ||
| self.assertEquals(info[4], '0') | ||
| self.assertTrue(info[5].startswith("min")) | ||
| def test_ping_fail(self): | ||
| info = ping("couscous") | ||
| self.assertEquals(info[0], 'PING') | ||
| self.assertEquals(info[1], 'couscous') | ||
| self.assertEquals(info[2], 600) | ||
| self.assertEquals(info[3], 'failed') | ||
| self.assertEquals(info[4], -1) | ||
| self.assertEquals(info[5], 'Fail to parser ping output') | ||
| def test_ping6_ok(self): | ||
| info = ping6("localhost") | ||
| self.assertEquals(info[0], 'PING6') | ||
| self.assertEquals(info[1], 'localhost') | ||
| self.assertEquals(info[2], 200) | ||
| self.assertTrue(float(info[3]) < 0.2) | ||
| self.assertEquals(info[4], '0') | ||
| self.assertTrue(info[5].startswith("min")) | ||
| def test_ping6_fail(self): | ||
| info = ping6("couscous") | ||
| self.assertEquals(info[0], 'PING6') | ||
| self.assertEquals(info[1], 'couscous') | ||
| self.assertEquals(info[2], 600) | ||
| self.assertEquals(info[3], 'failed') | ||
| self.assertEquals(info[4], -1) | ||
| self.assertEquals(info[5], 'Fail to parser ping output') | ||
| class TestHTTPBench(unittest.TestCase): | ||
| def test_request_ok(self): | ||
| """ This test is way to badly written as it depends on | ||
| www.erp5.com for now, please replace it | ||
|
||
| """ | ||
| info = request("https://www.erp5.com", {}) | ||
| self.assertEquals(info[0], 'GET') | ||
| self.assertEquals(info[1], 'https://www.erp5.com') | ||
| self.assertEquals(info[2], 200) | ||
| self.assertEquals(len(info[3].split(';')), 5 ) | ||
| self.assertEquals(info[4], "OK") | ||
| def test_request_expected_response(self): | ||
| """ This test is way to badly written as it depends on | ||
| www.erp5.com for now, please replace it | ||
| """ | ||
| info = request("https://www.erp5.com", {"expected_response": 200}) | ||
| self.assertEquals(info[0], 'GET') | ||
| self.assertEquals(info[1], 'https://www.erp5.com') | ||
| self.assertEquals(info[2], 200) | ||
| self.assertEquals(len(info[3].split(';')), 5 ) | ||
| self.assertEquals(info[4], "OK") | ||
| def test_request_expected_redirection(self): | ||
| """ This test is way to badly written as it depends on | ||
| www.erp5.com for now, please replace it | ||
| """ | ||
| info = request("http://www.erp5.com", {"expected_response": 302}) | ||
| self.assertEquals(info[0], 'GET') | ||
| self.assertEquals(info[1], 'http://www.erp5.com') | ||
| self.assertEquals(info[2], 302) | ||
| self.assertEquals(len(info[3].split(';')), 5 ) | ||
| self.assertEquals(info[4], "OK") | ||
| def test_request_expected_text(self): | ||
| """ This test is way to badly written as it depends on | ||
| www.erp5.com for now, please replace it | ||
| """ | ||
| info = request("https://www.erp5.com", {"expected_text": "ERP5"}) | ||
| self.assertEquals(info[0], 'GET') | ||
| self.assertEquals(info[1], 'https://www.erp5.com') | ||
| self.assertEquals(info[2], 200) | ||
| self.assertEquals(len(info[3].split(';')), 5 ) | ||
| self.assertEquals(info[4], "OK") | ||
| def test_request_fail(self): | ||
| """ Test unreachable URL | ||
| """ | ||
| info = request("http://thisurldontexist.erp5.com", {}) | ||
| self.assertEquals(info[0], 'GET') | ||
| self.assertEquals(info[1], 'http://thisurldontexist.erp5.com') | ||
| self.assertEquals(info[2], 0) | ||
| self.assertEquals(len(info[3].split(';')), 5 ) | ||
| self.assertEquals(info[4], "FAIL") | ||
| def test_request_unexpected_response(self): | ||
| """ This test is way to badly written as it depends on | ||
| www.erp5.com for now, please replace it | ||
| """ | ||
| info = request("http://www.erp5.com", {"expected_response": 200}) | ||
| self.assertEquals(info[0], 'GET') | ||
| self.assertEquals(info[1], 'http://www.erp5.com') | ||
| self.assertEquals(info[2], 302) | ||
| self.assertEquals(len(info[3].split(';')), 5 ) | ||
| self.assertEquals(info[4], "UNEXPECTED (200 != 302)") | ||
| def test_request_unexpected_text(self): | ||
| """ This test is way to badly written as it depends on | ||
| www.erp5.com for now, please replace it. | ||
| """ | ||
| info = request("https://www.erp5.com", {"expected_text": "COUSCOUS"}) | ||
| self.assertEquals(info[0], 'GET') | ||
| self.assertEquals(info[1], 'https://www.erp5.com') | ||
| self.assertEquals(info[2], 200) | ||
| self.assertEquals(len(info[3].split(';')), 5 ) | ||
| self.assertEquals(info[4], "UNEXPECTED (COUSCOUS not in page content)") | ||
| #def request(url, expected_dict): | ||
| # | ||
| # rendering_time = "%s;%s;%s;%s;%s" % \ | ||
| # (curl.getinfo(curl.NAMELOOKUP_TIME), | ||
| # curl.getinfo(curl.CONNECT_TIME), | ||
| # curl.getinfo(curl.PRETRANSFER_TIME), | ||
| # curl.getinfo(curl.STARTTRANSFER_TIME), | ||
| # curl.getinfo(curl.TOTAL_TIME)) | ||
| # | ||
| # response_code = curl.getinfo(pycurl.HTTP_CODE) | ||
| # | ||
| # expected_response = expected_dict.get("expected_response", None) | ||
| # if expected_response is not None and \ | ||
| # expected_response != response_code: | ||
| # result = "UNEXPECTED (%s != %s)" % (expected_response, response_code) | ||
| # | ||
| # expected_text = expected_dict.get("expected_text", None) | ||
| # if expected_text is not None and \ | ||
| # str(expected_text) not in str(body): | ||
| # result = "UNEXPECTED (%s not in page content)" % (expected_text) | ||
| # | ||
| # | ||
| # info_list = ('GET', url, response_code, rendering_time, result) | ||
| # | ||
| # return info_list | ||
| # | ||
| if __name__ == '__main__': | ||
| unittest.main() | ||
