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.documentsui.services; 18 19import static junit.framework.Assert.assertFalse; 20import static junit.framework.Assert.assertTrue; 21 22import android.app.Notification; 23import android.app.Notification.Builder; 24import android.content.Context; 25 26import com.android.documentsui.base.Features; 27import com.android.documentsui.clipping.UrisSupplier; 28import com.android.documentsui.R; 29import com.android.documentsui.base.DocumentInfo; 30import com.android.documentsui.base.DocumentStack; 31import com.android.documentsui.services.FileOperationService.OpType; 32 33import java.text.NumberFormat; 34 35public class TestJob extends Job { 36 37 private boolean mStarted; 38 private Runnable mStartRunnable; 39 40 private int mNumOfNotifications = 0; 41 42 TestJob(Context service, Listener listener, String id, @OpType int opType, DocumentStack stack, 43 UrisSupplier srcs, Runnable startRunnable, Features features) { 44 super(service, listener, id, opType, stack, srcs, features); 45 46 mStartRunnable = startRunnable; 47 } 48 49 @Override 50 void start() { 51 mStarted = true; 52 53 mStartRunnable.run(); 54 } 55 56 void assertStarted() { 57 assertTrue(mStarted); 58 } 59 60 void assertNotStarted() { 61 assertFalse(mStarted); 62 } 63 64 void fail(DocumentInfo doc) { 65 onFileFailed(doc); 66 } 67 68 int getNumOfNotifications() { 69 return mNumOfNotifications; 70 } 71 72 @Override 73 Notification getSetupNotification() { 74 ++mNumOfNotifications; 75 return getSetupNotification(service.getString(R.string.copy_preparing)); 76 } 77 78 @Override 79 Notification getProgressNotification() { 80 ++mNumOfNotifications; 81 double completed = mStarted ? 1F : 0F; 82 return mProgressBuilder 83 .setProgress(1, (int) completed, true) 84 .setSubText(NumberFormat.getPercentInstance().format(completed)) 85 .build(); 86 } 87 88 @Override 89 Notification getFailureNotification() { 90 // the "copy" stuff was just convenient and available :) 91 return getFailureNotification( 92 R.plurals.copy_error_notification_title, R.drawable.ic_menu_copy); 93 } 94 95 @Override 96 Notification getWarningNotification() { 97 throw new UnsupportedOperationException(); 98 } 99 100 @Override 101 Builder createProgressBuilder() { 102 ++mNumOfNotifications; 103 // the "copy" stuff was just convenient and available :) 104 return super.createProgressBuilder( 105 service.getString(R.string.copy_notification_title), 106 R.drawable.ic_menu_copy, 107 service.getString(android.R.string.cancel), 108 R.drawable.ic_cab_cancel); 109 } 110} 111