summaryrefslogtreecommitdiffstats
path: root/tests/distaf/distaf_libs/gluster_libs/snap_ops.py
blob: bc4067d862189cf581975c97c1f4147b2d1df507 (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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#!/usr/bin/env python
#  This file is part of DiSTAF
#  Copyright (C) 2015-2016  Red Hat, Inc. <http://www.redhat.com>
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License along
#  with this program; if not, write to the Free Software Foundation, Inc.,
#  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.


from distaf.util import tc
from distaf.gluster_libs.volume_ops import start_volume, stop_volume


def snap_create(volname, snapname, server='', desc=''):
    """
        Runs snap create command and returns the output
    """
    if server == '':
        server = tc.nodes[0]
    if desc != '':
        desc = "description %s" % desc
    ret = tc.run(server, "gluster snapshot create %s %s %s" \
            % (volname, snapname, desc))
    return ret


def snap_activate(snapname, server=''):
    """
        Activate the snap and returns the output
    """
    if server == '':
        server = tc.nodes[0]
    return tc.run(server, "gluster snapshot activate %s" % snapname)


def snap_delete(snapname, server=''):
    """
        Deletes snapshot and returns the output
    """
    if server == '':
        server = tc.nodes[0]
    cmd = "gluster snapshot delete %s --mode=script" % snapname
    return tc.run(server, cmd)


def snap_delete_all(volname, server=''):
    """
        Deletes one or more snaps and returns the output
    """
    if server == '':
        server = tc.nodes[0]
    cmd = 'ret=0; for i in `gluster snapshot list %s`; do \
gluster snapshot delete $i --mode=script || ret=1; done ; exit $ret' % volname
    return tc.run(server, cmd)


def snap_restore(volname, snapname, server=''):
    """
        stops the volume restore the snapshot and starts the volume

        Returns True upon success, False on in any step
    """
    if server == '':
        server = tc.nodes[0]
    ret = stop_volume(volname, server)
    if not ret:
        return False
    ret = tc.run(server, "gluster snapshot restore %s" % snapname)
    if ret[0] != 0:
        tc.logger.error("snapshot restore failed")
        return False
    ret = start_volume(volname, server)
    if not ret:
        return False
    return True