1/*
2 * Copyright (C) 2016 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
17package com.android.setupwizardlib.test;
18
19import static org.junit.Assert.assertEquals;
20import static org.junit.Assert.assertSame;
21
22import android.support.test.filters.SmallTest;
23import android.support.test.runner.AndroidJUnit4;
24import android.text.Annotation;
25import android.text.SpannableStringBuilder;
26
27import com.android.setupwizardlib.span.SpanHelper;
28
29import org.junit.Test;
30import org.junit.runner.RunWith;
31
32@RunWith(AndroidJUnit4.class)
33@SmallTest
34public class SpanHelperTest {
35
36    @Test
37    public void testReplaceSpan() {
38        SpannableStringBuilder ssb = new SpannableStringBuilder("Hello world");
39        Annotation oldSpan = new Annotation("key", "value");
40        Annotation newSpan = new Annotation("newkey", "newvalue");
41        ssb.setSpan(oldSpan, 2, 5, 0 /* flags */);
42
43        SpanHelper.replaceSpan(ssb, oldSpan, newSpan);
44
45        final Object[] spans = ssb.getSpans(0, ssb.length(), Object.class);
46        assertEquals("There should be one span in the builder", 1, spans.length);
47        assertSame("The span should be newSpan", newSpan, spans[0]);
48    }
49}
50