Python FakeConnection.set_state Examples

Example #1

File: channel0_tests.py Project: gaochunzy/amqp-storm

 def test_forcefully_closed_connection(self):
     connection = FakeConnection()
     connection.set_state(connection.OPEN)
     channel = Channel0(connection)
     channel._close_connection(
         Connection.Close(reply_text=b'',
                          reply_code=500)
     )
     self.assertTrue(connection.is_closed)
     self.assertRaises(AMQPConnectionError, connection.check_for_errors)

Example #2

File: channel0_tests.py Project: gaochunzy/amqp-storm

    def test_close_connection(self):
        connection = FakeConnection()
        connection.set_state(connection.OPEN)
        channel = Channel0(connection)

        self.assertTrue(connection.is_open)
        channel._close_connection(
            Connection.Close(reply_text=b'',
                             reply_code=200)
        )
        self.assertEqual(connection.exceptions, [])
        self.assertTrue(connection.is_closed)