def test_check_s3_url(self, mock_check_prefix, mock_check_bucket, mock_check_key, mock_client): mock_client.return_value = None hook = SageMakerHook() mock_check_bucket.side_effect = [False, True, True, True] mock_check_key.side_effect = [False, True, False] mock_check_prefix.side_effect = [False, True, True] with pytest.raises(AirflowException): hook.check_s3_url(data_url) with pytest.raises(AirflowException): hook.check_s3_url(data_url) assert hook.check_s3_url(data_url) is True assert hook.check_s3_url(data_url) is True
def test_check_s3_url(self, mock_check_prefix, mock_check_bucket, mock_check_key, mock_client): mock_client.return_value = None hook = SageMakerHook() mock_check_bucket.side_effect = [False, True, True, True] mock_check_key.side_effect = [False, True, False] mock_check_prefix.side_effect = [False, True, True] self.assertRaises(AirflowException, hook.check_s3_url, data_url) self.assertRaises(AirflowException, hook.check_s3_url, data_url) self.assertEqual(hook.check_s3_url(data_url), True) self.assertEqual(hook.check_s3_url(data_url), True)