1 /*
2  * Copyright (C) 2019 The Android Open Source Project
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 
17 package android.net.ip;
18 
19 import static org.junit.Assert.assertArrayEquals;
20 import static org.junit.Assert.assertEquals;
21 import static org.mockito.ArgumentMatchers.any;
22 import static org.mockito.Mockito.doNothing;
23 import static org.mockito.Mockito.times;
24 import static org.mockito.Mockito.verify;
25 
26 import android.net.INetd;
27 import android.net.InetAddresses;
28 import android.net.InterfaceConfigurationParcel;
29 import android.net.LinkAddress;
30 import android.net.util.SharedLog;
31 
32 import androidx.test.filters.SmallTest;
33 import androidx.test.runner.AndroidJUnit4;
34 
35 import org.junit.Before;
36 import org.junit.Test;
37 import org.junit.runner.RunWith;
38 import org.mockito.ArgumentCaptor;
39 import org.mockito.Captor;
40 import org.mockito.Mock;
41 import org.mockito.MockitoAnnotations;
42 
43 import java.util.ArrayList;
44 
45 @RunWith(AndroidJUnit4.class)
46 @SmallTest
47 public class InterfaceControllerTest {
48     private static final String TEST_IFACE = "testif";
49     private static final String TEST_IPV4_ADDR = "192.168.123.28";
50     private static final int TEST_PREFIXLENGTH = 31;
51 
52     @Mock private INetd mNetd;
53     @Mock private SharedLog mLog;
54     @Captor private ArgumentCaptor<InterfaceConfigurationParcel> mConfigCaptor;
55 
56     private InterfaceController mController;
57 
58     @Before
setUp()59     public void setUp() throws Exception {
60         MockitoAnnotations.initMocks(this);
61         mController = new InterfaceController(TEST_IFACE, mNetd, mLog);
62 
63         doNothing().when(mNetd).interfaceSetCfg(mConfigCaptor.capture());
64     }
65 
66     @Test
testSetIPv4Address()67     public void testSetIPv4Address() throws Exception {
68         mController.setIPv4Address(
69                 new LinkAddress(InetAddresses.parseNumericAddress(TEST_IPV4_ADDR),
70                         TEST_PREFIXLENGTH));
71         verify(mNetd, times(1)).interfaceSetCfg(any());
72         final InterfaceConfigurationParcel parcel = mConfigCaptor.getValue();
73         assertEquals(TEST_IFACE, parcel.ifName);
74         assertEquals(TEST_IPV4_ADDR, parcel.ipv4Addr);
75         assertEquals(TEST_PREFIXLENGTH, parcel.prefixLength);
76         assertEquals("", parcel.hwAddr);
77         assertArrayEquals(new String[0], parcel.flags);
78     }
79 
80     @Test
testClearIPv4Address()81     public void testClearIPv4Address() throws Exception {
82         mController.clearIPv4Address();
83         verify(mNetd, times(1)).interfaceSetCfg(any());
84         final InterfaceConfigurationParcel parcel = mConfigCaptor.getValue();
85         assertEquals(TEST_IFACE, parcel.ifName);
86         assertEquals("0.0.0.0", parcel.ipv4Addr);
87         assertEquals(0, parcel.prefixLength);
88         assertEquals("", parcel.hwAddr);
89         assertArrayEquals(new String[0], parcel.flags);
90     }
91 }
92