| |
| # Copyright 2017-present Open Networking Foundation |
| # |
| # 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. |
| |
| |
| # Distributed under the OpenFlow Software License (see LICENSE) |
| # Copyright (c) 2014 Big Switch Networks, Inc. |
| """ |
| Bundle test cases |
| """ |
| |
| import logging |
| |
| from oftest import config |
| import oftest.base_tests as base_tests |
| import ofp |
| |
| from oftest.testutils import * |
| |
| class Commit(base_tests.SimpleProtocol): |
| """ |
| Verify that messages added to a bundle are executed only after a commit |
| """ |
| def runTest(self): |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_OPEN_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_OPEN_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| for i in xrange(0, 10): |
| request = ofp.message.bundle_add_msg( |
| xid=i, |
| bundle_id=1, |
| data=ofp.message.echo_request(xid=i).pack()) |
| self.controller.message_send(request) |
| |
| # Make sure the messages aren't executed |
| msg, _ = self.controller.poll(ofp.message.echo_reply) |
| self.assertIsNone(msg) |
| |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_COMMIT_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_COMMIT_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| for i in xrange(0, 10): |
| response, _ = self.controller.poll(ofp.message.echo_reply) |
| self.assertIsNotNone(response) |
| self.assertEquals(response.xid, i) |
| |
| class Discard(base_tests.SimpleProtocol): |
| """ |
| Verify that messages in a discarded bundle are not executed |
| """ |
| def runTest(self): |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_OPEN_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_OPEN_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| for i in xrange(0, 10): |
| request = ofp.message.bundle_add_msg( |
| xid=i, |
| bundle_id=1, |
| data=ofp.message.echo_request(xid=i).pack()) |
| self.controller.message_send(request) |
| |
| # Make sure the messages aren't executed |
| msg, _ = self.controller.poll(ofp.message.echo_reply) |
| self.assertIsNone(msg) |
| |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_DISCARD_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_DISCARD_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| # Make sure the messages aren't executed |
| msg, _ = self.controller.poll(ofp.message.echo_reply) |
| self.assertIsNone(msg) |
| |
| class Disconnect(base_tests.SimpleProtocol): |
| """ |
| Disconnect without explicitly discarding the bundle |
| """ |
| def runTest(self): |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_OPEN_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_OPEN_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| for i in xrange(0, 10): |
| request = ofp.message.bundle_add_msg( |
| xid=i, |
| bundle_id=1, |
| data=ofp.message.echo_request(xid=i).pack()) |
| self.controller.message_send(request) |
| |
| # Disconnect without committing/discarding |
| |
| class TooManyMsgs(base_tests.SimpleProtocol): |
| """ |
| Verify that exactly 262144 messages can be added to a bundle |
| |
| This is tied to the limit in Indigo. |
| """ |
| def runTest(self): |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_OPEN_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_OPEN_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| request = ofp.message.bundle_add_msg( |
| xid=0, |
| bundle_id=1, |
| data=ofp.message.echo_request(xid=0).pack()) |
| |
| buf = request.pack() |
| |
| logging.debug("Sending bundle-add messages") |
| |
| count = 0 |
| for i in xrange(0, 256*1024): |
| with self.controller.tx_lock: |
| if self.controller.switch_socket.sendall(buf) is not None: |
| raise AssertionError("failed to send message to switch") |
| count += 1 |
| |
| logging.debug("Sent %d bundle-add messages", count) |
| |
| do_barrier(self.controller) |
| verify_no_errors(self.controller) |
| |
| request = ofp.message.bundle_add_msg( |
| xid=1, |
| bundle_id=1, |
| data=ofp.message.echo_request(xid=1).pack()) |
| self.controller.message_send(request) |
| |
| do_barrier(self.controller) |
| |
| msg, _ = self.controller.poll(exp_msg=ofp.message.error_msg) |
| self.assertIsNotNone(msg) |
| |
| class TooManyBytes(base_tests.SimpleProtocol): |
| """ |
| Verify that 50 MB of messages can be added to a bundle |
| |
| This is tied to the limit in Indigo. |
| """ |
| def runTest(self): |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_OPEN_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_OPEN_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| echo_buf = ofp.message.echo_request(xid=0, data="\x00" * 1016).pack() |
| self.assertEquals(len(echo_buf), 1024) |
| |
| request = ofp.message.bundle_add_msg( |
| xid=0, |
| bundle_id=1, |
| data=echo_buf) |
| |
| buf = request.pack() |
| |
| max_bytes = 50*1024*1024 |
| num_msgs = max_bytes / len(echo_buf) |
| |
| logging.debug("Sending bundle-add messages") |
| |
| count = 0 |
| for i in xrange(0, num_msgs): |
| with self.controller.tx_lock: |
| if self.controller.switch_socket.sendall(buf) is not None: |
| raise AssertionError("failed to send message to switch") |
| count += 1 |
| |
| logging.debug("Sent %d bundle-add messages", count) |
| |
| do_barrier(self.controller) |
| verify_no_errors(self.controller) |
| |
| request = ofp.message.bundle_add_msg( |
| xid=1, |
| bundle_id=1, |
| data=ofp.message.echo_request(xid=1).pack()) |
| self.controller.message_send(request) |
| |
| do_barrier(self.controller) |
| |
| msg, _ = self.controller.poll(exp_msg=ofp.message.error_msg) |
| self.assertIsNotNone(msg) |
| |
| class Barrier(base_tests.SimpleProtocol): |
| """ |
| Verify that barriers are allowed in a bundle |
| """ |
| def runTest(self): |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_OPEN_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_OPEN_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| request = ofp.message.bundle_add_msg( |
| xid=1, |
| bundle_id=1, |
| data=ofp.message.flow_stats_request(xid=1).pack()) |
| self.controller.message_send(request) |
| |
| request = ofp.message.bundle_add_msg( |
| xid=2, |
| bundle_id=1, |
| data=ofp.message.barrier_request(xid=2).pack()) |
| self.controller.message_send(request) |
| |
| # Make sure the messages aren't executed |
| msg, _ = self.controller.poll(ofp.message.echo_reply) |
| self.assertIsNone(msg) |
| |
| request = ofp.message.bundle_ctrl_msg( |
| bundle_ctrl_type=ofp.OFPBCT_COMMIT_REQUEST, |
| bundle_id=1) |
| |
| response, _ = self.controller.transact(request) |
| self.assertIsInstance(response, ofp.message.bundle_ctrl_msg) |
| self.assertEqual(response.bundle_ctrl_type, ofp.OFPBCT_COMMIT_REPLY) |
| self.assertEqual(response.bundle_id, 1) |
| |
| response, _ = self.controller.poll(ofp.message.flow_stats_reply) |
| self.assertIsNotNone(response) |
| self.assertEquals(response.xid, 1) |
| |
| response, _ = self.controller.poll(ofp.message.barrier_reply) |
| self.assertIsNotNone(response) |
| self.assertEquals(response.xid, 2) |