Python TCPConnection.change_state Examples

Example #1

File: Test.py Project: goagile/gof

class Test(unittest.TestCase):
    def setUp(self):
        self.tcp = TCPConnection()

    def test_init_TCPConnection(self):
        self.assertTrue(isinstance(self.tcp.state, TCPClosed))

    def test_change_state(self):
        self.tcp.change_state(TCPEstablished())
        self.assertTrue(isinstance(self.tcp.state, TCPEstablished))

    def test_open(self):
        self.tcp.open()
        self.assertTrue(isinstance(self.tcp.state, TCPEstablished))

Example #2

File: Test.py Project: khardi/design_patterns

class Test(unittest.TestCase):

    def setUp(self):
        self.tcp = TCPConnection()

    def test_init_TCPConnection(self):
        self.assertTrue(isinstance(self.tcp.state, TCPClosed))

    def test_change_state(self):
        self.tcp.change_state(TCPEstablished())
        self.assertTrue(isinstance(self.tcp.state, TCPEstablished))

    def test_open(self):
        self.tcp.open()
        self.assertTrue(isinstance(self.tcp.state, TCPEstablished))