def test_dashboard_delete(quicksight_application_stub, mininmal_data_sets_stub, template_arn): obj = Dashboard(quicksight_application=quicksight_application_stub, data_sets=mininmal_data_sets_stub.data_sets_stub, quicksight_template_arn=template_arn, props=None) sub_type = 'main' dump_state(obj, 'Before delete') DashboardStubber.stub_delete_dashboard_call(sub_type) obj.delete() dump_state(obj, 'After delete')
def buildboard_startup(cfg): dashbrd = Dashboard(cfg) name = dashbrd.getName() branch = dashbrd.getBranch() logger.info("name: {0} branch: {1}".format(name, branch)) # Initialize git repo repo = dashbrd.getRepo() gitProj = dashbrd.getBuildProject() if gitProj: if not os.path.exists(gitProj): check_call(["git", "clone", repo + '/' + gitProj]) gitRepo = Repo(gitProj) if gitRepo: dashbrd.setPyGitRepo(gitRepo) else: print("Error: Unable to clone {0}".format(cfg['gitProj'])) return None else: print("Error: Need to specify the github project") return None # # Add build product to dashboard.html # ret = html_update_dashboard_table(cfg['name']) if ret: # Load last build from DB and update dashboard instance # Load last build from DB and start monitoring # Do nothing for now print("DEBUG: Created table {0} in Dashboard".format(cfg['name'])) else: print("WARNING: Can't add {0} to Dashboard".format(cfg['name'])) return None return dashbrd
def test_dashboard_update_source_entity(quicksight_application_stub, mininmal_data_sets_stub, template_arn): obj = Dashboard(quicksight_application=quicksight_application_stub, data_sets=mininmal_data_sets_stub.data_sets_stub, quicksight_template_arn=template_arn, props=None) sub_type = 'main' assert sub_type in obj.config_data assert 'SourceEntity' in obj.config_data[sub_type] source_entity = obj.source_entity._get_map(sub_type, "SourceEntity") dump_state(source_entity, 'Dump SourceEntity before update') assert 'SourceTemplate' in source_entity source_template = source_entity.get('SourceTemplate', None) assert 'DataSetReferences' in source_template assert 'Arn' in source_template obj.source_entity._update_source_entity(source_entity) dump_state(source_entity, 'Dump SourceEntity after update') assert template_arn == source_template['Arn']
def buildboard_startup(cfg): dashbrd = Dashboard(cfg) name = dashbrd.getName() branch = dashbrd.getBranch() logger.info("name: {0} branch: {1}".format(name, branch)) # Initialize git repo repo = dashbrd.getRepo() gitProj = dashbrd.getBuildProject() if gitProj: if not os.path.exists(gitProj): check_call(["git", "clone", repo+'/'+ gitProj]) gitRepo = Repo(gitProj) if gitRepo: dashbrd.setPyGitRepo(gitRepo) else: print ("Error: Unable to clone {0}".format(cfg['gitProj'])) return None else: print ("Error: Need to specify the github project") return None # # Add build product to dashboard.html # ret = html_update_dashboard_table(cfg['name']) if ret: # Load last build from DB and update dashboard instance # Load last build from DB and start monitoring # Do nothing for now print ("DEBUG: Created table {0} in Dashboard".format(cfg['name'])) else: print ("WARNING: Can't add {0} to Dashboard".format(cfg['name'])) return None return dashbrd
def test_dashboard_init(quicksight_application_stub, mininmal_data_sets_stub): obj = Dashboard(quicksight_application=quicksight_application_stub, data_sets=mininmal_data_sets_stub.data_sets_stub, quicksight_template_arn=template_arn, props=None) dump_state(obj, 'Dump dashboard')
def __init__(self, resource_properties): # TODO: use the config data / file to figure out the supported_data_set_types supported_data_set_types = ['sentiment', 'image-text', 'text', 'topic'] self.resource_properties = resource_properties self.global_state = get_global_state() # use config data file if provided config_file = resource_properties.get('ConfigDataFile', None) if config_file: data = read_config(config_file) self.global_state.update(data) # TODO:RENAME: 'StackName' key name as this could be used out of stack. ApplicationName? # TODO: create/use a uuid if no attack provided self.prefix = resource_properties.get('StackName', 'Sample_Sol') # TODO:RENAME: quicksight_template_arn -> quicksight_source_template_arn self.quicksight_template_arn = resource_properties.get( 'QuickSightSourceTemplateArn', 'Uninitialized QuickSightSourceTemplateArn') logger.debug( f'Using QuickSightSourceTemplateArn: {self.quicksight_template_arn }' ) self.quicksight_principal_arn = resource_properties.get( 'QuickSightPrincipalArn', 'Uninitialized QuickSightPrincipalArn') logger.debug( f'Using QuickSightPrincipalArn: {self.quicksight_principal_arn }') self.data_source = DataSource(quicksight_application=self, props=self.global_state) self.data_source.athena_workgroup = resource_properties.get( 'WorkGroupName', 'primary') self.data_set_sub_types = supported_data_set_types self.data_sets = dict() for data_set_sub_type in self.data_set_sub_types: data_set = DataSet(quicksight_application=self, data_source=self.data_source, data_set_sub_type=data_set_sub_type, data_set_name=f"_{data_set_sub_type}_dataset", props=self.global_state) self.data_sets[data_set_sub_type] = data_set self.analysis = Analysis( quicksight_application=self, data_sets=self.data_sets, quicksight_template_arn=self.quicksight_template_arn, data_source=self.data_source, props=self.global_state) self.dashboard = Dashboard( quicksight_application=self, data_sets=self.data_sets, quicksight_template_arn=self.quicksight_template_arn, props=self.global_state) self.template = Template(quicksight_application=self, data_sets=self.data_sets, props=self.global_state) global_state_json = json.dumps(self.global_state, indent=2, sort_keys=True) logger.debug( f'QuicksightApi: after init, global data json: {global_state_json}' )
def __init__(self, resource_properties): # TODO: use the config data / file to figure out the supported_data_set_types supported_data_set_types = [ "image-text", "topic", "image-moderation-label", "geo", "topic-mapping", "feed" ] self.resource_properties = resource_properties self.global_state = get_global_state() # use config data file if provided config_file = resource_properties.get("ConfigDataFile", None) if config_file: data = read_config(config_file) self.global_state.update(data) self.prefix = resource_properties.get("StackName", "Sample_Sol") # TODO:RENAME: quicksight_template_arn -> quicksight_source_template_arn self.quicksight_template_arn = resource_properties.get( "QuickSightSourceTemplateArn", "Uninitialized QuickSightSourceTemplateArn") logger.debug( f"Using QuickSightSourceTemplateArn: {self.quicksight_template_arn }" ) self.quicksight_principal_arn = resource_properties.get( "QuickSightPrincipalArn", "Uninitialized QuickSightPrincipalArn") logger.debug( f"Using QuickSightPrincipalArn: {self.quicksight_principal_arn }") self.data_source = DataSource(quicksight_application=self, props=self.global_state) self.data_source.athena_workgroup = resource_properties.get( "WorkGroupName", "primary") self.data_set_sub_types = supported_data_set_types self.data_sets = dict() for data_set_sub_type in self.data_set_sub_types: data_set = DataSet( quicksight_application=self, data_source=self.data_source, data_set_sub_type=data_set_sub_type, props=self.global_state, ) self.data_sets[data_set_sub_type] = data_set self.analysis = Analysis( quicksight_application=self, data_sets=self.data_sets, quicksight_template_arn=self.quicksight_template_arn, data_source=self.data_source, props=self.global_state, ) self.dashboard = Dashboard( quicksight_application=self, data_source=self.data_source, data_sets=self.data_sets, quicksight_template_arn=self.quicksight_template_arn, props=self.global_state, ) self.template = Template(quicksight_application=self, data_sets=self.data_sets, props=self.global_state) global_state_json = json.dumps(self.global_state, indent=2, sort_keys=True) logger.debug( f"QuicksightApi: after init, global data json: {global_state_json}" )