(self, new_state)
| 494 | return default |
| 495 | |
| 496 | def _set_state(self, new_state): |
| 497 | allowed = False |
| 498 | |
| 499 | if new_state == SSLProtocolState.UNWRAPPED: |
| 500 | allowed = True |
| 501 | |
| 502 | elif ( |
| 503 | self._state == SSLProtocolState.UNWRAPPED and |
| 504 | new_state == SSLProtocolState.DO_HANDSHAKE |
| 505 | ): |
| 506 | allowed = True |
| 507 | |
| 508 | elif ( |
| 509 | self._state == SSLProtocolState.DO_HANDSHAKE and |
| 510 | new_state == SSLProtocolState.WRAPPED |
| 511 | ): |
| 512 | allowed = True |
| 513 | |
| 514 | elif ( |
| 515 | self._state == SSLProtocolState.WRAPPED and |
| 516 | new_state == SSLProtocolState.FLUSHING |
| 517 | ): |
| 518 | allowed = True |
| 519 | |
| 520 | elif ( |
| 521 | self._state == SSLProtocolState.FLUSHING and |
| 522 | new_state == SSLProtocolState.SHUTDOWN |
| 523 | ): |
| 524 | allowed = True |
| 525 | |
| 526 | if allowed: |
| 527 | self._state = new_state |
| 528 | |
| 529 | else: |
| 530 | raise RuntimeError( |
| 531 | 'cannot switch state from {} to {}'.format( |
| 532 | self._state, new_state)) |
| 533 | |
| 534 | # Handshake flow |
| 535 |
no test coverage detected