Skip to content

Commit

Permalink
Fixed the elasticsearch and ebs_in_use issue (#805)
Browse files Browse the repository at this point in the history
  • Loading branch information
athiruma authored Jun 18, 2024
1 parent e11f058 commit 941d573
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 9 deletions.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ azure-mgmt-resource==23.0.1
azure-mgmt-subscription==3.1.1
boto3==1.26.4
botocore==1.29.4
elasticsearch==7.11.0
elasticsearch==7.17.9
elasticsearch-dsl==7.4.0
google-api-python-client==2.57.0
google-auth-httplib2==0.1.0
Expand Down
Original file line number Diff line number Diff line change
@@ -1,13 +1,11 @@
import os

import boto3
from moto import mock_ec2, mock_s3

from cloud_governance.common.clouds.aws.s3.s3_operations import S3Operations
from cloud_governance.main.environment_variables import environment_variables
from cloud_governance.policy.aws.ebs_in_use import EbsInUse

os.environ['AWS_DEFAULT_REGION'] = 'us-east-2'
from tests.unittest.configs import AWS_DEFAULT_REGION


@mock_ec2
Expand All @@ -17,8 +15,8 @@ def test_ebs_in_use():
@return:
"""
environment_variables.environment_variables_dict['policy'] = 'ebs_in_use'
region = os.environ.get('AWS_DEFAULT_REGION', 'us-east-1')
ec2_client = boto3.client('ec2', region_name=region)
environment_variables.environment_variables_dict['AWS_DEFAULT_REGION'] = AWS_DEFAULT_REGION
ec2_client = boto3.client('ec2', region_name=AWS_DEFAULT_REGION)
volume_id = ec2_client.create_volume(Size=10, AvailabilityZone='us-east-1a')['VolumeId']
default_ami_id = 'ami-03cf127a'
instance_id = ec2_client.run_instances(ImageId=default_ami_id, InstanceType='t2.micro', MaxCount=1, MinCount=1)['Instances'][0]['InstanceId']
Expand All @@ -36,8 +34,8 @@ def test_ebs_in_use_s3_upload():
@return:
"""
environment_variables.environment_variables_dict['policy'] = 'ebs_in_use'
region = os.environ.get('AWS_DEFAULT_REGION', 'us-east-1')
ec2_client = boto3.client('ec2', region_name=region)
environment_variables.environment_variables_dict['AWS_DEFAULT_REGION'] = AWS_DEFAULT_REGION
ec2_client = boto3.client('ec2', region_name=AWS_DEFAULT_REGION)
default_ami_id = 'ami-03cf127a'
ec2_client.run_instances(ImageId=default_ami_id, InstanceType='t2.micro', MaxCount=1, MinCount=1)
s3_client = boto3.client('s3', region_name='us-east-1')
Expand Down
2 changes: 1 addition & 1 deletion tests_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ azure-mgmt-costmanagement==3.0.0
azure-mgmt-monitor==6.0.2
azure-mgmt-subscription==3.1.1
boto3==1.26.4
elasticsearch==7.11.0
elasticsearch==7.17.9
elasticsearch-dsl==7.4.0
freezegun==1.5.1
moto==4.0.1
Expand Down

0 comments on commit 941d573

Please sign in to comment.