def test_consolidate_query_filter(self): session_factory = self.replay_flight_data( 'test_emr_query_ids') ctx = Bag( session_factory=session_factory, log_dir='', options=Config.empty()) query = { 'query': [ {'tag:foo': 'val1'}, {'tag:foo': 'val2'}, {'tag:bar': 'val3'} ] } mgr = emr.EMRCluster(ctx, query) self.assertEqual( mgr.consolidate_query_filter(), [ {'Values': ['val1', 'val2'], 'Name': 'tag:foo'}, {'Values': ['val3'], 'Name': 'tag:bar'}, # default query { 'Values': ['WAITING', 'RUNNING', 'BOOTSTRAPPING'], 'Name': 'ClusterStates', }, ] ) query = { 'query': [ {'tag:foo': 'val1'}, {'tag:foo': 'val2'}, {'tag:bar': 'val3'}, {'ClusterStates': 'terminated'}, ] } mgr = emr.EMRCluster(ctx, query) self.assertEqual( mgr.consolidate_query_filter(), [ {'Values': ['val1', 'val2'], 'Name': 'tag:foo'}, {'Values': ['val3'], 'Name': 'tag:bar'}, # verify default is overridden { 'Values': ['terminated'], 'Name': 'ClusterStates', }, ] )
def test_consolidate_query_filter(self): session_factory = self.replay_flight_data('test_emr_query_ids') ctx = Bag(session_factory=session_factory, log_dir='', options=Config.empty()) query = { 'query': [{ 'tag:foo': 'val1' }, { 'tag:foo': 'val2' }, { 'tag:bar': 'val3' }] } mgr = emr.EMRCluster(ctx, query) self.assertEqual(mgr.consolidate_query_filter(), [{ 'Values': ['val1', 'val2'], 'Name': 'tag:foo' }, { 'Values': ['val3'], 'Name': 'tag:bar' }])
def test_get_emr_by_ids(self): session_factory = self.replay_flight_data("test_emr_query_ids") ctx = Bag(session_factory=session_factory, log_dir="", options=Config.empty()) mgr = emr.EMRCluster(ctx, {}) resources = mgr.get_resources(["j-1EJMJNTXC63JW"]) self.assertEqual(resources[0]["Id"], "j-1EJMJNTXC63JW")
def test_consolidate_query_filter(self): session_factory = self.replay_flight_data("test_emr_query_ids") ctx = Bag(session_factory=session_factory, log_dir="", options=Config.empty()) query = { "query": [{"tag:foo": "val1"}, {"tag:foo": "val2"}, {"tag:bar": "val3"}] } mgr = emr.EMRCluster(ctx, query) self.assertEqual( mgr.consolidate_query_filter(), [ {"Values": ["val1", "val2"], "Name": "tag:foo"}, {"Values": ["val3"], "Name": "tag:bar"}, # default query { "Values": ["WAITING", "RUNNING", "BOOTSTRAPPING"], "Name": "ClusterStates", }, ], ) query = { "query": [ {"tag:foo": "val1"}, {"tag:foo": "val2"}, {"tag:bar": "val3"}, {"ClusterStates": "terminated"}, ] } mgr = emr.EMRCluster(ctx, query) self.assertEqual( mgr.consolidate_query_filter(), [ {"Values": ["val1", "val2"], "Name": "tag:foo"}, {"Values": ["val3"], "Name": "tag:bar"}, # verify default is overridden {"Values": ["terminated"], "Name": "ClusterStates"}, ], )