Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 1 | /* |
Brian O'Connor | a09fe5b | 2017-08-03 21:12:30 -0700 | [diff] [blame] | 2 | * Copyright 2017-present Open Networking Foundation |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 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.mastership; |
| 17 | |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 18 | import com.google.common.collect.ImmutableMap; |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 19 | import com.google.common.testing.EqualsTester; |
| 20 | import org.junit.Test; |
| 21 | |
| 22 | import org.onosproject.cluster.NodeId; |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 23 | import org.onosproject.net.DeviceId; |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 24 | import org.onosproject.net.MastershipRole; |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 25 | |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 26 | import java.util.Optional; |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 27 | |
| 28 | import static org.hamcrest.Matchers.is; |
| 29 | import static org.junit.Assert.assertThat; |
| 30 | |
| 31 | /** |
| 32 | * Unit tests for the mastership event. |
| 33 | */ |
| 34 | public class MastershipEventTest { |
| 35 | private final long time = System.currentTimeMillis(); |
| 36 | private final DeviceId deviceId1 = DeviceId.deviceId("foo:bar"); |
| 37 | private final DeviceId deviceId2 = DeviceId.deviceId("bar:baz"); |
| 38 | private final NodeId node1 = new NodeId("1"); |
| 39 | private final NodeId node2 = new NodeId("2"); |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 40 | private final MastershipInfo mastershipInfo1 = new MastershipInfo( |
| 41 | 1, |
| 42 | Optional.of(node1), |
| 43 | ImmutableMap.<NodeId, MastershipRole>builder() |
| 44 | .put(node1, MastershipRole.MASTER) |
| 45 | .put(node2, MastershipRole.STANDBY) |
| 46 | .build()); |
| 47 | private final MastershipInfo mastershipInfo2 = new MastershipInfo( |
| 48 | 2, |
| 49 | Optional.of(node1), |
| 50 | ImmutableMap.<NodeId, MastershipRole>builder() |
| 51 | .put(node2, MastershipRole.MASTER) |
| 52 | .put(node1, MastershipRole.STANDBY) |
| 53 | .build()); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 54 | |
| 55 | private final MastershipEvent event1 = |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 56 | new MastershipEvent(MastershipEvent.Type.BACKUPS_CHANGED, deviceId1, mastershipInfo1); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 57 | private final MastershipEvent event2 = |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 58 | new MastershipEvent(MastershipEvent.Type.MASTER_CHANGED, deviceId1, mastershipInfo1); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 59 | private final MastershipEvent event3 = |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 60 | new MastershipEvent(MastershipEvent.Type.SUSPENDED, deviceId1, mastershipInfo1); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 61 | private final MastershipEvent event4 = |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 62 | new MastershipEvent(MastershipEvent.Type.MASTER_CHANGED, deviceId1, mastershipInfo2, time); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 63 | private final MastershipEvent sameAsEvent4 = |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 64 | new MastershipEvent(MastershipEvent.Type.MASTER_CHANGED, deviceId1, mastershipInfo2, time); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 65 | private final MastershipEvent event5 = |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 66 | new MastershipEvent(MastershipEvent.Type.BACKUPS_CHANGED, deviceId2, mastershipInfo1); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 67 | |
| 68 | /** |
| 69 | * Tests for proper operation of equals(), hashCode() and toString() methods. |
| 70 | */ |
| 71 | @Test |
| 72 | public void checkEquals() { |
| 73 | new EqualsTester() |
| 74 | .addEqualityGroup(event1) |
| 75 | .addEqualityGroup(event2) |
| 76 | .addEqualityGroup(event3) |
| 77 | .addEqualityGroup(event4, sameAsEvent4) |
| 78 | .addEqualityGroup(event5) |
| 79 | .testEquals(); |
| 80 | } |
| 81 | |
| 82 | /** |
| 83 | * Tests that objects are created properly. |
| 84 | */ |
| 85 | @Test |
| 86 | public void checkConstruction() { |
| 87 | assertThat(event1.type(), is(MastershipEvent.Type.BACKUPS_CHANGED)); |
| 88 | assertThat(event1.subject(), is(deviceId1)); |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 89 | assertThat(event1.mastershipInfo(), is(mastershipInfo1)); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 90 | |
| 91 | assertThat(event4.time(), is(time)); |
| 92 | assertThat(event4.type(), is(MastershipEvent.Type.MASTER_CHANGED)); |
| 93 | assertThat(event4.subject(), is(deviceId1)); |
Jordan Halterman | 0a2bd45 | 2018-06-13 17:24:58 -0700 | [diff] [blame] | 94 | assertThat(event4.mastershipInfo(), is(mastershipInfo2)); |
Jon Hall | 672a3ed | 2017-04-05 13:03:37 -0700 | [diff] [blame] | 95 | } |
| 96 | |
| 97 | } |