diff options
author | Mehant Baid <mehantr@gmail.com> | 2014-09-17 16:50:42 -0700 |
---|---|---|
committer | Steven Phillips <sphillips@maprtech.com> | 2014-09-29 18:21:44 -0700 |
commit | a44cb9170c660cd9ee66750b0917d562942df1fb (patch) | |
tree | 6d3ec502f7ed38055dc93d6abec1fa2f0837bb8d /exec/java-exec/src/test/java/org/apache/drill/exec/pop/TestFragmentChecker.java | |
parent | 1c40b5e745ab20da6725568c9e4790f34ae73c8f (diff) |
DRILL-1432: Propagate user credentials from client to the fragments
Diffstat (limited to 'exec/java-exec/src/test/java/org/apache/drill/exec/pop/TestFragmentChecker.java')
-rw-r--r-- | exec/java-exec/src/test/java/org/apache/drill/exec/pop/TestFragmentChecker.java | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/exec/java-exec/src/test/java/org/apache/drill/exec/pop/TestFragmentChecker.java b/exec/java-exec/src/test/java/org/apache/drill/exec/pop/TestFragmentChecker.java index 58ddd0663..6349b767c 100644 --- a/exec/java-exec/src/test/java/org/apache/drill/exec/pop/TestFragmentChecker.java +++ b/exec/java-exec/src/test/java/org/apache/drill/exec/pop/TestFragmentChecker.java @@ -26,7 +26,9 @@ import org.apache.drill.exec.planner.fragment.SimpleParallelizer; import org.apache.drill.exec.planner.fragment.StatsCollector; import org.apache.drill.exec.proto.BitControl.PlanFragment; import org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint; +import org.apache.drill.exec.proto.UserBitShared; import org.apache.drill.exec.proto.UserBitShared.QueryId; +import org.apache.drill.exec.rpc.user.UserSession; import org.apache.drill.exec.server.options.OptionList; import org.apache.drill.exec.work.QueryWorkUnit; import org.junit.Test; @@ -59,7 +61,8 @@ public class TestFragmentChecker extends PopUnitTestBase{ endpoints.add(b1); } - QueryWorkUnit qwu = par.getFragments(new OptionList(), localBit, QueryId.getDefaultInstance(), endpoints, ppr, fragmentRoot, planningSet); + QueryWorkUnit qwu = par.getFragments(new OptionList(), localBit, QueryId.getDefaultInstance(), endpoints, ppr, fragmentRoot, planningSet, + UserSession.Builder.newBuilder().withCredentials(UserBitShared.UserCredentials.newBuilder().setUserName("foo").build()).build()); System.out.println(String.format("=========ROOT FRAGMENT [%d:%d] =========", qwu.getRootFragment().getHandle().getMajorFragmentId(), qwu.getRootFragment().getHandle().getMinorFragmentId())); System.out.print(qwu.getRootFragment().getFragmentJson()); |