[Notes] [Git][BuildStream/buildstream][jonathan/workspace-fragment-guess-element] 16 commits: utils.py: Add a helper for searching upwards for files



Title: GitLab

Jonathan Maw pushed to branch jonathan/workspace-fragment-guess-element at BuildStream / buildstream

Commits:

13 changed files:

Changes:

  • NEWS
    ... ... @@ -70,6 +70,14 @@ buildstream 1.3.1
    70 70
       o Add sandbox API for command batching and use it for build, script, and
    
    71 71
         compose elements.
    
    72 72
     
    
    73
    +  o Opening a workspace now creates a .bstproject.yaml file that allows buildstream
    
    74
    +    commands to be run from a workspace that is not inside a project.
    
    75
    +
    
    76
    +  o Specifying an element is now optional for some commands when buildstream is run
    
    77
    +    from inside a workspace - the 'build', 'checkout', 'fetch', 'pull', 'push',
    
    78
    +    'shell', 'show', 'track', 'workspace close' and 'workspace reset' commands are
    
    79
    +    affected.
    
    80
    +
    
    73 81
     
    
    74 82
     =================
    
    75 83
     buildstream 1.1.5
    

  • buildstream/_context.py
    ... ... @@ -32,7 +32,7 @@ from ._message import Message, MessageType
    32 32
     from ._profile import Topics, profile_start, profile_end
    
    33 33
     from ._artifactcache import ArtifactCache
    
    34 34
     from ._artifactcache.cascache import CASCache
    
    35
    -from ._workspaces import Workspaces
    
    35
    +from ._workspaces import Workspaces, WorkspaceProjectCache
    
    36 36
     from .plugin import _plugin_lookup
    
    37 37
     
    
    38 38
     
    
    ... ... @@ -47,9 +47,13 @@ from .plugin import _plugin_lookup
    47 47
     # verbosity levels and basically anything pertaining to the context
    
    48 48
     # in which BuildStream was invoked.
    
    49 49
     #
    
    50
    +# Args:
    
    51
    +#    workspace_project_cache (WorkspaceProjectCache): A WorkspaceProjectCache
    
    52
    +#        for this invocation
    
    53
    +#
    
    50 54
     class Context():
    
    51 55
     
    
    52
    -    def __init__(self):
    
    56
    +    def __init__(self, workspace_project_cache=None):
    
    53 57
     
    
    54 58
             # Filename indicating which configuration file was used, or None for the defaults
    
    55 59
             self.config_origin = None
    
    ... ... @@ -122,6 +126,10 @@ class Context():
    122 126
             # remove a workspace directory.
    
    123 127
             self.prompt_workspace_close_remove_dir = None
    
    124 128
     
    
    129
    +        # Boolean, whether we double-check with the user that they meant to
    
    130
    +        # close the workspace when they're using it to access the project.
    
    131
    +        self.prompt_workspace_close_project_inaccessible = None
    
    132
    +
    
    125 133
             # Boolean, whether we double-check with the user that they meant to do
    
    126 134
             # a hard reset of a workspace, potentially losing changes.
    
    127 135
             self.prompt_workspace_reset_hard = None
    
    ... ... @@ -140,6 +148,7 @@ class Context():
    140 148
             self._projects = []
    
    141 149
             self._project_overrides = {}
    
    142 150
             self._workspaces = None
    
    151
    +        self._workspace_project_cache = workspace_project_cache or WorkspaceProjectCache()
    
    143 152
             self._log_handle = None
    
    144 153
             self._log_filename = None
    
    145 154
             self._cascache = None
    
    ... ... @@ -250,12 +259,15 @@ class Context():
    250 259
                 defaults, Mapping, 'prompt')
    
    251 260
             _yaml.node_validate(prompt, [
    
    252 261
                 'auto-init', 'really-workspace-close-remove-dir',
    
    262
    +            'really-workspace-close-project-inaccessible',
    
    253 263
                 'really-workspace-reset-hard',
    
    254 264
             ])
    
    255 265
             self.prompt_auto_init = _node_get_option_str(
    
    256 266
                 prompt, 'auto-init', ['ask', 'no']) == 'ask'
    
    257 267
             self.prompt_workspace_close_remove_dir = _node_get_option_str(
    
    258 268
                 prompt, 'really-workspace-close-remove-dir', ['ask', 'yes']) == 'ask'
    
    269
    +        self.prompt_workspace_close_project_inaccessible = _node_get_option_str(
    
    270
    +            prompt, 'really-workspace-close-project-inaccessible', ['ask', 'yes']) == 'ask'
    
    259 271
             self.prompt_workspace_reset_hard = _node_get_option_str(
    
    260 272
                 prompt, 'really-workspace-reset-hard', ['ask', 'yes']) == 'ask'
    
    261 273
     
    
    ... ... @@ -312,6 +324,16 @@ class Context():
    312 324
         def get_workspaces(self):
    
    313 325
             return self._workspaces
    
    314 326
     
    
    327
    +    # get_workspace_project_cache():
    
    328
    +    #
    
    329
    +    # Return the WorkspaceProjectCache object used for this BuildStream invocation
    
    330
    +    #
    
    331
    +    # Returns:
    
    332
    +    #    (WorkspaceProjectCache): The WorkspaceProjectCache object
    
    333
    +    #
    
    334
    +    def get_workspace_project_cache(self):
    
    335
    +        return self._workspace_project_cache
    
    336
    +
    
    315 337
         # get_overrides():
    
    316 338
         #
    
    317 339
         # Fetch the override dictionary for the active project. This returns
    

  • buildstream/_frontend/app.py
    ... ... @@ -39,6 +39,7 @@ from .._stream import Stream
    39 39
     from .._versions import BST_FORMAT_VERSION
    
    40 40
     from .. import _yaml
    
    41 41
     from .._scheduler import ElementJob
    
    42
    +from .._workspaces import WorkspaceProjectCache
    
    42 43
     
    
    43 44
     # Import frontend assets
    
    44 45
     from . import Profile, LogLine, Status
    
    ... ... @@ -79,6 +80,7 @@ class App():
    79 80
             self._fail_messages = {}           # Failure messages by unique plugin id
    
    80 81
             self._interactive_failures = None  # Whether to handle failures interactively
    
    81 82
             self._started = False              # Whether a session has started
    
    83
    +        self._workspace_project_cache = WorkspaceProjectCache()  # A collection of workspace local data
    
    82 84
     
    
    83 85
             # UI Colors Profiles
    
    84 86
             self._content_profile = Profile(fg='yellow')
    
    ... ... @@ -164,7 +166,7 @@ class App():
    164 166
             # Load the Context
    
    165 167
             #
    
    166 168
             try:
    
    167
    -            self.context = Context()
    
    169
    +            self.context = Context(self._workspace_project_cache)
    
    168 170
                 self.context.load(config)
    
    169 171
             except BstError as e:
    
    170 172
                 self._error_exit(e, "Error loading user configuration")
    
    ... ... @@ -402,6 +404,20 @@ class App():
    402 404
             if self.stream:
    
    403 405
                 self.stream.cleanup()
    
    404 406
     
    
    407
    +    # guess_element()
    
    408
    +    #
    
    409
    +    # Attempts to interpret which element the user intended to run commands on
    
    410
    +    #
    
    411
    +    # Returns:
    
    412
    +    #    (str) The name of the element, or an empty string
    
    413
    +    def guess_element(self):
    
    414
    +        directory = self._main_options['directory']
    
    415
    +        workspace_project = self._workspace_project_cache.get(directory)
    
    416
    +        if workspace_project:
    
    417
    +            return workspace_project.get_default_element()
    
    418
    +        else:
    
    419
    +            return ""
    
    420
    +
    
    405 421
         ############################################################
    
    406 422
         #                   Abstract Class Methods                 #
    
    407 423
         ############################################################
    

  • buildstream/_frontend/cli.py
    ... ... @@ -59,18 +59,9 @@ def complete_target(args, incomplete):
    59 59
         :return: all the possible user-specified completions for the param
    
    60 60
         """
    
    61 61
     
    
    62
    +    from .. import utils
    
    62 63
         project_conf = 'project.conf'
    
    63 64
     
    
    64
    -    def ensure_project_dir(directory):
    
    65
    -        directory = os.path.abspath(directory)
    
    66
    -        while not os.path.isfile(os.path.join(directory, project_conf)):
    
    67
    -            parent_dir = os.path.dirname(directory)
    
    68
    -            if directory == parent_dir:
    
    69
    -                break
    
    70
    -            directory = parent_dir
    
    71
    -
    
    72
    -        return directory
    
    73
    -
    
    74 65
         # First resolve the directory, in case there is an
    
    75 66
         # active --directory/-C option
    
    76 67
         #
    
    ... ... @@ -89,7 +80,7 @@ def complete_target(args, incomplete):
    89 80
         else:
    
    90 81
             # Check if this directory or any of its parent directories
    
    91 82
             # contain a project config file
    
    92
    -        base_directory = ensure_project_dir(base_directory)
    
    83
    +        base_directory = utils._search_upward_for_file(base_directory, project_conf)
    
    93 84
     
    
    94 85
         # Now parse the project.conf just to find the element path,
    
    95 86
         # this is unfortunately a bit heavy.
    
    ... ... @@ -321,6 +312,12 @@ def build(app, elements, all_, track_, track_save, track_all, track_except, trac
    321 312
         if track_save:
    
    322 313
             click.echo("WARNING: --track-save is deprecated, saving is now unconditional", err=True)
    
    323 314
     
    
    315
    +    if not all_ and not elements:
    
    316
    +        # Attempt to divine the element from the workspace you're in
    
    317
    +        guessed_target = app.guess_element()
    
    318
    +        if guessed_target:
    
    319
    +            elements = (guessed_target,)
    
    320
    +
    
    324 321
         if track_all:
    
    325 322
             track_ = elements
    
    326 323
     
    
    ... ... @@ -375,6 +372,12 @@ def fetch(app, elements, deps, track_, except_, track_cross_junctions):
    375 372
                        "Since tracking modifies the build plan, all elements will be tracked.", err=True)
    
    376 373
             deps = PipelineSelection.ALL
    
    377 374
     
    
    375
    +    if not elements:
    
    376
    +        # Attempt to divine the element from the workspace you're in
    
    377
    +        guessed_target = app.guess_element()
    
    378
    +        if guessed_target:
    
    379
    +            elements = (guessed_target,)
    
    380
    +
    
    378 381
         with app.initialized(session_name="Fetch"):
    
    379 382
             app.stream.fetch(elements,
    
    380 383
                              selection=deps,
    
    ... ... @@ -411,6 +414,12 @@ def track(app, elements, deps, except_, cross_junctions):
    411 414
             none:  No dependencies, just the specified elements
    
    412 415
             all:   All dependencies of all specified elements
    
    413 416
         """
    
    417
    +    if not elements:
    
    418
    +        # Attempt to divine the element from the workspace you're in
    
    419
    +        guessed_target = app.guess_element()
    
    420
    +        if guessed_target:
    
    421
    +            elements = (guessed_target,)
    
    422
    +
    
    414 423
         with app.initialized(session_name="Track"):
    
    415 424
             # Substitute 'none' for 'redirect' so that element redirections
    
    416 425
             # will be done
    
    ... ... @@ -447,6 +456,12 @@ def pull(app, elements, deps, remote):
    447 456
             none:  No dependencies, just the element itself
    
    448 457
             all:   All dependencies
    
    449 458
         """
    
    459
    +    if not elements:
    
    460
    +        # Attempt to divine the element from the workspace you're in
    
    461
    +        guessed_target = app.guess_element()
    
    462
    +        if guessed_target:
    
    463
    +            elements = (guessed_target,)
    
    464
    +
    
    450 465
         with app.initialized(session_name="Pull"):
    
    451 466
             app.stream.pull(elements, selection=deps, remote=remote)
    
    452 467
     
    
    ... ... @@ -475,6 +490,11 @@ def push(app, elements, deps, remote):
    475 490
             none:  No dependencies, just the element itself
    
    476 491
             all:   All dependencies
    
    477 492
         """
    
    493
    +    if not elements:
    
    494
    +        # Attempt to divine the element from the workspace you're in
    
    495
    +        guessed_target = app.guess_element()
    
    496
    +        if guessed_target:
    
    497
    +            elements = (guessed_target,)
    
    478 498
         with app.initialized(session_name="Push"):
    
    479 499
             app.stream.push(elements, selection=deps, remote=remote)
    
    480 500
     
    
    ... ... @@ -545,6 +565,12 @@ def show(app, elements, deps, except_, order, format_):
    545 565
             bst show target.bst --format \\
    
    546 566
                 $'---------- %{name} ----------\\n%{vars}'
    
    547 567
         """
    
    568
    +    if not elements:
    
    569
    +        # Attempt to divine the element from the workspace you're in
    
    570
    +        guessed_target = app.guess_element()
    
    571
    +        if guessed_target:
    
    572
    +            elements = (guessed_target,)
    
    573
    +
    
    548 574
         with app.initialized():
    
    549 575
             dependencies = app.stream.load_selection(elements,
    
    550 576
                                                      selection=deps,
    
    ... ... @@ -574,7 +600,7 @@ def show(app, elements, deps, except_, order, format_):
    574 600
                   help="Mount a file or directory into the sandbox")
    
    575 601
     @click.option('--isolate', is_flag=True, default=False,
    
    576 602
                   help='Create an isolated build sandbox')
    
    577
    -@click.argument('element',
    
    603
    +@click.argument('element', required=False,
    
    578 604
                     type=click.Path(readable=False))
    
    579 605
     @click.argument('command', type=click.STRING, nargs=-1)
    
    580 606
     @click.pass_obj
    
    ... ... @@ -605,6 +631,14 @@ def shell(app, element, sysroot, mount, isolate, build_, command):
    605 631
             scope = Scope.RUN
    
    606 632
     
    
    607 633
         with app.initialized():
    
    634
    +        if not element:
    
    635
    +            # Attempt to divine the element from the workspace you're in
    
    636
    +            guessed_target = app.guess_element()
    
    637
    +            if guessed_target:
    
    638
    +                element = guessed_target
    
    639
    +            else:
    
    640
    +                raise AppError('Error: Missing argument "ELEMENT".')
    
    641
    +
    
    608 642
             dependencies = app.stream.load_selection((element,), selection=PipelineSelection.NONE)
    
    609 643
             element = dependencies[0]
    
    610 644
             prompt = app.shell_prompt(element)
    
    ... ... @@ -642,14 +676,27 @@ def shell(app, element, sysroot, mount, isolate, build_, command):
    642 676
                   help="Create a tarball from the artifact contents instead "
    
    643 677
                        "of a file tree. If LOCATION is '-', the tarball "
    
    644 678
                        "will be dumped to the standard output.")
    
    645
    -@click.argument('element',
    
    679
    +@click.argument('element', required=False,
    
    646 680
                     type=click.Path(readable=False))
    
    647
    -@click.argument('location', type=click.Path())
    
    681
    +@click.argument('location', type=click.Path(), required=False)
    
    648 682
     @click.pass_obj
    
    649 683
     def checkout(app, element, location, force, deps, integrate, hardlinks, tar):
    
    650 684
         """Checkout a built artifact to the specified location
    
    651 685
         """
    
    652 686
     
    
    687
    +    if not element and not location:
    
    688
    +        click.echo("ERROR: LOCATION is not specified", err=True)
    
    689
    +        sys.exit(-1)
    
    690
    +
    
    691
    +    if element and not location:
    
    692
    +        # Nasty hack to get around click's optional args
    
    693
    +        location = element
    
    694
    +        element = app.guess_element()
    
    695
    +
    
    696
    +    if not element:
    
    697
    +        click.echo("ERROR: ELEMENT is not specified", err=True)
    
    698
    +        sys.exit(-1)
    
    699
    +
    
    653 700
         if hardlinks and tar:
    
    654 701
             click.echo("ERROR: options --hardlinks and --tar conflict", err=True)
    
    655 702
             sys.exit(-1)
    
    ... ... @@ -741,8 +788,14 @@ def workspace_close(app, remove_dir, all_, elements):
    741 788
         """Close a workspace"""
    
    742 789
     
    
    743 790
         if not (all_ or elements):
    
    744
    -        click.echo('ERROR: no elements specified', err=True)
    
    745
    -        sys.exit(-1)
    
    791
    +        # NOTE: I may need to revisit this when implementing multiple projects
    
    792
    +        # opening one workspace.
    
    793
    +        element = app.guess_element()
    
    794
    +        if element:
    
    795
    +            elements = (element,)
    
    796
    +        else:
    
    797
    +            click.echo('ERROR: no elements specified', err=True)
    
    798
    +            sys.exit(-1)
    
    746 799
     
    
    747 800
         with app.initialized():
    
    748 801
     
    
    ... ... @@ -756,11 +809,18 @@ def workspace_close(app, remove_dir, all_, elements):
    756 809
     
    
    757 810
             elements = app.stream.redirect_element_names(elements)
    
    758 811
     
    
    759
    -        # Check that the workspaces in question exist
    
    812
    +        # Check that the workspaces in question exist, and that it's safe to
    
    813
    +        # remove them.
    
    760 814
             nonexisting = []
    
    761 815
             for element_name in elements:
    
    762 816
                 if not app.stream.workspace_exists(element_name):
    
    763 817
                     nonexisting.append(element_name)
    
    818
    +            if (app.stream.workspace_is_required(element_name) and app.interactive and
    
    819
    +                    app.context.prompt_workspace_close_project_inaccessible):
    
    820
    +                click.echo("Removing '{}' will prevent you from running buildstream commands".format(element_name))
    
    821
    +                if not click.confirm('Are you sure you want to close this workspace?'):
    
    822
    +                    click.echo('Aborting', err=True)
    
    823
    +                    sys.exit(-1)
    
    764 824
             if nonexisting:
    
    765 825
                 raise AppError("Workspace does not exist", detail="\n".join(nonexisting))
    
    766 826
     
    
    ... ... @@ -793,7 +853,11 @@ def workspace_reset(app, soft, track_, all_, elements):
    793 853
         with app.initialized():
    
    794 854
     
    
    795 855
             if not (all_ or elements):
    
    796
    -            raise AppError('No elements specified to reset')
    
    856
    +            element = app.guess_element()
    
    857
    +            if element:
    
    858
    +                elements = (element,)
    
    859
    +            else:
    
    860
    +                raise AppError('No elements specified to reset')
    
    797 861
     
    
    798 862
             if all_ and not app.stream.workspace_exists():
    
    799 863
                 raise AppError("No open workspaces to reset")
    

  • buildstream/_project.py
    ... ... @@ -95,8 +95,10 @@ class Project():
    95 95
             # The project name
    
    96 96
             self.name = None
    
    97 97
     
    
    98
    -        # The project directory
    
    99
    -        self.directory = self._ensure_project_dir(directory)
    
    98
    +        self._context = context  # The invocation Context, a private member
    
    99
    +
    
    100
    +        # The project directory, and whether the project was found from an external workspace
    
    101
    +        self.directory, self._required_workspace_element = self._find_project_dir(directory)
    
    100 102
     
    
    101 103
             # Absolute path to where elements are loaded from within the project
    
    102 104
             self.element_path = None
    
    ... ... @@ -117,7 +119,6 @@ class Project():
    117 119
             #
    
    118 120
             # Private Members
    
    119 121
             #
    
    120
    -        self._context = context  # The invocation Context
    
    121 122
     
    
    122 123
             self._default_mirror = default_mirror    # The name of the preferred mirror.
    
    123 124
     
    
    ... ... @@ -371,6 +372,14 @@ class Project():
    371 372
     
    
    372 373
             self._load_second_pass()
    
    373 374
     
    
    375
    +    # required_workspace_element()
    
    376
    +    #
    
    377
    +    # Returns the element whose workspace is required to load this project,
    
    378
    +    # if any.
    
    379
    +    #
    
    380
    +    def required_workspace_element(self):
    
    381
    +        return self._required_workspace_element
    
    382
    +
    
    374 383
         # cleanup()
    
    375 384
         #
    
    376 385
         # Cleans up resources used loading elements
    
    ... ... @@ -650,7 +659,7 @@ class Project():
    650 659
             # Source url aliases
    
    651 660
             output._aliases = _yaml.node_get(config, Mapping, 'aliases', default_value={})
    
    652 661
     
    
    653
    -    # _ensure_project_dir()
    
    662
    +    # _find_project_dir()
    
    654 663
         #
    
    655 664
         # Returns path of the project directory, if a configuration file is found
    
    656 665
         # in given directory or any of its parent directories.
    
    ... ... @@ -661,18 +670,26 @@ class Project():
    661 670
         # Raises:
    
    662 671
         #    LoadError if project.conf is not found
    
    663 672
         #
    
    664
    -    def _ensure_project_dir(self, directory):
    
    665
    -        directory = os.path.abspath(directory)
    
    666
    -        while not os.path.isfile(os.path.join(directory, _PROJECT_CONF_FILE)):
    
    667
    -            parent_dir = os.path.dirname(directory)
    
    668
    -            if directory == parent_dir:
    
    673
    +    # Returns:
    
    674
    +    #    (str) - the directory that contains the project, and
    
    675
    +    #    (str) - the name of the element required to find the project, or an empty string
    
    676
    +    #
    
    677
    +    def _find_project_dir(self, directory):
    
    678
    +        workspace_element = ""
    
    679
    +        project_directory = utils._search_upward_for_file(directory, _PROJECT_CONF_FILE)
    
    680
    +        if not project_directory:
    
    681
    +            workspace_project_cache = self._context.get_workspace_project_cache()
    
    682
    +            workspace_project = workspace_project_cache.get(directory)
    
    683
    +            if workspace_project:
    
    684
    +                project_directory = workspace_project.get_default_path()
    
    685
    +                workspace_element = workspace_project.get_default_element()
    
    686
    +            else:
    
    669 687
                     raise LoadError(
    
    670 688
                         LoadErrorReason.MISSING_PROJECT_CONF,
    
    671 689
                         '{} not found in current directory or any of its parent directories'
    
    672 690
                         .format(_PROJECT_CONF_FILE))
    
    673
    -            directory = parent_dir
    
    674 691
     
    
    675
    -        return directory
    
    692
    +        return project_directory, workspace_element
    
    676 693
     
    
    677 694
         def _load_plugin_factories(self, config, output):
    
    678 695
             plugin_source_origins = []   # Origins of custom sources
    

  • buildstream/_stream.py
    ... ... @@ -28,7 +28,7 @@ import tarfile
    28 28
     from contextlib import contextmanager
    
    29 29
     from tempfile import TemporaryDirectory
    
    30 30
     
    
    31
    -from ._exceptions import StreamError, ImplError, BstError, set_last_task_error
    
    31
    +from ._exceptions import StreamError, ImplError, BstError, set_last_task_error, LoadError, LoadErrorReason
    
    32 32
     from ._message import Message, MessageType
    
    33 33
     from ._scheduler import Scheduler, SchedStatus, TrackQueue, FetchQueue, BuildQueue, PullQueue, PushQueue
    
    34 34
     from ._pipeline import Pipeline, PipelineSelection
    
    ... ... @@ -550,6 +550,8 @@ class Stream():
    550 550
             # So far this function has tried to catch as many issues as possible with out making any changes
    
    551 551
             # Now it dose the bits that can not be made atomic.
    
    552 552
             targetGenerator = zip(elements, expanded_directories)
    
    553
    +        workspace_project_cache = self._context.get_workspace_project_cache()
    
    554
    +        project = self._context.get_toplevel_project()
    
    553 555
             for target, directory in targetGenerator:
    
    554 556
                 self._message(MessageType.INFO, "Creating workspace for element {}"
    
    555 557
                               .format(target.name))
    
    ... ... @@ -574,6 +576,10 @@ class Stream():
    574 576
                     with target.timed_activity("Staging sources to {}".format(directory)):
    
    575 577
                         target._open_workspace()
    
    576 578
     
    
    579
    +            workspace_project = workspace_project_cache.add(directory, project.directory,
    
    580
    +                                                            target._get_full_name())
    
    581
    +            workspace_project.write()
    
    582
    +
    
    577 583
                 # Saving the workspace once it is set up means that if the next workspace fails to be created before
    
    578 584
                 # the configuration gets saved. The successfully created workspace still gets saved.
    
    579 585
                 workspaces.save_config()
    
    ... ... @@ -601,6 +607,16 @@ class Stream():
    601 607
                     except OSError as e:
    
    602 608
                         raise StreamError("Could not remove  '{}': {}"
    
    603 609
                                           .format(workspace.get_absolute_path(), e)) from e
    
    610
    +        else:
    
    611
    +            workspace_project_cache = self._context.get_workspace_project_cache()
    
    612
    +            try:
    
    613
    +                workspace_project_cache.remove(workspace.get_absolute_path())
    
    614
    +            except LoadError as e:
    
    615
    +                # We might be closing a workspace with a deleted directory
    
    616
    +                if e.reason == LoadErrorReason.MISSING_FILE:
    
    617
    +                    pass
    
    618
    +                else:
    
    619
    +                    raise
    
    604 620
     
    
    605 621
             # Delete the workspace and save the configuration
    
    606 622
             workspaces.delete_workspace(element_name)
    
    ... ... @@ -644,6 +660,8 @@ class Stream():
    644 660
             for element in elements:
    
    645 661
                 workspace = workspaces.get_workspace(element._get_full_name())
    
    646 662
                 workspace_path = workspace.get_absolute_path()
    
    663
    +            workspace_project_cache = self._context.get_workspace_project_cache()
    
    664
    +            workspace_project = workspace_project_cache.get(workspace_path)
    
    647 665
                 if soft:
    
    648 666
                     workspace.prepared = False
    
    649 667
                     self._message(MessageType.INFO, "Reset workspace state for {} at: {}"
    
    ... ... @@ -664,6 +682,8 @@ class Stream():
    664 682
                 with element.timed_activity("Staging sources to {}".format(workspace_path)):
    
    665 683
                     element._open_workspace()
    
    666 684
     
    
    685
    +            workspace_project.write()
    
    686
    +
    
    667 687
                 self._message(MessageType.INFO,
    
    668 688
                               "Reset workspace for {} at: {}".format(element.name,
    
    669 689
                                                                      workspace_path))
    
    ... ... @@ -694,6 +714,20 @@ class Stream():
    694 714
     
    
    695 715
             return False
    
    696 716
     
    
    717
    +    # workspace_is_required()
    
    718
    +    #
    
    719
    +    # Checks whether the workspace belonging to element_name is required to
    
    720
    +    # load the project
    
    721
    +    #
    
    722
    +    # Args:
    
    723
    +    #    element_name (str): The element whose workspace may be required
    
    724
    +    #
    
    725
    +    # Returns:
    
    726
    +    #    (bool): True if the workspace is required
    
    727
    +    def workspace_is_required(self, element_name):
    
    728
    +        required_elm = self._project.required_workspace_element()
    
    729
    +        return required_elm == element_name
    
    730
    +
    
    697 731
         # workspace_list
    
    698 732
         #
    
    699 733
         # Serializes the workspaces and dumps them in YAML to stdout.
    

  • buildstream/_workspaces.py
    ... ... @@ -25,6 +25,211 @@ from ._exceptions import LoadError, LoadErrorReason
    25 25
     
    
    26 26
     
    
    27 27
     BST_WORKSPACE_FORMAT_VERSION = 3
    
    28
    +BST_WORKSPACE_PROJECT_FORMAT_VERSION = 1
    
    29
    +WORKSPACE_PROJECT_FILE = ".bstproject.yaml"
    
    30
    +
    
    31
    +
    
    32
    +# WorkspaceProject()
    
    33
    +#
    
    34
    +# An object to contain various helper functions and data required for
    
    35
    +# referring from a workspace back to buildstream.
    
    36
    +#
    
    37
    +# Args:
    
    38
    +#    directory (str): The directory that the workspace exists in
    
    39
    +#    project_path (str): The project path used to refer back
    
    40
    +#                        to buildstream projects.
    
    41
    +#    element_name (str): The name of the element used to create this workspace.
    
    42
    +class WorkspaceProject():
    
    43
    +    def __init__(self, directory, project_path="", element_name=""):
    
    44
    +        self._projects = []
    
    45
    +        self._directory = directory
    
    46
    +
    
    47
    +        assert (project_path and element_name) or (not project_path and not element_name)
    
    48
    +        if project_path:
    
    49
    +            self._add_project(project_path, element_name)
    
    50
    +
    
    51
    +    # get_default_path()
    
    52
    +    #
    
    53
    +    # Retrieves the default path to a project.
    
    54
    +    #
    
    55
    +    # Returns:
    
    56
    +    #    (str): The path to a project
    
    57
    +    def get_default_path(self):
    
    58
    +        return self._projects[0]['project-path']
    
    59
    +
    
    60
    +    # get_default_element()
    
    61
    +    #
    
    62
    +    # Retrieves the name of the element that owns this workspace.
    
    63
    +    #
    
    64
    +    # Returns:
    
    65
    +    #    (str): The name of an element
    
    66
    +    def get_default_element(self):
    
    67
    +        return self._projects[0]['element-name']
    
    68
    +
    
    69
    +    # to_dict()
    
    70
    +    #
    
    71
    +    # Turn the members data into a dict for serialization purposes
    
    72
    +    #
    
    73
    +    # Returns:
    
    74
    +    #    (dict): A dict representation of the WorkspaceProject
    
    75
    +    #
    
    76
    +    def to_dict(self):
    
    77
    +        ret = {
    
    78
    +            'projects': self._projects,
    
    79
    +            'format-version': BST_WORKSPACE_PROJECT_FORMAT_VERSION,
    
    80
    +        }
    
    81
    +        return ret
    
    82
    +
    
    83
    +    # from_dict()
    
    84
    +    #
    
    85
    +    # Loads a new WorkspaceProject from a simple dictionary
    
    86
    +    #
    
    87
    +    # Args:
    
    88
    +    #    directory (str): The directory that the workspace exists in
    
    89
    +    #    dictionary (dict): The dict to generate a WorkspaceProject from
    
    90
    +    #
    
    91
    +    # Returns:
    
    92
    +    #   (WorkspaceProject): A newly instantiated WorkspaceProject
    
    93
    +    @classmethod
    
    94
    +    def from_dict(cls, directory, dictionary):
    
    95
    +        # Only know how to handle one format-version at the moment.
    
    96
    +        format_version = int(dictionary['format-version'])
    
    97
    +        assert format_version == BST_WORKSPACE_PROJECT_FORMAT_VERSION, \
    
    98
    +            "Format version {} not found in {}".format(BST_WORKSPACE_PROJECT_FORMAT_VERSION, dictionary)
    
    99
    +
    
    100
    +        workspace_project = cls(directory)
    
    101
    +        for item in dictionary['projects']:
    
    102
    +            workspace_project._add_project(item['project-path'], item['element-name'])
    
    103
    +
    
    104
    +        return workspace_project
    
    105
    +
    
    106
    +    # load()
    
    107
    +    #
    
    108
    +    # Loads the WorkspaceProject for a given directory. This directory may be a
    
    109
    +    # subdirectory of the workspace's directory.
    
    110
    +    #
    
    111
    +    # Args:
    
    112
    +    #    directory (str): The directory
    
    113
    +    # Returns:
    
    114
    +    #    (WorkspaceProject): The created WorkspaceProject, if in a workspace, or
    
    115
    +    #    (NoneType): None, if the directory is not inside a workspace.
    
    116
    +    @classmethod
    
    117
    +    def load(cls, directory):
    
    118
    +        project_dir = cls.search_for_dir(directory)
    
    119
    +        if project_dir:
    
    120
    +            workspace_file = os.path.join(project_dir, WORKSPACE_PROJECT_FILE)
    
    121
    +            data_dict = _yaml.load(workspace_file)
    
    122
    +            return cls.from_dict(project_dir, data_dict)
    
    123
    +        else:
    
    124
    +            return None
    
    125
    +
    
    126
    +    # write()
    
    127
    +    #
    
    128
    +    # Writes the WorkspaceProject to disk
    
    129
    +    def write(self):
    
    130
    +        os.makedirs(self._directory, exist_ok=True)
    
    131
    +        _yaml.dump(self.to_dict(), self._get_filename())
    
    132
    +
    
    133
    +    # search_for_dir()
    
    134
    +    #
    
    135
    +    # Returns the directory that contains the workspace local project file,
    
    136
    +    # searching upwards from search_dir.
    
    137
    +    @staticmethod
    
    138
    +    def search_for_dir(search_dir):
    
    139
    +        return utils._search_upward_for_file(search_dir, WORKSPACE_PROJECT_FILE)
    
    140
    +
    
    141
    +    def _get_filename(self):
    
    142
    +        return os.path.join(self._directory, WORKSPACE_PROJECT_FILE)
    
    143
    +
    
    144
    +    def _add_project(self, project_path, element_name):
    
    145
    +        assert (project_path and element_name)
    
    146
    +        self._projects.append({'project-path': project_path, 'element-name': element_name})
    
    147
    +
    
    148
    +
    
    149
    +# WorkspaceProjectCache()
    
    150
    +#
    
    151
    +# A class to manage workspace project data for multiple workspaces.
    
    152
    +#
    
    153
    +class WorkspaceProjectCache():
    
    154
    +    def __init__(self):
    
    155
    +        self._projects = {}  # Mapping of a workspace directory to its WorkspaceProject
    
    156
    +
    
    157
    +    # get()
    
    158
    +    #
    
    159
    +    # Returns a WorkspaceProject for a given directory, retrieving from the cache if
    
    160
    +    # present, and searching the filesystem for the file and loading it if not.
    
    161
    +    #
    
    162
    +    # Args:
    
    163
    +    #    directory (str): The directory to search for a WorkspaceProject.
    
    164
    +    #
    
    165
    +    # Returns:
    
    166
    +    #    (WorkspaceProject): The WorkspaceProject that was found for that directory.
    
    167
    +    #    or      (NoneType): None, if no WorkspaceProject can be found.
    
    168
    +    #
    
    169
    +    def get(self, directory):
    
    170
    +        try:
    
    171
    +            workspace_project = self._projects[directory]
    
    172
    +        except KeyError:
    
    173
    +            found_dir = WorkspaceProject.search_for_dir(directory)
    
    174
    +            if found_dir:
    
    175
    +                try:
    
    176
    +                    workspace_project = self._projects[found_dir]
    
    177
    +                except KeyError:
    
    178
    +                    workspace_project = WorkspaceProject.load(found_dir)
    
    179
    +                    self._projects[found_dir] = workspace_project
    
    180
    +            else:
    
    181
    +                workspace_project = None
    
    182
    +
    
    183
    +        return workspace_project
    
    184
    +
    
    185
    +    # add()
    
    186
    +    #
    
    187
    +    # Adds the project path and element name to the WorkspaceProject that exists
    
    188
    +    # for that directory
    
    189
    +    #
    
    190
    +    # Args:
    
    191
    +    #    directory (str): The directory to search for a WorkspaceProject.
    
    192
    +    #    project_path (str): The path to the project that refers to this workspace
    
    193
    +    #    element_name (str): The element in the project that was refers to this workspace
    
    194
    +    #
    
    195
    +    # Returns:
    
    196
    +    #    (WorkspaceProject): The WorkspaceProject that was found for that directory.
    
    197
    +    #
    
    198
    +    def add(self, directory, project_path='', element_name=''):
    
    199
    +        workspace_project = self.get(directory)
    
    200
    +        if not workspace_project:
    
    201
    +            workspace_project = WorkspaceProject(directory)
    
    202
    +            self._projects[directory] = workspace_project
    
    203
    +        if project_path:
    
    204
    +            workspace_project._add_project(project_path, element_name)
    
    205
    +        return workspace_project
    
    206
    +
    
    207
    +    # remove()
    
    208
    +    #
    
    209
    +    # Removes the project path and element name from the WorkspaceProject that exists
    
    210
    +    # for that directory.
    
    211
    +    #
    
    212
    +    # NOTE: This currently just deletes the file, but with support for multiple
    
    213
    +    # projects opening the same workspace, this will involve decreasing the count
    
    214
    +    # and deleting the file if there are no more projects.
    
    215
    +    #
    
    216
    +    # Args:
    
    217
    +    #    directory (str): The directory to search for a WorkspaceProject.
    
    218
    +    #    project_path (str): **UNUSED** The path to the project that refers to this workspace
    
    219
    +    #    element_name (str): **UNUSED** The element in the project that was refers to this workspace
    
    220
    +    #
    
    221
    +    def remove(self, directory, project_path='', element_name=''):
    
    222
    +        # NOTE: project_path and element_name will only be used when I implement
    
    223
    +        #       multiple owners of a workspace
    
    224
    +        workspace_project = self.get(directory)
    
    225
    +        if not workspace_project:
    
    226
    +            raise LoadError(LoadErrorReason.MISSING_FILE,
    
    227
    +                            "Failed to find a {} file to remove".format(WORKSPACE_PROJECT_FILE))
    
    228
    +        path = workspace_project._get_filename()
    
    229
    +        try:
    
    230
    +            os.unlink(path)
    
    231
    +        except FileNotFoundError:
    
    232
    +            pass
    
    28 233
     
    
    29 234
     
    
    30 235
     # Workspace()
    
    ... ... @@ -174,10 +379,15 @@ class Workspace():
    174 379
             if recalculate or self._key is None:
    
    175 380
                 fullpath = self.get_absolute_path()
    
    176 381
     
    
    382
    +            excluded_files = (WORKSPACE_PROJECT_FILE,)
    
    383
    +
    
    177 384
                 # Get a list of tuples of the the project relative paths and fullpaths
    
    178 385
                 if os.path.isdir(fullpath):
    
    179 386
                     filelist = utils.list_relative_paths(fullpath)
    
    180
    -                filelist = [(relpath, os.path.join(fullpath, relpath)) for relpath in filelist]
    
    387
    +                filelist = [
    
    388
    +                    (relpath, os.path.join(fullpath, relpath)) for relpath in filelist
    
    389
    +                    if relpath not in excluded_files
    
    390
    +                ]
    
    181 391
                 else:
    
    182 392
                     filelist = [(self.get_absolute_path(), fullpath)]
    
    183 393
     
    

  • buildstream/data/userconfig.yaml
    ... ... @@ -128,6 +128,14 @@ prompt:
    128 128
       #
    
    129 129
       really-workspace-close-remove-dir: ask
    
    130 130
     
    
    131
    +  # Whether to really proceed with 'bst workspace close' when doing so would
    
    132
    +  # stop them from running bst commands in this workspace.
    
    133
    +  #
    
    134
    +  #  ask - Ask the user if they are sure.
    
    135
    +  #  yes - Always close, without asking.
    
    136
    +  #
    
    137
    +  really-workspace-close-project-inaccessible: ask
    
    138
    +
    
    131 139
       # Whether to really proceed with 'bst workspace reset' doing a hard reset of
    
    132 140
       # a workspace, potentially losing changes.
    
    133 141
       #
    

  • buildstream/utils.py
    ... ... @@ -1242,3 +1242,17 @@ def _deduplicate(iterable, key=None):
    1242 1242
     def _get_link_mtime(path):
    
    1243 1243
         path_stat = os.lstat(path)
    
    1244 1244
         return path_stat.st_mtime
    
    1245
    +
    
    1246
    +
    
    1247
    +# Returns the first directory to contain filename, or an empty string if
    
    1248
    +# none found
    
    1249
    +#
    
    1250
    +def _search_upward_for_file(directory, filename):
    
    1251
    +    directory = os.path.abspath(directory)
    
    1252
    +    while not os.path.isfile(os.path.join(directory, filename)):
    
    1253
    +        parent_dir = os.path.dirname(directory)
    
    1254
    +        if directory == parent_dir:
    
    1255
    +            return ""
    
    1256
    +        directory = parent_dir
    
    1257
    +
    
    1258
    +    return directory

  • doc/sessions/developing.run
    ... ... @@ -24,6 +24,11 @@ commands:
    24 24
       output: ../source/sessions/developing-build-after-changes.html
    
    25 25
       command: build hello.bst
    
    26 26
     
    
    27
    +# Rebuild, from the workspace
    
    28
    +- directory: ../examples/developing/workspace_hello
    
    29
    +  output: ../source/sessions/developing-build-after-changes-workspace.html
    
    30
    +  command: build
    
    31
    +
    
    27 32
     # Capture shell output with changes
    
    28 33
     - directory: ../examples/developing/
    
    29 34
       output: ../source/sessions/developing-shell-after-changes.html
    

  • doc/source/developing/workspaces.rst
    ... ... @@ -50,11 +50,16 @@ We can open workspace_hello/hello.c and make the following change:
    50 50
     .. literalinclude:: ../../examples/developing/update.patch
    
    51 51
         :language: diff
    
    52 52
     
    
    53
    -Now, rebuild the hello.bst element
    
    53
    +Now, rebuild the hello.bst element.
    
    54 54
     
    
    55 55
     .. raw:: html
    
    56 56
        :file: ../sessions/developing-build-after-changes.html
    
    57 57
     
    
    58
    +Note that if you run the command from inside the workspace, the element name is optional.
    
    59
    +
    
    60
    +.. raw:: html
    
    61
    +   :file: ../sessions/developing-build-after-changes-workspace.html
    
    62
    +
    
    58 63
     Now running the hello command using bst shell:
    
    59 64
     
    
    60 65
     .. raw:: html
    

  • tests/frontend/workspace.py
    ... ... @@ -31,6 +31,7 @@ import shutil
    31 31
     import subprocess
    
    32 32
     from ruamel.yaml.comments import CommentedSet
    
    33 33
     from tests.testutils import cli, create_repo, ALL_REPO_KINDS, wait_for_cache_granularity
    
    34
    +from tests.testutils import create_artifact_share
    
    34 35
     
    
    35 36
     from buildstream import _yaml
    
    36 37
     from buildstream._exceptions import ErrorDomain, LoadError, LoadErrorReason
    
    ... ... @@ -615,9 +616,15 @@ def test_list(cli, tmpdir, datafiles):
    615 616
     @pytest.mark.datafiles(DATA_DIR)
    
    616 617
     @pytest.mark.parametrize("kind", repo_kinds)
    
    617 618
     @pytest.mark.parametrize("strict", [("strict"), ("non-strict")])
    
    618
    -def test_build(cli, tmpdir, datafiles, kind, strict):
    
    619
    +@pytest.mark.parametrize("call_from,guess_element", [
    
    620
    +    ("project", "no-guess"), ("workspace", "guess"), ("workspace", "no-guess")
    
    621
    +])
    
    622
    +def test_build(cli, tmpdir_factory, datafiles, kind, strict, call_from, guess_element):
    
    623
    +    tmpdir = tmpdir_factory.mktemp('')
    
    619 624
         element_name, project, workspace = open_workspace(cli, tmpdir, datafiles, kind, False)
    
    620 625
         checkout = os.path.join(str(tmpdir), 'checkout')
    
    626
    +    args_dir = ['-C', workspace] if call_from == "workspace" else []
    
    627
    +    args_elm = [element_name] if guess_element == "no-guess" else []
    
    621 628
     
    
    622 629
         # Modify workspace
    
    623 630
         shutil.rmtree(os.path.join(workspace, 'usr', 'bin'))
    
    ... ... @@ -640,15 +647,14 @@ def test_build(cli, tmpdir, datafiles, kind, strict):
    640 647
         # Build modified workspace
    
    641 648
         assert cli.get_element_state(project, element_name) == 'buildable'
    
    642 649
         assert cli.get_element_key(project, element_name) == "{:?<64}".format('')
    
    643
    -    result = cli.run(project=project, args=['build', element_name])
    
    650
    +    result = cli.run(project=project, args=args_dir + ['build'] + args_elm)
    
    644 651
         result.assert_success()
    
    645 652
         assert cli.get_element_state(project, element_name) == 'cached'
    
    646 653
         assert cli.get_element_key(project, element_name) != "{:?<64}".format('')
    
    647 654
     
    
    648 655
         # Checkout the result
    
    649
    -    result = cli.run(project=project, args=[
    
    650
    -        'checkout', element_name, checkout
    
    651
    -    ])
    
    656
    +    result = cli.run(project=project,
    
    657
    +                     args=args_dir + ['checkout'] + args_elm + [checkout])
    
    652 658
         result.assert_success()
    
    653 659
     
    
    654 660
         # Check that the pony.conf from the modified workspace exists
    
    ... ... @@ -1055,3 +1061,144 @@ def test_multiple_failed_builds(cli, tmpdir, datafiles):
    1055 1061
             result = cli.run(project=project, args=["build", element_name])
    
    1056 1062
             assert "BUG" not in result.stderr
    
    1057 1063
             assert cli.get_element_state(project, element_name) != "cached"
    
    1064
    +
    
    1065
    +
    
    1066
    +@pytest.mark.datafiles(DATA_DIR)
    
    1067
    +@pytest.mark.parametrize("guess_element", [("guess"), ("no-guess")])
    
    1068
    +def test_external_fetch(cli, datafiles, tmpdir_factory, guess_element):
    
    1069
    +    # Fetching from a workspace outside a project doesn't fail horribly
    
    1070
    +    tmpdir = tmpdir_factory.mktemp('')
    
    1071
    +    element_name, project, workspace = open_workspace(cli, tmpdir, datafiles, "git", False)
    
    1072
    +    arg_elm = [element_name] if guess_element == "no-guess" else []
    
    1073
    +
    
    1074
    +    result = cli.run(project=project, args=['-C', workspace, 'fetch'] + arg_elm)
    
    1075
    +    result.assert_success()
    
    1076
    +
    
    1077
    +    # We already fetched it by opening the workspace, but we're also checking
    
    1078
    +    # `bst show` works here
    
    1079
    +    result = cli.run(project=project,
    
    1080
    +                     args=['-C', workspace, 'show', '--deps', 'none', '--format', '%{state}'] + arg_elm)
    
    1081
    +    result.assert_success()
    
    1082
    +    assert result.output.strip() == 'buildable'
    
    1083
    +
    
    1084
    +
    
    1085
    +@pytest.mark.datafiles(DATA_DIR)
    
    1086
    +@pytest.mark.parametrize("guess_element", [("guess"), ("no-guess")])
    
    1087
    +def test_external_push_pull(cli, datafiles, tmpdir_factory, guess_element):
    
    1088
    +    # Pushing and pulling to/from an artifact cache works from an external workspace
    
    1089
    +    tmpdir = tmpdir_factory.mktemp('')
    
    1090
    +    element_name, project, workspace = open_workspace(cli, tmpdir, datafiles, "git", False)
    
    1091
    +    args_elm = [element_name] if guess_element == "no-guess" else []
    
    1092
    +
    
    1093
    +    with create_artifact_share(os.path.join(str(tmpdir), 'artifactshare')) as share:
    
    1094
    +        result = cli.run(project=project, args=['-C', workspace, 'build', element_name])
    
    1095
    +        result.assert_success()
    
    1096
    +
    
    1097
    +        cli.configure({
    
    1098
    +            'artifacts': {'url': share.repo, 'push': True}
    
    1099
    +        })
    
    1100
    +
    
    1101
    +        result = cli.run(project=project, args=['-C', workspace, 'push'] + args_elm)
    
    1102
    +        result.assert_success()
    
    1103
    +
    
    1104
    +        result = cli.run(project=project, args=['-C', workspace, 'pull', '--deps', 'all'] + args_elm)
    
    1105
    +        result.assert_success()
    
    1106
    +
    
    1107
    +
    
    1108
    +@pytest.mark.datafiles(DATA_DIR)
    
    1109
    +@pytest.mark.parametrize("guess_element", [("guess"), ("no-guess")])
    
    1110
    +def test_external_track(cli, datafiles, tmpdir_factory, guess_element):
    
    1111
    +    # Tracking does not get horribly confused
    
    1112
    +    tmpdir = tmpdir_factory.mktemp('')
    
    1113
    +    element_name, project, workspace = open_workspace(cli, tmpdir, datafiles, "git", True)
    
    1114
    +    args_elm = [element_name] if guess_element == "no-guess" else []
    
    1115
    +
    
    1116
    +    # The workspace is necessarily already tracked, so we only care that
    
    1117
    +    # there's no weird errors.
    
    1118
    +    result = cli.run(project=project, args=['-C', workspace, 'track'] + args_elm)
    
    1119
    +    result.assert_success()
    
    1120
    +
    
    1121
    +
    
    1122
    +@pytest.mark.datafiles(DATA_DIR)
    
    1123
    +def test_external_open_other(cli, datafiles, tmpdir_factory):
    
    1124
    +    # >From inside an external workspace, open another workspace
    
    1125
    +    tmpdir1 = tmpdir_factory.mktemp('')
    
    1126
    +    tmpdir2 = tmpdir_factory.mktemp('')
    
    1127
    +    # Making use of the assumption that it's the same project in both invocations of open_workspace
    
    1128
    +    alpha_element, project, alpha_workspace = open_workspace(cli, tmpdir1, datafiles, "git", False, suffix="-alpha")
    
    1129
    +    beta_element, _, beta_workspace = open_workspace(cli, tmpdir2, datafiles, "git", False, suffix="-beta")
    
    1130
    +
    
    1131
    +    # Closing the other element first, because I'm too lazy to create an
    
    1132
    +    # element without opening it
    
    1133
    +    result = cli.run(project=project, args=['workspace', 'close', beta_element])
    
    1134
    +    result.assert_success()
    
    1135
    +
    
    1136
    +    result = cli.run(project=project, args=[
    
    1137
    +        '-C', alpha_workspace, 'workspace', 'open', '--force', '--directory', beta_workspace, beta_element
    
    1138
    +    ])
    
    1139
    +    result.assert_success()
    
    1140
    +
    
    1141
    +
    
    1142
    +@pytest.mark.datafiles(DATA_DIR)
    
    1143
    +def test_external_close_other(cli, datafiles, tmpdir_factory):
    
    1144
    +    # >From inside an external workspace, close the other workspace
    
    1145
    +    tmpdir1 = tmpdir_factory.mktemp('')
    
    1146
    +    tmpdir2 = tmpdir_factory.mktemp('')
    
    1147
    +    # Making use of the assumption that it's the same project in both invocations of open_workspace
    
    1148
    +    alpha_element, project, alpha_workspace = open_workspace(cli, tmpdir1, datafiles, "git", False, suffix="-alpha")
    
    1149
    +    beta_element, _, beta_workspace = open_workspace(cli, tmpdir2, datafiles, "git", False, suffix="-beta")
    
    1150
    +
    
    1151
    +    result = cli.run(project=project, args=['-C', alpha_workspace, 'workspace', 'close', beta_element])
    
    1152
    +    result.assert_success()
    
    1153
    +
    
    1154
    +
    
    1155
    +@pytest.mark.datafiles(DATA_DIR)
    
    1156
    +@pytest.mark.parametrize("guess_element", [("guess"), ("no-guess")])
    
    1157
    +def test_external_close_self(cli, datafiles, tmpdir_factory, guess_element):
    
    1158
    +    # >From inside an external workspace, close it
    
    1159
    +    tmpdir1 = tmpdir_factory.mktemp('')
    
    1160
    +    tmpdir2 = tmpdir_factory.mktemp('')
    
    1161
    +    # Making use of the assumption that it's the same project in both invocations of open_workspace
    
    1162
    +    alpha_element, project, alpha_workspace = open_workspace(cli, tmpdir1, datafiles, "git", False, suffix="-alpha")
    
    1163
    +    beta_element, _, beta_workspace = open_workspace(cli, tmpdir2, datafiles, "git", False, suffix="-beta")
    
    1164
    +    arg_elm = [alpha_element] if guess_element == "no-guess" else []
    
    1165
    +
    
    1166
    +    result = cli.run(project=project, args=['-C', alpha_workspace, 'workspace', 'close'] + arg_elm)
    
    1167
    +    result.assert_success()
    
    1168
    +
    
    1169
    +
    
    1170
    +@pytest.mark.datafiles(DATA_DIR)
    
    1171
    +def test_external_reset_other(cli, datafiles, tmpdir_factory):
    
    1172
    +    tmpdir1 = tmpdir_factory.mktemp('')
    
    1173
    +    tmpdir2 = tmpdir_factory.mktemp('')
    
    1174
    +    # Making use of the assumption that it's the same project in both invocations of open_workspace
    
    1175
    +    alpha_element, project, alpha_workspace = open_workspace(cli, tmpdir1, datafiles, "git", False, suffix="-alpha")
    
    1176
    +    beta_element, _, beta_workspace = open_workspace(cli, tmpdir2, datafiles, "git", False, suffix="-beta")
    
    1177
    +
    
    1178
    +    result = cli.run(project=project, args=['-C', alpha_workspace, 'workspace', 'reset', beta_element])
    
    1179
    +    result.assert_success()
    
    1180
    +
    
    1181
    +
    
    1182
    +@pytest.mark.datafiles(DATA_DIR)
    
    1183
    +@pytest.mark.parametrize("guess_element", [("guess"), ("no-guess")])
    
    1184
    +def test_external_reset_self(cli, datafiles, tmpdir, guess_element):
    
    1185
    +    element, project, workspace = open_workspace(cli, tmpdir, datafiles, "git", False)
    
    1186
    +    arg_elm = [element] if guess_element == "no-guess" else []
    
    1187
    +
    
    1188
    +    # Command succeeds
    
    1189
    +    result = cli.run(project=project, args=['-C', workspace, 'workspace', 'reset'] + arg_elm)
    
    1190
    +    result.assert_success()
    
    1191
    +
    
    1192
    +    # Successive commands still work (i.e. .bstproject.yaml hasn't been deleted)
    
    1193
    +    result = cli.run(project=project, args=['-C', workspace, 'workspace', 'list'])
    
    1194
    +    result.assert_success()
    
    1195
    +
    
    1196
    +
    
    1197
    +@pytest.mark.datafiles(DATA_DIR)
    
    1198
    +def test_external_list(cli, datafiles, tmpdir_factory):
    
    1199
    +    tmpdir = tmpdir_factory.mktemp('')
    
    1200
    +    # Making use of the assumption that it's the same project in both invocations of open_workspace
    
    1201
    +    element, project, workspace = open_workspace(cli, tmpdir, datafiles, "git", False)
    
    1202
    +
    
    1203
    +    result = cli.run(project=project, args=['-C', workspace, 'workspace', 'list'])
    
    1204
    +    result.assert_success()

  • tests/integration/shell.py
    ... ... @@ -353,3 +353,39 @@ def test_integration_devices(cli, tmpdir, datafiles):
    353 353
     
    
    354 354
         result = execute_shell(cli, project, ["true"], element=element_name)
    
    355 355
         assert result.exit_code == 0
    
    356
    +
    
    357
    +
    
    358
    +# Test that a shell can be opened from an external workspace
    
    359
    +@pytest.mark.datafiles(DATA_DIR)
    
    360
    +@pytest.mark.parametrize("build_shell", [("build"), ("nobuild")])
    
    361
    +@pytest.mark.parametrize("guess_element", [("guess"), ("no-guess")])
    
    362
    +@pytest.mark.skipif(IS_LINUX and not HAVE_BWRAP, reason='Only available with bubblewrap on Linux')
    
    363
    +def test_integration_external_workspace(cli, tmpdir_factory, datafiles, build_shell, guess_element):
    
    364
    +    tmpdir = tmpdir_factory.mktemp("")
    
    365
    +    project = os.path.join(datafiles.dirname, datafiles.basename)
    
    366
    +    element_name = 'autotools/amhello.bst'
    
    367
    +    workspace_dir = os.path.join(str(tmpdir), 'workspace')
    
    368
    +
    
    369
    +    if guess_element == "guess":
    
    370
    +        # Mutate the project.conf to use a default shell command
    
    371
    +        project_file = os.path.join(project, 'project.conf')
    
    372
    +        config_text = "shell:\n"\
    
    373
    +                      "  command: ['true']\n"
    
    374
    +        with open(project_file, 'a') as f:
    
    375
    +            f.write(config_text)
    
    376
    +
    
    377
    +    result = cli.run(project=project, args=[
    
    378
    +        'workspace', 'open', '--directory', workspace_dir, element_name
    
    379
    +    ])
    
    380
    +    result.assert_success()
    
    381
    +
    
    382
    +    result = cli.run(project=project, args=['-C', workspace_dir, 'build', element_name])
    
    383
    +    result.assert_success()
    
    384
    +
    
    385
    +    command = ['-C', workspace_dir, 'shell']
    
    386
    +    if build_shell == 'build':
    
    387
    +        command.append('--build')
    
    388
    +    if guess_element == "no-guess":
    
    389
    +        command.extend([element_name, '--', 'true'])
    
    390
    +    result = cli.run(project=project, cwd=workspace_dir, args=command)
    
    391
    +    result.assert_success()



  • [Date Prev][Date Next]   [Thread Prev][Thread Next]   [Thread Index] [Date Index] [Author Index]