blob: 406915bdf08ec5da950444fce52a42da1ac4cc28 (
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
|
# Copyright 2013-2019 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.pattern as pattern
@pytest.fixture()
def interface():
"""Returns the interface class for the composite."""
class Base:
counter = 0
def add(self):
raise NotImplementedError('add not implemented')
def subtract(self):
raise NotImplementedError('subtract not implemented')
return Base
@pytest.fixture()
def implementation(interface):
"""Returns an implementation of the interface"""
class Implementation(interface):
def __init__(self, value):
self.value = value
def add(self):
interface.counter += self.value
def subtract(self):
interface.counter -= self.value
return Implementation
@pytest.fixture(params=[
'interface',
'method_list'
])
def composite(interface, implementation, request):
"""Returns a composite that contains an instance of `implementation(1)`
and one of `implementation(2)`.
"""
if request.param == 'interface':
@pattern.composite(interface=interface)
class Composite:
pass
else:
@pattern.composite(method_list=['add', 'subtract'])
class Composite:
pass
c = Composite()
c.append(implementation(1))
c.append(implementation(2))
return c
def test_composite_interface_calls(interface, composite):
composite.add()
assert interface.counter == 3
composite.pop()
composite.subtract()
assert interface.counter == 2
def test_composite_wrong_container(interface):
with pytest.raises(TypeError):
@pattern.composite(interface=interface, container=2)
class CompositeFromInterface:
pass
def test_composite_no_methods():
with pytest.raises(TypeError):
@pattern.composite()
class CompositeFromInterface:
pass
|
