summaryrefslogtreecommitdiff
path: root/lib/spack/spack/test/namespace_trie.py
blob: 39f7d3ca844cc65477a293bf7c1e8cb82467efd7 (plain) (blame)
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
# Copyright 2013-2023 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

import pytest

import spack.util.naming


@pytest.fixture()
def trie():
    return spack.util.naming.NamespaceTrie()


def test_add_single(trie):
    trie["foo"] = "bar"

    assert trie.is_prefix("foo")
    assert trie.has_value("foo")
    assert trie["foo"] == "bar"


def test_add_multiple(trie):
    trie["foo.bar"] = "baz"

    assert not trie.has_value("foo")
    assert trie.is_prefix("foo")

    assert trie.is_prefix("foo.bar")
    assert trie.has_value("foo.bar")
    assert trie["foo.bar"] == "baz"

    assert not trie.is_prefix("foo.bar.baz")
    assert not trie.has_value("foo.bar.baz")


def test_add_three(trie):
    # add a three-level namespace
    trie["foo.bar.baz"] = "quux"

    assert trie.is_prefix("foo")
    assert not trie.has_value("foo")

    assert trie.is_prefix("foo.bar")
    assert not trie.has_value("foo.bar")

    assert trie.is_prefix("foo.bar.baz")
    assert trie.has_value("foo.bar.baz")
    assert trie["foo.bar.baz"] == "quux"

    assert not trie.is_prefix("foo.bar.baz.quux")
    assert not trie.has_value("foo.bar.baz.quux")

    # Try to add a second element in a prefix namespace
    trie["foo.bar"] = "blah"

    assert trie.is_prefix("foo")
    assert not trie.has_value("foo")

    assert trie.is_prefix("foo.bar")
    assert trie.has_value("foo.bar")
    assert trie["foo.bar"] == "blah"

    assert trie.is_prefix("foo.bar.baz")
    assert trie.has_value("foo.bar.baz")
    assert trie["foo.bar.baz"] == "quux"

    assert not trie.is_prefix("foo.bar.baz.quux")
    assert not trie.has_value("foo.bar.baz.quux")


def test_add_none_single(trie):
    trie["foo"] = None
    assert trie.is_prefix("foo")
    assert trie.has_value("foo")
    assert trie["foo"] is None

    assert not trie.is_prefix("foo.bar")
    assert not trie.has_value("foo.bar")


def test_add_none_multiple(trie):
    trie["foo.bar"] = None

    assert trie.is_prefix("foo")
    assert not trie.has_value("foo")

    assert trie.is_prefix("foo.bar")
    assert trie.has_value("foo.bar")
    assert trie["foo.bar"] is None

    assert not trie.is_prefix("foo.bar.baz")
    assert not trie.has_value("foo.bar.baz")