test_b2help_recommendation.py

 
 
'''
Test for checking:
    - b2help-recommendation
    - b2help-recommendation -t gamma
'''
 
import basf2
import subprocess
import shlex
 
 
def call_command(command):
    """Call command print output after having removed some lines from output
 
    Args:
        command (string): command to be called
    """
    output = subprocess.check_output(shlex.split(command), encoding='utf-8').strip().split('\n')
    for line in output:
        if "provider = " not in line:
            print(line)
 
 
if __name__ == '__main__':
    check_tags = ['all', 'gamma', 'Tracking']
    filename = basf2.find_file('framework/tests/test_b2help_recommendation_payload.json')
    # List available tags
    call_command(f'b2help-recommendation -l {filename}')
    # Call the script for tags
    for tag in check_tags:
        call_command(f'b2help-recommendation -t {tag} -l {filename}')
    # Call convert to payload
    call_command(f'b2help-recommendation -l {filename} -t {check_tags[1]} -c -p test')