utest/test_frontend.py

#!/usr/bin/env python
# Copyright 2013, Big Switch Networks, Inc.
#
# LoxiGen is licensed under the Eclipse Public License, version 1.0 (EPL), with
# the following special exception:
#
# LOXI Exception
#
# As a special exception to the terms of the EPL, you may distribute libraries
# generated by LoxiGen (LoxiGen Libraries) under the terms of your choice, provided
# that copyright and licensing notices generated by LoxiGen are not altered or removed
# from the LoxiGen Libraries and the notice provided below is (i) included in
# the LoxiGen Libraries, if distributed in source code form and (ii) included in any
# documentation for the LoxiGen Libraries, if distributed in binary form.
#
# Notice: "Copyright 2013, Big Switch Networks, Inc. This library was generated by the LoxiGen Compiler."
#
# You may not use this file except in compliance with the EPL or LOXI Exception. You may obtain
# a copy of the EPL at:
#
# http://www.eclipse.org/legal/epl-v10.html
#
# 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
# EPL for the specific language governing permissions and limitations
# under the EPL.

import sys
import os
import unittest

root_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..')
sys.path.insert(0, root_dir)

import loxi_front_end.parser as parser
import loxi_front_end.frontend as frontend
from loxi_ir import *

class FrontendTests(unittest.TestCase):
    maxDiff = None

    def test_simple(self):
        ast = parser.parse("""
#version 1

enum ofp_port_config {
    OFPPC_PORT_DOWN = 0x1,
    OFPPC_NO_STP = 0x2,
    OFPPC_NO_RECV = 0x4,
    OFPPC_NO_RECV_STP = 0x8,
    OFPPC_NO_FLOOD = 0x10,
    OFPPC_NO_FWD = 0x20,
    OFPPC_NO_PACKET_IN = 0x40,
};

#version 2

struct of_echo_reply {
    uint8_t version;
    uint8_t type == 3;
    uint16_t length;
    uint32_t xid;
    of_octets_t data;
};

enum ofp_queue_op_failed_code {
    OFPQOFC_BAD_PORT = 0,
    OFPQOFC_BAD_QUEUE = 1,
    OFPQOFC_EPERM = 2,
};

struct of_packet_queue {
    uint32_t queue_id;
    uint16_t len;
    pad(2);
    list(of_queue_prop_t) properties;
};
""")

        # Not testing the parser, just making sure the AST is what we expect
        expected_ast = [
            ['metadata', 'version', '1'],
            ['enum', 'ofp_port_config', [
                ['OFPPC_PORT_DOWN', 1],
                ['OFPPC_NO_STP', 2],
                ['OFPPC_NO_RECV', 4],
                ['OFPPC_NO_RECV_STP', 8],
                ['OFPPC_NO_FLOOD', 16],
                ['OFPPC_NO_FWD', 32],
                ['OFPPC_NO_PACKET_IN', 64]]],
            ['metadata', 'version', '2'],
            ['struct', 'of_echo_reply', None, [
                ['data', 'uint8_t', 'version'],
                ['type', 'uint8_t', 'type', 3],
                ['data', 'uint16_t', 'length'],
                ['data', 'uint32_t', 'xid'],
                ['data', 'of_octets_t', 'data']]],
            ['enum', 'ofp_queue_op_failed_code', [
                ['OFPQOFC_BAD_PORT', 0],
                ['OFPQOFC_BAD_QUEUE', 1],
                ['OFPQOFC_EPERM', 2]]],
            ['struct', 'of_packet_queue', None, [
                ['data', 'uint32_t', 'queue_id'],
                ['data', 'uint16_t', 'len'],
                ['pad', 2],
                ['data', 'list(of_queue_prop_t)', 'properties']]],
        ]
        self.assertEquals(expected_ast, ast)

        ofinput = frontend.create_ofinput(ast)
        self.assertEquals(set([1, 2]), ofinput.wire_versions)
        expected_classes = [
            OFClass('of_echo_reply', None, [
                OFDataMember('version', 'uint8_t'), # XXX
                OFTypeMember('type', 'uint8_t', 3),
                OFLengthMember('length', 'uint16_t'),
                OFDataMember('xid', 'uint32_t'),
                OFDataMember('data', 'of_octets_t')]),
            OFClass('of_packet_queue', None, [
                OFDataMember('queue_id', 'uint32_t'),
                OFLengthMember('len', 'uint16_t'),
                OFPadMember(2),
                OFDataMember('properties', 'list(of_queue_prop_t)')]),
        ]
        self.assertEquals(expected_classes, ofinput.classes)
        expected_enums = [
            OFEnum('ofp_port_config', [
                ('OFPPC_PORT_DOWN', 1),
                ('OFPPC_NO_STP', 2),
                ('OFPPC_NO_RECV', 4),
                ('OFPPC_NO_RECV_STP', 8),
                ('OFPPC_NO_FLOOD', 16),
                ('OFPPC_NO_FWD', 32),
                ('OFPPC_NO_PACKET_IN', 64)]),
            OFEnum('ofp_queue_op_failed_code', [
                ('OFPQOFC_BAD_PORT', 0),
                ('OFPQOFC_BAD_QUEUE', 1),
                ('OFPQOFC_EPERM', 2)]),
        ]
        self.assertEquals(expected_enums, ofinput.enums)

    def test_inheritance(self):
        ast = parser.parse("""
#version 1

struct of_queue_prop {
    uint16_t type;
    uint16_t len;
    pad(4);
};

struct of_queue_prop_min_rate : of_queue_prop {
    uint16_t type == 1;
    uint16_t len;
    pad(4);
    uint16_t rate;
    pad(6);
};
""")

        # Not testing the parser, just making sure the AST is what we expect
        expected_ast = [
            ['metadata', 'version', '1'],

            ['struct', 'of_queue_prop', None, [
                ['data', 'uint16_t', 'type'],
                ['data', 'uint16_t', 'len'],
                ['pad', 4]]],

            ['struct', 'of_queue_prop_min_rate', 'of_queue_prop', [
                ['type', 'uint16_t', 'type', 1],
                ['data', 'uint16_t', 'len'],
                ['pad', 4],
                ['data', 'uint16_t', 'rate'],
                ['pad', 6]]],
        ]
        self.assertEquals(expected_ast, ast)

        ofinput = frontend.create_ofinput(ast)
        expected_classes = [
            OFClass('of_queue_prop', None, [
                OFDataMember('type', 'uint16_t'),
                OFLengthMember('len', 'uint16_t'),
                OFPadMember(4)]),
            OFClass('of_queue_prop_min_rate', 'of_queue_prop', [
                OFTypeMember('type', 'uint16_t', 1),
                OFLengthMember('len', 'uint16_t'),
                OFPadMember(4),
                OFDataMember('rate', 'uint16_t'),
                OFPadMember(6)]),
        ]
        self.assertEquals(expected_classes, ofinput.classes)

if __name__ == '__main__':
    unittest.main()