Module: check_mk
Branch: master
Commit: 7281142be4d4382769f47652130868bbf8588068
URL:
http://git.mathias-kettner.de/git/?p=check_mk.git;a=commit;h=7281142be4d438…
Author: Lars Michelsen <lm(a)mathias-kettner.de>
Date: Wed Sep 27 10:05:16 2017 +0200
First try to add some unit test like mkeventd test
Change-Id: I615545e33a28f395c3cd7cf1d14e9428728230cc
---
tests/mkeventd/test_mkeventd_status.py | 27 +++++++++++++++++++++++----
1 file changed, 23 insertions(+), 4 deletions(-)
diff --git a/tests/mkeventd/test_mkeventd_status.py
b/tests/mkeventd/test_mkeventd_status.py
index 1d0fff5..c18e1dd 100644
--- a/tests/mkeventd/test_mkeventd_status.py
+++ b/tests/mkeventd/test_mkeventd_status.py
@@ -3,6 +3,7 @@
import pytest
import time
+import ast
from testlib import web, ec, cmk_path
#
@@ -17,24 +18,42 @@ def test_mkeventd_unit():
def test_handle_client(monkeypatch):
- class FakeSocket(object):
+ class FakeQueries(object):
def __init__(self, sock):
+ self._sent = False
pass
def __iter__(self):
return self
def next(self):
+ if self._sent:
+ raise StopIteration()
+
+ self._sent = True
return ["GET events"]
+
+ class FakeSocket(object):
+ def sendall(self, data):
+ if data == "\n":
+ return
+ response = ast.literal_eval(data)
+ assert type(response) == list
+ assert len(response) == 1
+ assert "event_id" in response[0]
+
def close(self):
pass
- monkeypatch.setattr(mkeventd, "Queries", FakeSocket)
+
+ monkeypatch.setattr(mkeventd, "Queries", FakeQueries)
monkeypatch.setattr(mkeventd.StatusServer, "open_sockets", lambda x: None)
- status = mkeventd.StatusServer()
- print status.handle_client(None, True, "127.0.0.1")
+ mkeventd.g_status_server = mkeventd.StatusServer()
+ mkeventd.g_event_status = mkeventd.EventStatus()
+
+ mkeventd.g_status_server.handle_client(FakeSocket(), True, "127.0.0.1")
#