Home | History | Annotate | Download | only in server 
      1 /*
      2  * Copyright (C) 2011 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 com.android.server;
     18 
     19 import static com.android.server.NativeDaemonConnector.appendEscaped;
     20 import static com.android.server.NativeDaemonConnector.makeCommand;
     21 
     22 import android.test.AndroidTestCase;
     23 import android.test.suitebuilder.annotation.MediumTest;
     24 
     25 import com.android.server.NativeDaemonConnector.SensitiveArg;
     26 
     27 /**
     28  * Tests for {@link NativeDaemonConnector}.
     29  */
     30 @MediumTest
     31 public class NativeDaemonConnectorTest extends AndroidTestCase {
     32     private static final String TAG = "NativeDaemonConnectorTest";
     33 
     34     public void testArgumentNormal() throws Exception {
     35         final StringBuilder builder = new StringBuilder();
     36 
     37         builder.setLength(0);
     38         appendEscaped(builder, "");
     39         assertEquals("", builder.toString());
     40 
     41         builder.setLength(0);
     42         appendEscaped(builder, "foo");
     43         assertEquals("foo", builder.toString());
     44 
     45         builder.setLength(0);
     46         appendEscaped(builder, "foo\"bar");
     47         assertEquals("foo\\\"bar", builder.toString());
     48 
     49         builder.setLength(0);
     50         appendEscaped(builder, "foo\\bar\\\"baz");
     51         assertEquals("foo\\\\bar\\\\\\\"baz", builder.toString());
     52     }
     53 
     54     public void testArgumentWithSpaces() throws Exception {
     55         final StringBuilder builder = new StringBuilder();
     56 
     57         builder.setLength(0);
     58         appendEscaped(builder, "foo bar");
     59         assertEquals("\"foo bar\"", builder.toString());
     60 
     61         builder.setLength(0);
     62         appendEscaped(builder, "foo\"bar\\baz foo");
     63         assertEquals("\"foo\\\"bar\\\\baz foo\"", builder.toString());
     64     }
     65 
     66     public void testArgumentWithUtf() throws Exception {
     67         final StringBuilder builder = new StringBuilder();
     68 
     69         builder.setLength(0);
     70         appendEscaped(builder, "caf\u00E9 c\u00F6ffee");
     71         assertEquals("\"caf\u00E9 c\u00F6ffee\"", builder.toString());
     72     }
     73 
     74     public void testSensitiveArgs() throws Exception {
     75         final StringBuilder rawBuilder = new StringBuilder();
     76         final StringBuilder logBuilder = new StringBuilder();
     77 
     78         rawBuilder.setLength(0);
     79         logBuilder.setLength(0);
     80         makeCommand(rawBuilder, logBuilder, 1, "foo", "bar", "baz");
     81         assertEquals("1 foo bar baz\0", rawBuilder.toString());
     82         assertEquals("1 foo bar baz", logBuilder.toString());
     83 
     84         rawBuilder.setLength(0);
     85         logBuilder.setLength(0);
     86         makeCommand(rawBuilder, logBuilder, 1, "foo", new SensitiveArg("bar"), "baz");
     87         assertEquals("1 foo bar baz\0", rawBuilder.toString());
     88         assertEquals("1 foo [scrubbed] baz", logBuilder.toString());
     89 
     90         rawBuilder.setLength(0);
     91         logBuilder.setLength(0);
     92         makeCommand(rawBuilder, logBuilder, 1, "foo", new SensitiveArg("foo bar"), "baz baz",
     93                 new SensitiveArg("wat"));
     94         assertEquals("1 foo \"foo bar\" \"baz baz\" wat\0", rawBuilder.toString());
     95         assertEquals("1 foo [scrubbed] \"baz baz\" [scrubbed]", logBuilder.toString());
     96     }
     97 }
     98