1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
# Copyright (C) 2016. BMW Car IT GmbH. All rights reserved.
"""Basic unittests for DLTClient class"""
import unittest
from unittest.mock import patch, Mock
from dlt.dlt import DLTClient, DLT_RETURN_OK, DLT_RETURN_ERROR
class TestDLTClient(unittest.TestCase):
def setUp(self):
# - patch port so that connect fails even if dlt-daemon is running
self.client = DLTClient(servIP="127.0.0.1", port=424242)
def test_connect_with_timeout_failed(self):
# - timeout error
self.assertFalse(self.client.connect(timeout=2))
# - dlt_receiver_init error
with patch("socket.create_connection", return_value=Mock(fileno=Mock(return_value=2000000))), patch(
"dlt.dlt.dltlib.dlt_receiver_init", return_value=DLT_RETURN_ERROR
):
self.assertFalse(self.client.connect(timeout=2))
def test_connect_with_timeout_success(self):
with patch("socket.create_connection", return_value=Mock(fileno=Mock(return_value=2000000))), patch(
"dlt.dlt.dltlib.dlt_receiver_init", return_value=DLT_RETURN_OK
):
self.assertTrue(self.client.connect(timeout=2))
|
