mobile/android/geckoview/src/androidTest/java/org/mozilla/geckoview/test/TemporaryProfileRule.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

package org.mozilla.geckoview.test;

import java.io.File;
import java.io.IOException;
import org.junit.rules.ExternalResource;
import org.junit.rules.TemporaryFolder;
import org.mozilla.geckoview.test.rule.TestHarnessException;

/** Lazily provides a temporary profile folder for tests. */
public class TemporaryProfileRule extends ExternalResource {
  TemporaryFolder mTemporaryFolder;
  File mProfileFolder;

  @Override
  protected void after() {
    if (mTemporaryFolder != null) {
      mTemporaryFolder.delete();
      mProfileFolder = null;
    }
  }

  public File get() {
    if (mProfileFolder == null) {
      mTemporaryFolder = new TemporaryFolder();
      try {
        mTemporaryFolder.create();
        mProfileFolder = mTemporaryFolder.newFolder("test-profile");
      } catch (IOException ex) {
        throw new TestHarnessException(ex);
      }
    }

    return mProfileFolder;
  }
}