diff options
Diffstat (limited to 'tests/test_dhcp.py')
-rw-r--r-- | tests/test_dhcp.py | 68 |
1 files changed, 0 insertions, 68 deletions
diff --git a/tests/test_dhcp.py b/tests/test_dhcp.py deleted file mode 100644 index 2206325..0000000 --- a/tests/test_dhcp.py +++ /dev/null @@ -1,68 +0,0 @@ -# Copyright 2014 Microsoft Corporation -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. -# -# Requires Python 2.4+ and Openssl 1.0+ -# -# Implements parts of RFC 2131, 1541, 1497 and -# http://msdn.microsoft.com/en-us/library/cc227282%28PROT.10%29.aspx -# http://msdn.microsoft.com/en-us/library/cc227259%28PROT.13%29.aspx - -import tests.env as env -from tests.tools import * -import uuid -import unittest -import os -import json -import azurelinuxagent.utils.fileutil as fileutil -import azurelinuxagent.distro.default.dhcp as dhcp_handler - -SampleDhcpResponse = None -with open(os.path.join(env.test_root, "dhcp"), 'rb') as F: - SampleDhcpResponse = F.read() - -mock_socket_send = MockFunc('socket_send', SampleDhcpResponse) -mock_gen_trans_id = MockFunc('gen_trans_id', "\xC6\xAA\xD1\x5D") -mock_get_mac_addr = MockFunc('get_mac_addr', "\x00\x15\x5D\x38\xAA\x38") -mock_send_dhcp_failed = MockFunc(retval=None) - -class TestdhcpHandler(unittest.TestCase): - - def test_build_dhcp_req(self): - req = dhcp_handler.build_dhcp_request(mock_get_mac_addr()) - self.assertNotEquals(None, req) - - @mock(dhcp_handler, "gen_trans_id", mock_gen_trans_id) - @mock(dhcp_handler, "socket_send", mock_socket_send) - def test_send_dhcp_req(self): - req = dhcp_handler.build_dhcp_request(mock_get_mac_addr()) - resp = dhcp_handler.send_dhcp_request(req) - self.assertNotEquals(None, resp) - - @mock(dhcp_handler, "send_dhcp_request", mock_send_dhcp_failed) - def test_send_dhcp_failed(self): - dhcp = dhcp_handler.DhcpHandler() - dhcp.probe() - - @mock(dhcp_handler, "socket_send", mock_socket_send) - @mock(dhcp_handler, "gen_trans_id", mock_gen_trans_id) - @mock(dhcp_handler.OSUTIL, "get_mac_addr", mock_get_mac_addr) - @mock(dhcp_handler.fileutil, "write_file", MockFunc()) - def test_handle_dhcp(self): - dh = dhcp_handler.DhcpHandler() - dh.probe() - self.assertEquals("10.62.144.1", dh.gateway) - self.assertEquals("10.62.144.140", dh.endpoint) - -if __name__ == '__main__': - unittest.main() |