Commit 136deb2e authored by Jim MacArthur's avatar Jim MacArthur
Browse files

Add basic storage test 'storage-test.py'

parent 0213f68d

tests/sandboxes/storage-test/original/bin/bash

0 → 100644
+1 −0
Original line number Diff line number Diff line 
This is the original /bin/bash.

tests/sandboxes/storage-test/original/bin/hello

0 → 100644
+1 −0
Original line number Diff line number Diff line 
This is the original /bin/hello.

tests/sandboxes/storage-test/overlay/bin/bash

0 → 100644
+1 −0
Original line number Diff line number Diff line 
This is the replacement /bin/bash.

tests/sandboxes/storage-tests.py

0 → 100644
+57 −0
Original line number Diff line number Diff line 
import os

import pytest



from buildstream._exceptions import ErrorDomain



from buildstream._context import Context

from buildstream.storage._casbaseddirectory import CasBasedDirectory

from buildstream.storage._filebaseddirectory import FileBasedDirectory



DATA_DIR = os.path.join(

    os.path.dirname(os.path.realpath(__file__)),

    "storage-test"

)





def setup_backend(backend_class, tmpdir):

    if backend_class == FileBasedDirectory:

        return backend_class(os.path.join(tmpdir, "vdir"))

    else:

        context = Context()

        context.artifactdir = os.path.join(tmpdir, "cas")

        return backend_class(context)





@pytest.mark.parametrize("backend", [

    FileBasedDirectory, CasBasedDirectory])

@pytest.mark.datafiles(DATA_DIR)

def test_import(tmpdir, datafiles, backend):

    original = os.path.join(str(datafiles), "original")



    c = setup_backend(backend, str(tmpdir))



    c.import_files(original)



    assert("bin/bash" in c.list_relative_paths())

    assert("bin/hello" in c.list_relative_paths())





@pytest.mark.parametrize("backend", [

    FileBasedDirectory, CasBasedDirectory])

@pytest.mark.datafiles(DATA_DIR)

def test_modified_file_list(tmpdir, datafiles, backend):

    original = os.path.join(str(datafiles), "original")

    overlay = os.path.join(str(datafiles), "overlay")



    c = setup_backend(backend, str(tmpdir))



    c.import_files(original)



    c.mark_unmodified()



    c.import_files(overlay)



    print("List of all paths in imported results: {}".format(c.list_relative_paths()))

    assert("bin/bash" in c.list_relative_paths())

    assert("bin/bash" in c.list_modified_paths())

    assert("bin/hello" not in c.list_modified_paths())