コード例 #1
0
ファイル: test_session.py プロジェクト: bbhunt-2020/Canivete 
def test_terminate_ssm_session(ssm_mock, instance_id):
    sess = SSMSession(instance_id=instance_id, ssm=ssm_mock)

    sess.create()
    sess.terminate()

    assert ssm_mock.terminate_session.called
コード例 #2
0
ファイル: test_session.py プロジェクト: jwarlander/aws-gate 
    def test_terminate_ssm_session(self):
        with patch.object(self.ssm,
                          'terminate_session',
                          return_value=self.response):
            sess = SSMSession(instance_id=self.instance_id, ssm=self.ssm)

            sess.create()
            sess.terminate()

            self.assertTrue(self.ssm.terminate_session.called)