summaryrefslogtreecommitdiffstats
path: root/src/software/test/test_software_identity.py
blob: 47eee28047f1517262784a8b8bd8fe57ca0c49f3 (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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
#!/usr/bin/env python
#
# Copyright (C) 2012-2013 Red Hat, Inc.  All rights reserved.
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
#
# Authors: Michal Minar <miminar@redhat.com>
#
"""
Unit tests for LMI_SoftwareIdentity provider.
"""

from datetime import datetime, timedelta
import itertools
import pywbem
import unittest

import base
import rpmcache
import util

class TestSoftwareIdentity(base.SoftwareBaseTestCase): #pylint: disable=R0904
    """
    Basic cim operations test.
    """

    CLASS_NAME = "LMI_SoftwareIdentity"
    KEYS = ("InstanceID", )

    def make_op(self, pkg, newer=True):
        """
        @return object path of SoftwareIdentity
        """
        objpath = self.objpath.copy()
        objpath["InstanceID"] = 'LMI:LMI_SoftwareIdentity:'+pkg.get_nevra(newer, "ALWAYS")
        return objpath

    @base.mark_dangerous
    def test_get_instance(self):
        """
        Dangerous test, making sure, that properties are set correctly
        for installed and removed packages.
        """
        for pkg in self.dangerous_pkgs:
            if rpmcache.is_pkg_installed(pkg.name):
                rpmcache.remove_pkg(pkg.name)
            objpath = self.make_op(pkg)
            inst = self.conn.GetInstance(InstanceName=objpath, LocalOnly=False)
            self.assertIsNone(inst["InstallDate"])
            time_stamp = datetime.now(pywbem.cim_types.MinutesFromUTC(0)) \
                    - timedelta(seconds=0.01)
            rpmcache.install_pkg(pkg)
            inst2 = self.conn.GetInstance(
                    InstanceName=objpath, LocalOnly=False)
            self.assertIsNotNone(inst2["InstallDate"])
            self.assertGreater(inst2["InstallDate"].datetime, time_stamp)

    def test_get_instance_safe(self):
        """
        Tests GetInstance call on packages from our rpm cache.
        """
        for pkg in self.safe_pkgs:
            objpath = self.make_op(pkg)
            inst = self.conn.GetInstance(InstanceName=objpath, LocalOnly=False)
            self.assertEqual(inst.path, objpath,
                    "Object paths should match for package %s"%pkg)
            for key in self.KEYS:
                self.assertTrue(inst.properties.has_key(key),
                        "OP is missing \"%s\" key for package %s"%(key, pkg))
            self.assertIsInstance(inst["Caption"], basestring)
            self.assertIsInstance(inst["Description"], basestring)
            self.assertEqual(pkg.up_evra, inst["VersionString"],
                    "VersionString does not match evra for pkg %s" % pkg)
            self.assertTrue(inst['IsEntity'])
            self.assertEqual(pkg.name, inst["Name"],
                    "Name does not match for pkg %s" % pkg)
            self.assertIsInstance(inst["Epoch"], pywbem.Uint32,
                    "Epoch does not match for pkg %s" % pkg)
            self.assertEqual(int(pkg.epoch), inst["Epoch"])
            self.assertEqual(pkg.ver, inst["Version"],
                    "Version does not match for pkg %s" % pkg)
            self.assertEqual(pkg.rel, inst["Release"],
                    "Release does not match for pkg %s" % pkg)
            self.assertEqual(pkg.arch, inst["Architecture"],
                    "Architecture does not match for pkg %s" % pkg)

            # try to leave out epoch part
            if pkg.epoch == "0":
                op_no_epoch = objpath.copy()
                op_no_epoch["SoftwareElementID"] = util.make_nevra(pkg.name,
                        pkg.up_epoch, pkg.up_ver, pkg.up_rel, pkg.arch, "NEVER")
                self.assertNotIn(":", op_no_epoch["SoftwareElementID"])
                inst = self.conn.GetInstance(
                        InstanceName=op_no_epoch, LocalOnly=False)
                self.assertIn(inst.path, (objpath, op_no_epoch))

    @base.mark_tedious
    def test_enum_instance_names_safe(self):
        """
        Tests EnumInstanceNames call on installed packages.
        """
        inames = self.conn.EnumerateInstanceNames(ClassName=self.CLASS_NAME)
        self.assertGreater(len(inames), 0)
        for iname in inames:
            self.assertIsInstance(iname, pywbem.CIMInstanceName)
            self.assertEqual(iname.namespace, 'root/cimv2')
            self.assertEqual(sorted(iname.keys()), sorted(self.KEYS))
        nevra_set = set(i["InstanceID"] for i in inames)
        for pkg in self.safe_pkgs:
            self.assertIn('LMI:LMI_SoftwareIdentity:'+pkg.get_nevra(with_epoch="ALWAYS"),
                    nevra_set)

#    @base.mark_tedious
#    def test_enum_instances(self):
#        """
#        Tests EnumInstances call on installed packages.
#        """
#        insts = self.conn.EnumerateInstances(ClassName=self.CLASS_NAME)
#        self.assertGreater(len(insts), 0)
#        for inst in insts:
#            self.assertIsInstance(inst, pywbem.CIMInstance)
#            self.assertEqual(inst.namespace, 'root/cimv2')
#            self.assertEqual(sorted(inst.keys()), sorted(self.KEYS))
#            self.assertEqual(inst["InstanceID"], inst.path["InstanceID"])
#        nevra_set = set()
#        name_set = set()
#        for inst in insts:
#            nevra_set.add(inst["InstanceID"])
#            name_set.add(inst["Name"])
#        for pkg in self.safe_pkgs:
#            self.assertIn("LMI:LMI_SoftwareIdentity:"+pkg.get_nevra(with_epoch="ALWAYS"),
#                    nevra_set)
#            self.assertIn(pkg.name, name_set)

def suite():
    """For unittest loaders."""
    return unittest.TestLoader().loadTestsFromTestCase(
            TestSoftwareIdentity)

if __name__ == '__main__':
    unittest.main()