/*
 * Copyright 2018 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package androidx.slice.render;

import static android.os.Build.VERSION.SDK_INT;

import static androidx.slice.render.SliceRenderer.SCREENSHOT_DIR;

import android.content.BroadcastReceiver;
import android.content.Context;
import android.content.Intent;
import android.content.IntentFilter;
import android.os.Build;
import android.support.test.InstrumentationRegistry;
import android.support.test.filters.SmallTest;
import android.support.test.runner.AndroidJUnit4;

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

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

@SmallTest
@RunWith(AndroidJUnit4.class)
public class RenderTest {

    private final Context mContext = InstrumentationRegistry.getContext();

    @Test
    public void testRender() throws Exception {
        final CountDownLatch latch = new CountDownLatch(1);
        BroadcastReceiver receiver = new BroadcastReceiver() {
            @Override
            public void onReceive(Context context, Intent intent) {
                latch.countDown();
            }
        };
        mContext.registerReceiver(receiver,
                new IntentFilter(SliceRenderActivity.ACTION_RENDER_DONE));
        mContext.startActivity(new Intent(mContext, SliceRenderActivity.class)
                .addFlags(Intent.FLAG_ACTIVITY_NEW_TASK));

        latch.await(30000, TimeUnit.MILLISECONDS);
        String path = new File(mContext.getFilesDir(), SCREENSHOT_DIR).getAbsolutePath();
        if (SDK_INT >= Build.VERSION_CODES.LOLLIPOP) {
            InstrumentationRegistry.getInstrumentation().getUiAutomation().executeShellCommand(
                    "mv " + path + " " + "/sdcard/");
        } else {
            deleteDir(new File("/sdcard/" + SCREENSHOT_DIR));
            copyDirectory(new File(path), new File("/sdcard/" + SCREENSHOT_DIR));
        }
    }


    public static void copyDirectory(File sourceLocation, File targetLocation) throws Exception {
        if (sourceLocation.isDirectory()) {
            if (!targetLocation.exists()) {
                targetLocation.mkdirs();
            }

            String[] children = sourceLocation.list();
            for (int i = 0; i < children.length; i++) {
                copyDirectory(new File(sourceLocation, children[i]), new File(
                        targetLocation, children[i]));
            }
        } else {
            copyFile(sourceLocation, targetLocation);
        }
    }

    public static void copyFile(File sourceLocation, File targetLocation) throws Exception {
        InputStream in = new FileInputStream(sourceLocation);
        OutputStream out = new FileOutputStream(targetLocation);

        byte[] buf = new byte[1024];
        int len;
        while ((len = in.read(buf)) > 0) {
            out.write(buf, 0, len);
        }
        in.close();
        out.close();
    }

    static void deleteDir(File file) {
        File[] contents = file.listFiles();
        if (contents != null) {
            for (File f : contents) {
                deleteDir(f);
            }
        }
        file.delete();
    }
}

@@ -758,16 +758,130 @@
 T_DIR;%0A%0A
+import static org.junit.Assert.assertEquals;%0Aimport static org.junit.Assert.assertTrue;%0A%0Aimport android.Manifest;%0A
 import a
@@ -910,24 +910,24 @@
 stReceiver;%0A
-
 import andro
@@ -1135,24 +1135,78 @@
 .SmallTest;%0A
+import android.support.test.rule.GrantPermissionRule;%0A
 import andro
@@ -1244,16 +1244,39 @@
 Unit4;%0A%0A
+import org.junit.Rule;%0A
 import o
@@ -1690,24 +1690,174 @@
 Context();%0A%0A
+    @Rule%0A    public GrantPermissionRule mRuntimePermissionRule =%0A            GrantPermissionRule.grant(Manifest.permission.WRITE_EXTERNAL_STORAGE);%0A%0A
     @Test%0A  
@@ -2484,16 +2484,59 @@
 CONDS);%0A
+        assertEquals(0, latch.getCount());%0A
         
@@ -3195,32 +3195,43 @@
                 
+assertTrue(
 targetLocation.m
@@ -3237,16 +3237,17 @@
 mkdirs()
+)
 ;%0A      

Fix render tests not running on new devices