mirror of
https://github.com/eblot/pybootd.git
synced 2024-09-11 22:17:44 +03:00
Add a unit test to reproduce #16
This commit is contained in:
parent
35d930e9c1
commit
5cc8737c23
148
pybootd/tests/simple.py
Executable file
148
pybootd/tests/simple.py
Executable file
@ -0,0 +1,148 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Copyright (c) 2020, Emmanuel Blot <emmanuel.blot@free.fr>
|
||||
# All rights reserved.
|
||||
#
|
||||
# This library is free software; you can redistribute it and/or
|
||||
# modify it under the terms of the GNU Lesser General Public
|
||||
# License as published by the Free Software Foundation; either
|
||||
# version 2 of the License, or (at your option) any later version.
|
||||
#
|
||||
# This library 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
|
||||
# Lesser General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Lesser General Public
|
||||
# License along with this library; if not, write to the Free Software
|
||||
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
|
||||
#pylint: disable-msg=empty-docstring
|
||||
#pylint: disable-msg=missing-docstring
|
||||
#pylint: disable-msg=invalid-name
|
||||
#pylint: disable-msg=global-statement
|
||||
|
||||
from binascii import hexlify, unhexlify
|
||||
from io import StringIO
|
||||
from socket import (socket, timeout, AF_INET, SOCK_DGRAM, IPPROTO_UDP,
|
||||
SOL_SOCKET, SO_BROADCAST, SO_REUSEADDR)
|
||||
from sys import modules, stdout
|
||||
from textwrap import fill
|
||||
from time import sleep
|
||||
from unittest import TestCase, TestSuite, SkipTest, makeSuite, main as ut_main
|
||||
from pybootd.daemons import BootpDaemon
|
||||
from pybootd.util import EasyConfigParser, logger_factory
|
||||
|
||||
|
||||
class PxeSimpleTestCase(TestCase):
|
||||
"""Simple PXE test case.
|
||||
"""
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
cls.logger = logger_factory(logtype='stdout',
|
||||
level='DEBUG')
|
||||
cfgparser = EasyConfigParser()
|
||||
config = StringIO("""
|
||||
[bootpd]
|
||||
address = 0.0.0.0
|
||||
pool_start = 127.0.0.100
|
||||
pool_count = 5
|
||||
servername = localhost
|
||||
domain = localdomain
|
||||
server_name = debug
|
||||
lease_time = 86400
|
||||
access = mac
|
||||
allow_simple_dhcp = enable
|
||||
|
||||
[bootfile]
|
||||
; BIOS boot file
|
||||
default = pxelinux.0
|
||||
|
||||
[mac]
|
||||
00-0A-86-A2-2F-3E = enable
|
||||
""")
|
||||
cfgparser.read_file(config)
|
||||
cls.config = cfgparser
|
||||
|
||||
def setUp(self):
|
||||
self.server = BootpDaemon(logger=self.logger, config=self.config,
|
||||
debug=True)
|
||||
self.sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)
|
||||
self.sock.setsockopt(SOL_SOCKET, SO_BROADCAST, 1)
|
||||
self.sock.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
|
||||
self.sock.settimeout(1)
|
||||
self.sock.bind(('', 68))
|
||||
self.server.start()
|
||||
# be sure the server can be scheduled and started before resuming
|
||||
sleep(0.1)
|
||||
|
||||
def tearDown(self):
|
||||
if self.sock:
|
||||
self.sock.close()
|
||||
self.server.stop()
|
||||
|
||||
def test(self):
|
||||
"""Doc.
|
||||
"""
|
||||
discover = """
|
||||
010106002233445a0000800000000000000000000000000000000000000a86a22f3e
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000006382
|
||||
5363ff00000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000
|
||||
"""
|
||||
|
||||
request = """
|
||||
010106002233445a0000800000000000000000000000000000000000000a86a22f3e
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
0000000000003c24496e7465724e6963686520506f727461626c65205443502f4950
|
||||
2c2076302e3042657461000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000ff6382
|
||||
53633501033204c0a801190104ffffff000304c0a8013a3304000151803704010306
|
||||
0f3604c0a8013a5106696e69636865000000000000000000340101ff000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000000000000000000000000000000000000000000000000000000000000000
|
||||
00000000
|
||||
"""
|
||||
for hexdata in (discover, request):
|
||||
req = unhexlify(hexdata.replace(' ', '').replace('\n', ''))
|
||||
self.sock.sendto(req, ('<broadcast>', 67))
|
||||
try:
|
||||
resp = self.sock.recv(1024)
|
||||
print('response:\n',
|
||||
fill(hexlify(resp).decode(),
|
||||
initial_indent=' ',
|
||||
subsequent_indent=' '))
|
||||
except timeout:
|
||||
print('Timeout')
|
||||
|
||||
def suite():
|
||||
suite_ = TestSuite()
|
||||
suite_.addTest(makeSuite(PxeSimpleTestCase, 'test'))
|
||||
return suite_
|
||||
|
||||
|
||||
def main():
|
||||
import doctest
|
||||
doctest.testmod(modules[__name__])
|
||||
try:
|
||||
ut_main(defaultTest='suite')
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
@ -24,7 +24,7 @@ from re import match
|
||||
from socket import inet_aton, inet_ntoa
|
||||
from subprocess import run
|
||||
from struct import pack as spack, unpack as sunpack
|
||||
from sys import platform, stderr
|
||||
from sys import platform, stderr, stdout
|
||||
|
||||
try:
|
||||
import netifaces as nif
|
||||
@ -138,6 +138,8 @@ def logger_factory(logtype='syslog', logfile=None, level='WARNING',
|
||||
hdlr = SysLogHandler('/dev/log')
|
||||
elif logtype in ('stderr'):
|
||||
hdlr = StreamHandler(stderr)
|
||||
elif logtype in ('stdout'):
|
||||
hdlr = StreamHandler(stdout)
|
||||
else:
|
||||
hdlr = BufferingHandler(0)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user