Advertisement
Guest User

Untitled

a guest
Apr 26th, 2017
74
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.05 KB | None | 0 0
  1. @Test
  2. public void testGroupByWithLookupDimensionSpec()
  3. {
  4. Map<String, String> map = new HashMap<>();
  5. map.put("automotive", "health105");
  6. map.put("business", "health20");
  7. map.put("entertainment", "travel47");
  8. map.put("health", "health55");
  9. map.put("mezzanine", "health09");
  10. map.put("news", "health0000");
  11. map.put("premium", "health999");
  12. map.put("technology", "travel123");
  13. map.put("travel", "travel555");
  14.  
  15. GroupByQuery innerQuery = GroupByQuery
  16. .builder()
  17. .setDataSource(QueryRunnerTestHelper.dataSource)
  18. .setQuerySegmentSpec(QueryRunnerTestHelper.firstToThird)
  19. .setDimensions(
  20. Lists.<DimensionSpec>newArrayList(
  21. new LookupDimensionSpec(
  22. "quality",
  23. "alias",
  24. new MapLookupExtractor(map, false),
  25. false,
  26. null,
  27. null,
  28. null,
  29. true
  30. )
  31. )
  32. )
  33. .setAggregatorSpecs(
  34. Arrays.<AggregatorFactory>asList(
  35. QueryRunnerTestHelper.rowsCount
  36. )
  37. )
  38. .setGranularity(QueryRunnerTestHelper.allGran)
  39. .build();
  40.  
  41. GroupByQuery outerQuery = GroupByQuery
  42. .builder()
  43. .setDataSource(new QueryDataSource(innerQuery))
  44. .setQuerySegmentSpec(QueryRunnerTestHelper.firstToThird)
  45. .setDimensions(
  46. Lists.<DimensionSpec>newArrayList(
  47. new DefaultDimensionSpec("alias", "alias")
  48. )
  49. )
  50. .setAggregatorSpecs(
  51. Arrays.<AggregatorFactory>asList(
  52. new LongSumAggregatorFactory(
  53. QueryRunnerTestHelper.rowsCount.getName(),
  54. QueryRunnerTestHelper.rowsCount.getName()
  55. )
  56. )
  57. )
  58. .setGranularity(QueryRunnerTestHelper.allGran)
  59. .build();
  60.  
  61. List<Row> expectedResults = Arrays.asList(
  62. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "health0000", "rows", 2L),
  63. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "health09", "rows", 6L),
  64. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "health105", "rows", 2L),
  65. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "health20", "rows", 2L),
  66. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "health55", "rows", 2L),
  67. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "health999", "rows", 6L),
  68. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "travel123", "rows", 2L),
  69. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "travel47", "rows", 2L),
  70. GroupByQueryRunnerTestHelper.createExpectedRow("2011-04-01", "alias", "travel555", "rows", 2L)
  71. );
  72.  
  73. Iterable<Row> results = GroupByQueryRunnerTestHelper.runQuery(factory, runner, outerQuery);
  74. TestHelper.assertExpectedObjects(expectedResults, results, "");
  75. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement