root/umpa/branches/protocols/tests/a_unit/test_protocols/test_icmp.py @ 5802

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Copyright (C) 2010 Adriano Monteiro Marques.
#
# Author: Kosma Moczek <kosma at kosma dot pl>
#
# 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.1 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 

import py.test
from umit.umpa.protocols import ICMP, Payload

class TestICMP(object):

    # test if the active fields are set properly for supported ICMP types
    def test_active_fields(self):
        i = ICMP()
        # default - no type
        for name in ( 'ident', 'seq', 'redir_gw', 'pointer', 'pointer_unused',
                      'addressmask', 'originate', 'receive', 'transmit' ):
            assert i.get_field(name).active == False
        for name in ( 'unused', 'data', ):
            assert i.get_field(name).active == True
        # id+seq types
        for type in ( 0, 8, 13, 14, 15, 16, 17, 18 ):
            i.type = type
            for name in ( 'redir_gw', 'pointer', 'pointer_unused', 'unused', ):
                assert i.get_field(name).active == False
            for name in ( 'ident', 'seq', ):
                assert i.get_field(name).active == True
        # pointer+unused3 types
        for type in ( 12, ):
            i.type = type
            for name in ( 'ident', 'seq', 'redir_gw', 'unused', ):
                assert i.get_field(name).active == False
            for name in ( 'pointer', 'pointer_unused', ):
                assert i.get_field(name).active == True
        # GIA types
        for type in ( 5, ):
            i.type = type
            for name in ( 'ident', 'seq', 'pointer', 'pointer_unused', 'unused', ):
                assert i.get_field(name).active == False
            for name in ( 'redir_gw', ):
                assert i.get_field(name).active == True
        # timestamp data fields
        for type in ( 13, 14 ):
            i.type = type
            for name in ( 'data', 'addressmask', ):
                assert i.get_field(name).active == False
            for name in ( 'originate', 'receive', 'transmit', ):
                assert i.get_field(name).active == True
        # address mask request/response
        for type in ( 17, 18 ):
            i.type = type
            for name in ( 'originate', 'receive', 'transmit', 'data' ):
                assert i.get_field(name).active == False
            for name in ( 'addressmask', ):
                assert i.get_field(name).active == True

#    # this tests pre/post raw methods as well
#    def test_get_raw(self):
#        ip = IP(src='1.2.3.4', dst='5.6.7.8')
#        udp = UDP(srcport=9, dstport=10)
#
#        assert udp._raw(0, 0, [ip, udp], 0) == (0x9000A0008EFB7, 64)
#
#    # test length and checksum calculation
#    def test_get_raw_with_payload(self):
#        ip = IP(src='1.2.3.4', dst='5.6.7.8')
#        udp = UDP(srcport=9, dstport=10)
#        payload = Payload('12345678')
#
#        # the next two lines are normally done by Packet.get_raw()
#        payload.get_raw([ip, udp, payload], 8*len(payload.data))
#        udp.__dict__['payload'] = payload
#
#        assert udp._raw(0, 0, [ip, udp, payload], 8*len(payload.data)) == \
#                                                       (0x9000a00101ed3L, 64)