summaryrefslogtreecommitdiff
path: root/tests/testutils/repo/repo.py
blob: 03d616a9083d45f8620e8353a4e7690be5e40a9c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import os
import shutil


# Repo()
#
# Abstract class providing scaffolding for
# generating data to be used with various sources
#
# Args:
#    directory (str): The base temp directory for the test
#
class Repo():

    def __init__(self, directory):

        # The working directory for the repo object
        #
        self.directory = os.path.abspath(directory)

        # The directory the actual repo will be stored in
        self.repo = os.path.join(self.directory, 'repo')

        os.makedirs(self.repo)

    # create():
    #
    # Create a repository in self.directory and add the initial content
    #
    # Args:
    #    directory: A directory with content to commit
    #
    # Returns:
    #    (smth): A new ref corresponding to this commit, which can
    #            be passed as the ref in the Repo.source_config() API.
    #
    def create(self, directory):
        pass

    # source_config()
    #
    # Args:
    #    ref (smth): An optional abstract ref object, usually a string.
    #
    # Returns:
    #    (dict): A configuration which can be serialized as a
    #            source when generating an element file on the fly
    #
    def source_config(self, ref=None):
        pass

    # copy_directory():
    #
    # Copies the content of src to the directory dest
    #
    # Like shutil.copytree(), except dest is expected
    # to exist.
    #
    # Args:
    #    src (str): The source directory
    #    dest (str): The destination directory
    #
    def copy_directory(self, src, dest):
        for filename in os.listdir(src):
            src_path = os.path.join(src, filename)
            dest_path = os.path.join(dest, filename)
            if os.path.isdir(src_path):
                shutil.copytree(src_path, dest_path)
            else:
                shutil.copy2(src_path, dest_path)