Add unit tests for Installing phase
Change-Id: I8955d20d523daf3abd8e1ef597bf5c26254ca68a
Showing
1 changed file
with
144 additions
and
0 deletions
1 | +/* | ||
2 | + * Copyright 2015 Open Networking Laboratory | ||
3 | + * | ||
4 | + * Licensed under the Apache License, Version 2.0 (the "License"); | ||
5 | + * you may not use this file except in compliance with the License. | ||
6 | + * You may obtain a copy of the License at | ||
7 | + * | ||
8 | + * http://www.apache.org/licenses/LICENSE-2.0 | ||
9 | + * | ||
10 | + * Unless required by applicable law or agreed to in writing, software | ||
11 | + * distributed under the License is distributed on an "AS IS" BASIS, | ||
12 | + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
13 | + * See the License for the specific language governing permissions and | ||
14 | + * limitations under the License. | ||
15 | + */ | ||
16 | +package org.onosproject.net.intent.impl.phase; | ||
17 | + | ||
18 | +import org.junit.After; | ||
19 | +import org.junit.Before; | ||
20 | +import org.junit.Test; | ||
21 | +import org.onosproject.TestApplicationId; | ||
22 | +import org.onosproject.core.ApplicationId; | ||
23 | +import org.onosproject.core.IdGenerator; | ||
24 | +import org.onosproject.net.ConnectPoint; | ||
25 | +import org.onosproject.net.DefaultLink; | ||
26 | +import org.onosproject.net.DefaultPath; | ||
27 | +import org.onosproject.net.Link; | ||
28 | +import org.onosproject.net.Path; | ||
29 | +import org.onosproject.net.flow.DefaultTrafficSelector; | ||
30 | +import org.onosproject.net.flow.DefaultTrafficTreatment; | ||
31 | +import org.onosproject.net.flow.FlowRuleOperations; | ||
32 | +import org.onosproject.net.flow.TrafficSelector; | ||
33 | +import org.onosproject.net.flow.TrafficTreatment; | ||
34 | +import org.onosproject.net.intent.Intent; | ||
35 | +import org.onosproject.net.intent.IntentData; | ||
36 | +import org.onosproject.net.intent.MockIdGenerator; | ||
37 | +import org.onosproject.net.intent.PathIntent; | ||
38 | +import org.onosproject.net.intent.PointToPointIntent; | ||
39 | +import org.onosproject.net.intent.impl.IntentInstallationException; | ||
40 | +import org.onosproject.net.intent.impl.IntentProcessor; | ||
41 | +import org.onosproject.net.provider.ProviderId; | ||
42 | +import org.onosproject.store.Timestamp; | ||
43 | + | ||
44 | +import java.util.Arrays; | ||
45 | +import java.util.List; | ||
46 | +import java.util.Optional; | ||
47 | + | ||
48 | +import static org.easymock.EasyMock.createMock; | ||
49 | +import static org.easymock.EasyMock.expectLastCall; | ||
50 | +import static org.easymock.EasyMock.replay; | ||
51 | +import static org.easymock.EasyMock.verify; | ||
52 | +import static org.hamcrest.Matchers.instanceOf; | ||
53 | +import static org.hamcrest.Matchers.is; | ||
54 | +import static org.junit.Assert.assertThat; | ||
55 | +import static org.onosproject.net.DeviceId.deviceId; | ||
56 | +import static org.onosproject.net.Link.Type.DIRECT; | ||
57 | +import static org.onosproject.net.PortNumber.portNumber; | ||
58 | +import static org.onosproject.net.intent.IntentState.INSTALL_REQ; | ||
59 | + | ||
60 | +/** | ||
61 | + * Unit tests for Installing phase. | ||
62 | + */ | ||
63 | +public class InstallingTest { | ||
64 | + | ||
65 | + private final ApplicationId appId = new TestApplicationId("test"); | ||
66 | + private final ProviderId pid = new ProviderId("of", "test"); | ||
67 | + private final TrafficSelector selector = DefaultTrafficSelector.builder().build(); | ||
68 | + private final TrafficTreatment treatment = DefaultTrafficTreatment.builder().build(); | ||
69 | + private final ConnectPoint cp1 = new ConnectPoint(deviceId("1"), portNumber(1)); | ||
70 | + private final ConnectPoint cp2 = new ConnectPoint(deviceId("1"), portNumber(2)); | ||
71 | + private final ConnectPoint cp3 = new ConnectPoint(deviceId("2"), portNumber(1)); | ||
72 | + private final ConnectPoint cp4 = new ConnectPoint(deviceId("2"), portNumber(2)); | ||
73 | + | ||
74 | + private final List<Link> links = Arrays.asList(new DefaultLink(pid, cp2, cp4, DIRECT)); | ||
75 | + private final Path path = new DefaultPath(pid, links, 10); | ||
76 | + | ||
77 | + private PointToPointIntent input; | ||
78 | + private PathIntent compiled; | ||
79 | + | ||
80 | + private IdGenerator idGenerator; | ||
81 | + private IntentProcessor processor; | ||
82 | + private Timestamp version; | ||
83 | + | ||
84 | + @Before | ||
85 | + public void setUp() { | ||
86 | + processor = createMock(IntentProcessor.class); | ||
87 | + version = createMock(Timestamp.class); | ||
88 | + | ||
89 | + idGenerator = new MockIdGenerator(); | ||
90 | + | ||
91 | + Intent.bindIdGenerator(idGenerator); | ||
92 | + | ||
93 | + // Intent creation should be placed after binding an ID generator | ||
94 | + input = new PointToPointIntent(appId, selector, treatment, cp1, cp3); | ||
95 | + compiled = new PathIntent(appId, selector, treatment, path); | ||
96 | + } | ||
97 | + | ||
98 | + | ||
99 | + @After | ||
100 | + public void tearDown() { | ||
101 | + Intent.unbindIdGenerator(idGenerator); | ||
102 | + } | ||
103 | + | ||
104 | + /** | ||
105 | + * Tests a next phase when no exception occurs. | ||
106 | + */ | ||
107 | + @Test | ||
108 | + public void testMoveToNextPhaseWithoutError() { | ||
109 | + IntentData pending = new IntentData(input, INSTALL_REQ, version); | ||
110 | + pending.setInstallables(Arrays.asList(compiled)); | ||
111 | + | ||
112 | + FlowRuleOperations operations = createMock(FlowRuleOperations.class); | ||
113 | + | ||
114 | + processor.applyFlowRules(operations); | ||
115 | + replay(processor); | ||
116 | + | ||
117 | + Installing sut = new Installing(processor, pending, operations); | ||
118 | + | ||
119 | + Optional<IntentProcessPhase> executed = sut.execute(); | ||
120 | + verify(processor); | ||
121 | + assertThat(executed.get(), is(instanceOf(Installed.class))); | ||
122 | + } | ||
123 | + | ||
124 | + /** | ||
125 | + * Test a next phase when IntentInstallationException occurs. | ||
126 | + */ | ||
127 | + @Test | ||
128 | + public void testWhenIntentInstallationExceptionOccurs() { | ||
129 | + IntentData pending = new IntentData(input, INSTALL_REQ, version); | ||
130 | + pending.setInstallables(Arrays.asList(compiled)); | ||
131 | + | ||
132 | + FlowRuleOperations operations = createMock(FlowRuleOperations.class); | ||
133 | + | ||
134 | + processor.applyFlowRules(operations); | ||
135 | + expectLastCall().andThrow(new IntentInstallationException()); | ||
136 | + replay(processor); | ||
137 | + | ||
138 | + Installing sut = new Installing(processor, pending, operations); | ||
139 | + | ||
140 | + Optional<IntentProcessPhase> executed = sut.execute(); | ||
141 | + verify(processor); | ||
142 | + assertThat(executed.get(), is(instanceOf(InstallingFailed.class))); | ||
143 | + } | ||
144 | +} |
-
Please register or login to post a comment