Python AFF4ObjectLabelsListの例

コード例 #1

ファイル: aff4_test.py プロジェクト: rainser/grr

    def testRegexForStringifiedValueMatchMatchesLabelsInList(self):
        labels_list = rdf_aff4.AFF4ObjectLabelsList()

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="ein", owner="GRR"))
        labels_list.AddLabel(
            rdf_aff4.AFF4ObjectLabel(name="zwei", owner="test"))
        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="drei",
                                                      owner="GRR"))
        labels_list.AddLabel(
            rdf_aff4.AFF4ObjectLabel(name="vier", owner="test"))

        self.assertTrue(
            re.match(
                rdf_aff4.AFF4ObjectLabelsList.RegexForStringifiedValueMatch(
                    "ein"), str(labels_list)))
        self.assertTrue(
            re.match(
                rdf_aff4.AFF4ObjectLabelsList.RegexForStringifiedValueMatch(
                    "zwei"), str(labels_list)))
        self.assertTrue(
            re.match(
                rdf_aff4.AFF4ObjectLabelsList.RegexForStringifiedValueMatch(
                    "drei"), str(labels_list)))
        self.assertTrue(
            re.match(
                rdf_aff4.AFF4ObjectLabelsList.RegexForStringifiedValueMatch(
                    "vier"), str(labels_list)))

コード例 #2

ファイル: aff4_test.py プロジェクト: rainser/grr

    def testStringifiedRepresentationIsSorted(self):
        labels_list = rdf_aff4.AFF4ObjectLabelsList()

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo", owner="GRR"))
        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="bar",
                                                      owner="test"))

        self.assertEqual(utils.SmartStr(labels_list), "bar,foo")

コード例 #3

ファイル: aff4_test.py プロジェクト: rainser/grr

    def testStringifiedValueIsLabelsNamesWithoutOwners(self):
        labels_list = rdf_aff4.AFF4ObjectLabelsList()

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="bar", owner="GRR"))
        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo",
                                                      owner="test"))

        self.assertEqual(utils.SmartStr(labels_list), "bar,foo")

コード例 #4

ファイル: aff4_test.py プロジェクト: rainser/grr

    def testAddLabelAddsLabelWithSameNameButDifferentOwner(self):
        labels_list = rdf_aff4.AFF4ObjectLabelsList()

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo",
                                                      owner="test"))
        self.assertEqual(len(labels_list.labels), 1)

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo", owner="GRR"))
        self.assertEqual(len(labels_list.labels), 2)

コード例 #5

ファイル: aff4_test.py プロジェクト: rainser/grr

    def testStringifiedValueDoesNotHaveDuplicates(self):
        labels_list = rdf_aff4.AFF4ObjectLabelsList()

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo", owner="GRR"))
        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="bar", owner="GRR"))
        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo",
                                                      owner="test"))

        self.assertEqual(utils.SmartStr(labels_list), "bar,foo")

コード例 #6

ファイル: aff4_test.py プロジェクト: rainser/grr

    def testAddLabelDoesNotAddLabelWithSameNameAndOwner(self):
        labels_list = rdf_aff4.AFF4ObjectLabelsList()

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo",
                                                      owner="test"))
        self.assertEqual(len(labels_list.labels), 1)

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo",
                                                      owner="test"))
        self.assertEqual(len(labels_list.labels), 1)

コード例 #7

ファイル: aff4_test.py プロジェクト: rainser/grr

    def GenerateSample(self, number=0):
        label1 = rdf_aff4.AFF4ObjectLabel(
            name="foo_%d" % number,
            owner="test",
            timestamp=rdfvalue.RDFDatetime.FromSecondsSinceEpoch(42))

        label2 = rdf_aff4.AFF4ObjectLabel(
            name="bar_%d" % number,
            owner="test",
            timestamp=rdfvalue.RDFDatetime.FromSecondsSinceEpoch(42))

        return rdf_aff4.AFF4ObjectLabelsList(labels=[label1, label2])

コード例 #8

ファイル: aff4_test.py プロジェクト: rainser/grr

    def testGetSortedLabelSet(self):
        labels_list = rdf_aff4.AFF4ObjectLabelsList()

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="foo",
                                                      owner="test"))
        labels_list.AddLabel(
            rdf_aff4.AFF4ObjectLabel(name="foo2", owner="test2"))
        labels_list.AddLabel(
            rdf_aff4.AFF4ObjectLabel(name="foo3", owner="test2"))

        self.assertItemsEqual(labels_list.GetLabelNames(),
                              ["foo", "foo2", "foo3"])
        self.assertItemsEqual(labels_list.GetLabelNames(owner="test2"),
                              ["foo2", "foo3"])
        self.assertEqual(labels_list.GetLabelNames(owner="test4"), [])

コード例 #9

    def testRegexForStringifiedValueDoesNotMatchLabelsNotInList(self):
        labels_list = rdf_aff4.AFF4ObjectLabelsList()

        labels_list.AddLabel(rdf_aff4.AFF4ObjectLabel(name="ein", owner="GRR"))
        labels_list.AddLabel(
            rdf_aff4.AFF4ObjectLabel(name="zwei", owner="test"))
        self.assertNotRegexpMatches(
            str(labels_list),
            rdf_aff4.AFF4ObjectLabelsList.RegexForStringifiedValueMatch("e"))
        self.assertNotRegexpMatches(
            str(labels_list),
            rdf_aff4.AFF4ObjectLabelsList.RegexForStringifiedValueMatch("in"))
        self.assertNotRegexpMatches(
            str(labels_list),
            rdf_aff4.AFF4ObjectLabelsList.RegexForStringifiedValueMatch(
                "a.zwer"))
        self.assertNotRegexpMatches(
            str(labels_list),
            rdf_aff4.AFF4ObjectLabelsList.RegexForStringifiedValueMatch(
                "ein."))