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