chromium/chrome/browser/download/android/java/src/org/chromium/chrome/browser/download/MimeUtilsTest.java

// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.chrome.browser.download;

import androidx.test.filters.SmallTest;

import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;

import org.chromium.base.test.util.Batch;
import org.chromium.base.test.util.Feature;
import org.chromium.chrome.test.ChromeJUnit4ClassRunner;

/** Tests for MimeUtils class. */
@RunWith(ChromeJUnit4ClassRunner.class)
@Batch(Batch.UNIT_TESTS)
public class MimeUtilsTest {
    @Test
    @SmallTest
    @Feature({"Download"})
    public void testGetFileExtension() {
        Assert.assertEquals("ext", MimeUtils.getFileExtension("", "file.ext"));
        Assert.assertEquals("ext", MimeUtils.getFileExtension("http://file.ext", ""));
        Assert.assertEquals("txt", MimeUtils.getFileExtension("http://file.ext", "file.txt"));
        Assert.assertEquals("txt", MimeUtils.getFileExtension("http://file.ext", "file name.txt"));
    }

    /**
     * Test to make sure {@link DownloadUtils#shouldAutoOpenDownload} returns the right result for
     * varying MIME types and Content-Dispositions.
     */
    @Test
    @SmallTest
    @Feature({"Download"})
    public void testCanAutoOpenMimeType() {
        // Should not open any download type MIME types.
        Assert.assertFalse(MimeUtils.canAutoOpenMimeType("application/download"));
        Assert.assertFalse(MimeUtils.canAutoOpenMimeType("application/x-download"));
        Assert.assertFalse(MimeUtils.canAutoOpenMimeType("application/octet-stream"));
        Assert.assertTrue(MimeUtils.canAutoOpenMimeType("application/pdf"));
        Assert.assertTrue(MimeUtils.canAutoOpenMimeType("application/x-x509-server-cert"));
        Assert.assertTrue(MimeUtils.canAutoOpenMimeType("application/x-wifi-config"));
        Assert.assertTrue(MimeUtils.canAutoOpenMimeType("application/pkix-cert"));
    }
}