{"action":"created","comment":{"url":"https://api.github.com/repos/caskdata/cdap/pulls/comments/39017849","id":39017849,"diff_hunk":"@@ -0,0 +1,124 @@\n+/*\n+ * Copyright © 2015 Cask Data, Inc.\n+ *\n+ * Licensed under the Apache License, Version 2.0 (the \"License\"); you may not\n+ * use this file except in compliance with the License. You may obtain a copy of\n+ * the License at\n+ *\n+ * http://www.apache.org/licenses/LICENSE-2.0\n+ *\n+ * Unless required by applicable law or agreed to in writing, software\n+ * distributed under the License is distributed on an \"AS IS\" BASIS, WITHOUT\n+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the\n+ * License for the specific language governing permissions and limitations under\n+ * the License.\n+ */\n+\n+package co.cask.cdap.data2.metadata.lineage;\n+\n+import co.cask.cdap.api.dataset.DatasetProperties;\n+import co.cask.cdap.common.app.RunIds;\n+import co.cask.cdap.data2.datafabric.dataset.DatasetsUtil;\n+import co.cask.cdap.data2.dataset2.DatasetFrameworkTestUtil;\n+import co.cask.cdap.proto.Id;\n+import co.cask.cdap.proto.ProgramType;\n+import com.google.common.collect.ImmutableSet;\n+import org.apache.twill.api.RunId;\n+import org.junit.Assert;\n+import org.junit.ClassRule;\n+import org.junit.Test;\n+\n+import java.util.Set;\n+\n+/**\n+ * Tests lineage computation.\n+ */\n+public class LineageServiceTest {\n+  @ClassRule\n+  public static DatasetFrameworkTestUtil dsFrameworkUtil = new DatasetFrameworkTestUtil();\n+\n+  @Test\n+  public void testSimpleLineage() throws Exception {\n+    LineageDataset lineageDataset = getLineageDataset(\"testSimpleLineage\");\n+    LineageService lineageService = new LineageService(lineageDataset);\n+\n+    // Define a run\n+    Id.DatasetInstance datasetInstance1 = Id.DatasetInstance.from(\"default\", \"dataset1\");\n+    Id.DatasetInstance datasetInstance2 = Id.DatasetInstance.from(\"default\", \"dataset2\");\n+    Id.Program program1 = Id.Program.from(\"default\", \"app1\", ProgramType.FLOW, \"flow1\");\n+    Id.Flow.Flowlet flowlet1 = Id.Flow.Flowlet.from(program1.getApplication(), program1.getId(), \"flowlet1\");\n+    Id.Run run1 = new Id.Run(program1, RunIds.generate(10000).getId());\n+\n+    // Define another run\n+    Id.DatasetInstance datasetInstance3 = Id.DatasetInstance.from(\"default\", \"dataset3\");\n+    Id.Program program2 = Id.Program.from(\"default\", \"app2\", ProgramType.FLOW, \"flow2\");\n+    Id.Flow.Flowlet flowlet2 = Id.Flow.Flowlet.from(program2.getApplication(), program2.getId(), \"flowlet2\");\n+    Id.Run run2 = new Id.Run(program2, RunIds.generate(900).getId());\n+\n+    // Add accesses for D3 -> P2 -> D2 -> P1 -> D1\n+    lineageDataset.addAccess(run1, datasetInstance1, AccessType.WRITE, \"metadata11\", flowlet1);\n+    lineageDataset.addAccess(run1, datasetInstance2, AccessType.READ, \"metadata12\", flowlet1);\n+\n+    lineageDataset.addAccess(run2, datasetInstance2, AccessType.WRITE, \"metadata21\", flowlet2);\n+    lineageDataset.addAccess(run2, datasetInstance3, AccessType.READ, \"metadata22\", flowlet2);\n+\n+    // Lineage for D1 should be D3 -> P2 -> D2 -> P1 -> D1\n+    Lineage lineage = lineageService.computeLineage(Id.DatasetInstance.from(\"default\", \"dataset1\"), 500, 20000, 5);\n+    Assert.assertNotNull(lineage);\n+\n+    Assert.assertEquals(ImmutableSet.of(program1, program2), lineage.getPrograms());\n+    Assert.assertEquals(ImmutableSet.of(datasetInstance1, datasetInstance2, datasetInstance3), lineage.getData());\n+\n+    Assert.assertEquals(\n+      ImmutableSet.of(\n+        new Relation(datasetInstance1, program1, AccessType.WRITE, iSet(twillRunId(run1)), iSet(flowlet1)),\n+        new Relation(datasetInstance2, program1, AccessType.READ, iSet(twillRunId(run1)), iSet(flowlet1)),\n+        new Relation(datasetInstance2, program2, AccessType.WRITE, iSet(twillRunId(run2)), iSet(flowlet2)),\n+        new Relation(datasetInstance3, program2, AccessType.READ, iSet(twillRunId(run2)), iSet(flowlet2))\n+      ),\n+      lineage.getRelations());\n+\n+    // Lineage for D2 should be D3 -> P2 -> D2\n+    Lineage lineage2 = lineageService.computeLineage(Id.DatasetInstance.from(\"default\", \"dataset2\"), 500, 20000, 5);\n+    Assert.assertNotNull(lineage2);\n+\n+    Assert.assertEquals(ImmutableSet.of(program2), lineage2.getPrograms());\n+    Assert.assertEquals(ImmutableSet.of(datasetInstance2, datasetInstance3), lineage2.getData());\n+\n+    Assert.assertEquals(\n+      ImmutableSet.of(\n+        new Relation(datasetInstance2, program2, AccessType.WRITE, iSet(twillRunId(run2)), iSet(flowlet2)),\n+        new Relation(datasetInstance3, program2, AccessType.READ, iSet(twillRunId(run2)), iSet(flowlet2))\n+      ),\n+      lineage2.getRelations());\n+\n+    // Lineage for D1 for one level should be D2 -> P1 -> D1\n+    Lineage lineage3 = lineageService.computeLineage(Id.DatasetInstance.from(\"default\", \"dataset1\"), 500, 20000, 1);\n+    Assert.assertNotNull(lineage3);\n+\n+    Assert.assertEquals(ImmutableSet.of(program1), lineage3.getPrograms());\n+    Assert.assertEquals(ImmutableSet.of(datasetInstance1, datasetInstance2), lineage3.getData());\n+\n+    Assert.assertEquals(\n+      ImmutableSet.of(\n+        new Relation(datasetInstance1, program1, AccessType.WRITE, iSet(twillRunId(run1)), iSet(flowlet1)),\n+        new Relation(datasetInstance2, program1, AccessType.READ, iSet(twillRunId(run1)), iSet(flowlet1))\n+      ),\n+      lineage3.getRelations());\n+  }\n+\n+  @SafeVarargs\n+  private static <T> Set<T> iSet(T... elements) {","path":"cdap-data-fabric/src/test/java/co/cask/cdap/data2/metadata/lineage/LineageServiceTest.java","position":111,"original_position":111,"commit_id":"f73d979939d1e2cbc75c8b2e6bcbf8184833e421","original_commit_id":"f73d979939d1e2cbc75c8b2e6bcbf8184833e421","user":{"login":"chtyim","id":885032,"avatar_url":"https://avatars.githubusercontent.com/u/885032?v=3","gravatar_id":"","url":"https://api.github.com/users/chtyim","html_url":"https://github.com/chtyim","followers_url":"https://api.github.com/users/chtyim/followers","following_url":"https://api.github.com/users/chtyim/following{/other_user}","gists_url":"https://api.github.com/users/chtyim/gists{/gist_id}","starred_url":"https://api.github.com/users/chtyim/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/chtyim/subscriptions","organizations_url":"https://api.github.com/users/chtyim/orgs","repos_url":"https://api.github.com/users/chtyim/repos","events_url":"https://api.github.com/users/chtyim/events{/privacy}","received_events_url":"https://api.github.com/users/chtyim/received_events","type":"User","site_admin":false},"body":"iSet? Should it be asSet or toSet?","created_at":"2015-09-09T08:27:52Z","updated_at":"2015-09-09T08:27:52Z","html_url":"https://github.com/caskdata/cdap/pull/3855#discussion_r39017849","pull_request_url":"https://api.github.com/repos/caskdata/cdap/pulls/3855","_links":{"self":{"href":"https://api.github.com/repos/caskdata/cdap/pulls/comments/39017849"},"html":{"href":"https://github.com/caskdata/cdap/pull/3855#discussion_r39017849"},"pull_request":{"href":"https://api.github.com/repos/caskdata/cdap/pulls/3855"}}},"pull_request":{"url":"https://api.github.com/repos/caskdata/cdap/pulls/3855","id":44519961,"html_url":"https://github.com/caskdata/cdap/pull/3855","diff_url":"https://github.com/caskdata/cdap/pull/3855.diff","patch_url":"https://github.com/caskdata/cdap/pull/3855.patch","issue_url":"https://api.github.com/repos/caskdata/cdap/issues/3855","number":3855,"state":"open","locked":false,"title":"CDAP-3519 First cut of Lineage computation based on Dataset accesses by a Program","user":{"login":"poornachandra","id":3181212,"avatar_url":"https://avatars.githubusercontent.com/u/3181212?v=3","gravatar_id":"","url":"https://api.github.com/users/poornachandra","html_url":"https://github.com/poornachandra","followers_url":"https://api.github.com/users/poornachandra/followers","following_url":"https://api.github.com/users/poornachandra/following{/other_user}","gists_url":"https://api.github.com/users/poornachandra/gists{/gist_id}","starred_url":"https://api.github.com/users/poornachandra/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/poornachandra/subscriptions","organizations_url":"https://api.github.com/users/poornachandra/orgs","repos_url":"https://api.github.com/users/poornachandra/repos","events_url":"https://api.github.com/users/poornachandra/events{/privacy}","received_events_url":"https://api.github.com/users/poornachandra/received_events","type":"User","site_admin":false},"body":"JIRA - https://issues.cask.co/browse/CDAP-3519\r\n\r\nTODO: Will add more extensive tests in a follow-up PR.\r\n","created_at":"2015-09-09T07:54:28Z","updated_at":"2015-09-09T08:27:52Z","closed_at":null,"merged_at":null,"merge_commit_sha":"70ae7b31d2be6ea58fbedca568df34595e6ab263","assignee":null,"milestone":null,"commits_url":"https://api.github.com/repos/caskdata/cdap/pulls/3855/commits","review_comments_url":"https://api.github.com/repos/caskdata/cdap/pulls/3855/comments","review_comment_url":"https://api.github.com/repos/caskdata/cdap/pulls/comments{/number}","comments_url":"https://api.github.com/repos/caskdata/cdap/issues/3855/comments","statuses_url":"https://api.github.com/repos/caskdata/cdap/statuses/f73d979939d1e2cbc75c8b2e6bcbf8184833e421","head":{"label":"caskdata:feature/lineage-dataset","ref":"feature/lineage-dataset","sha":"f73d979939d1e2cbc75c8b2e6bcbf8184833e421","user":{"login":"caskdata","id":8123023,"avatar_url":"https://avatars.githubusercontent.com/u/8123023?v=3","gravatar_id":"","url":"https://api.github.com/users/caskdata","html_url":"https://github.com/caskdata","followers_url":"https://api.github.com/users/caskdata/followers","following_url":"https://api.github.com/users/caskdata/following{/other_user}","gists_url":"https://api.github.com/users/caskdata/gists{/gist_id}","starred_url":"https://api.github.com/users/caskdata/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/caskdata/subscriptions","organizations_url":"https://api.github.com/users/caskdata/orgs","repos_url":"https://api.github.com/users/caskdata/repos","events_url":"https://api.github.com/users/caskdata/events{/privacy}","received_events_url":"https://api.github.com/users/caskdata/received_events","type":"Organization","site_admin":false},"repo":{"id":22542759,"name":"cdap","full_name":"caskdata/cdap","owner":{"login":"caskdata","id":8123023,"avatar_url":"https://avatars.githubusercontent.com/u/8123023?v=3","gravatar_id":"","url":"https://api.github.com/users/caskdata","html_url":"https://github.com/caskdata","followers_url":"https://api.github.com/users/caskdata/followers","following_url":"https://api.github.com/users/caskdata/following{/other_user}","gists_url":"https://api.github.com/users/caskdata/gists{/gist_id}","starred_url":"https://api.github.com/users/caskdata/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/caskdata/subscriptions","organizations_url":"https://api.github.com/users/caskdata/orgs","repos_url":"https://api.github.com/users/caskdata/repos","events_url":"https://api.github.com/users/caskdata/events{/privacy}","received_events_url":"https://api.github.com/users/caskdata/received_events","type":"Organization","site_admin":false},"private":false,"html_url":"https://github.com/caskdata/cdap","description":"Cask Data Application Platform (CDAP)","fork":false,"url":"https://api.github.com/repos/caskdata/cdap","forks_url":"https://api.github.com/repos/caskdata/cdap/forks","keys_url":"https://api.github.com/repos/caskdata/cdap/keys{/key_id}","collaborators_url":"https://api.github.com/repos/caskdata/cdap/collaborators{/collaborator}","teams_url":"https://api.github.com/repos/caskdata/cdap/teams","hooks_url":"https://api.github.com/repos/caskdata/cdap/hooks","issue_events_url":"https://api.github.com/repos/caskdata/cdap/issues/events{/number}","events_url":"https://api.github.com/repos/caskdata/cdap/events","assignees_url":"https://api.github.com/repos/caskdata/cdap/assignees{/user}","branches_url":"https://api.github.com/repos/caskdata/cdap/branches{/branch}","tags_url":"https://api.github.com/repos/caskdata/cdap/tags","blobs_url":"https://api.github.com/repos/caskdata/cdap/git/blobs{/sha}","git_tags_url":"https://api.github.com/repos/caskdata/cdap/git/tags{/sha}","git_refs_url":"https://api.github.com/repos/caskdata/cdap/git/refs{/sha}","trees_url":"https://api.github.com/repos/caskdata/cdap/git/trees{/sha}","statuses_url":"https://api.github.com/repos/caskdata/cdap/statuses/{sha}","languages_url":"https://api.github.com/repos/caskdata/cdap/languages","stargazers_url":"https://api.github.com/repos/caskdata/cdap/stargazers","contributors_url":"https://api.github.com/repos/caskdata/cdap/contributors","subscribers_url":"https://api.github.com/repos/caskdata/cdap/subscribers","subscription_url":"https://api.github.com/repos/caskdata/cdap/subscription","commits_url":"https://api.github.com/repos/caskdata/cdap/commits{/sha}","git_commits_url":"https://api.github.com/repos/caskdata/cdap/git/commits{/sha}","comments_url":"https://api.github.com/repos/caskdata/cdap/comments{/number}","issue_comment_url":"https://api.github.com/repos/caskdata/cdap/issues/comments{/number}","contents_url":"https://api.github.com/repos/caskdata/cdap/contents/{+path}","compare_url":"https://api.github.com/repos/caskdata/cdap/compare/{base}...{head}","merges_url":"https://api.github.com/repos/caskdata/cdap/merges","archive_url":"https://api.github.com/repos/caskdata/cdap/{archive_format}{/ref}","downloads_url":"https://api.github.com/repos/caskdata/cdap/downloads","issues_url":"https://api.github.com/repos/caskdata/cdap/issues{/number}","pulls_url":"https://api.github.com/repos/caskdata/cdap/pulls{/number}","milestones_url":"https://api.github.com/repos/caskdata/cdap/milestones{/number}","notifications_url":"https://api.github.com/repos/caskdata/cdap/notifications{?since,all,participating}","labels_url":"https://api.github.com/repos/caskdata/cdap/labels{/name}","releases_url":"https://api.github.com/repos/caskdata/cdap/releases{/id}","created_at":"2014-08-02T09:00:10Z","updated_at":"2015-09-04T03:31:50Z","pushed_at":"2015-09-09T07:54:28Z","git_url":"git://github.com/caskdata/cdap.git","ssh_url":"git@github.com:caskdata/cdap.git","clone_url":"https://github.com/caskdata/cdap.git","svn_url":"https://github.com/caskdata/cdap","homepage":"http://cdap.io","size":857269,"stargazers_count":64,"watchers_count":64,"language":"Java","has_issues":false,"has_downloads":true,"has_wiki":false,"has_pages":true,"forks_count":19,"mirror_url":null,"open_issues_count":14,"forks":19,"open_issues":14,"watchers":64,"default_branch":"develop"}},"base":{"label":"caskdata:develop","ref":"develop","sha":"21d59a3f85b7ef3a129ec42c195f8b3e89c2d573","user":{"login":"caskdata","id":8123023,"avatar_url":"https://avatars.githubusercontent.com/u/8123023?v=3","gravatar_id":"","url":"https://api.github.com/users/caskdata","html_url":"https://github.com/caskdata","followers_url":"https://api.github.com/users/caskdata/followers","following_url":"https://api.github.com/users/caskdata/following{/other_user}","gists_url":"https://api.github.com/users/caskdata/gists{/gist_id}","starred_url":"https://api.github.com/users/caskdata/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/caskdata/subscriptions","organizations_url":"https://api.github.com/users/caskdata/orgs","repos_url":"https://api.github.com/users/caskdata/repos","events_url":"https://api.github.com/users/caskdata/events{/privacy}","received_events_url":"https://api.github.com/users/caskdata/received_events","type":"Organization","site_admin":false},"repo":{"id":22542759,"name":"cdap","full_name":"caskdata/cdap","owner":{"login":"caskdata","id":8123023,"avatar_url":"https://avatars.githubusercontent.com/u/8123023?v=3","gravatar_id":"","url":"https://api.github.com/users/caskdata","html_url":"https://github.com/caskdata","followers_url":"https://api.github.com/users/caskdata/followers","following_url":"https://api.github.com/users/caskdata/following{/other_user}","gists_url":"https://api.github.com/users/caskdata/gists{/gist_id}","starred_url":"https://api.github.com/users/caskdata/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/caskdata/subscriptions","organizations_url":"https://api.github.com/users/caskdata/orgs","repos_url":"https://api.github.com/users/caskdata/repos","events_url":"https://api.github.com/users/caskdata/events{/privacy}","received_events_url":"https://api.github.com/users/caskdata/received_events","type":"Organization","site_admin":false},"private":false,"html_url":"https://github.com/caskdata/cdap","description":"Cask Data Application Platform (CDAP)","fork":false,"url":"https://api.github.com/repos/caskdata/cdap","forks_url":"https://api.github.com/repos/caskdata/cdap/forks","keys_url":"https://api.github.com/repos/caskdata/cdap/keys{/key_id}","collaborators_url":"https://api.github.com/repos/caskdata/cdap/collaborators{/collaborator}","teams_url":"https://api.github.com/repos/caskdata/cdap/teams","hooks_url":"https://api.github.com/repos/caskdata/cdap/hooks","issue_events_url":"https://api.github.com/repos/caskdata/cdap/issues/events{/number}","events_url":"https://api.github.com/repos/caskdata/cdap/events","assignees_url":"https://api.github.com/repos/caskdata/cdap/assignees{/user}","branches_url":"https://api.github.com/repos/caskdata/cdap/branches{/branch}","tags_url":"https://api.github.com/repos/caskdata/cdap/tags","blobs_url":"https://api.github.com/repos/caskdata/cdap/git/blobs{/sha}","git_tags_url":"https://api.github.com/repos/caskdata/cdap/git/tags{/sha}","git_refs_url":"https://api.github.com/repos/caskdata/cdap/git/refs{/sha}","trees_url":"https://api.github.com/repos/caskdata/cdap/git/trees{/sha}","statuses_url":"https://api.github.com/repos/caskdata/cdap/statuses/{sha}","languages_url":"https://api.github.com/repos/caskdata/cdap/languages","stargazers_url":"https://api.github.com/repos/caskdata/cdap/stargazers","contributors_url":"https://api.github.com/repos/caskdata/cdap/contributors","subscribers_url":"https://api.github.com/repos/caskdata/cdap/subscribers","subscription_url":"https://api.github.com/repos/caskdata/cdap/subscription","commits_url":"https://api.github.com/repos/caskdata/cdap/commits{/sha}","git_commits_url":"https://api.github.com/repos/caskdata/cdap/git/commits{/sha}","comments_url":"https://api.github.com/repos/caskdata/cdap/comments{/number}","issue_comment_url":"https://api.github.com/repos/caskdata/cdap/issues/comments{/number}","contents_url":"https://api.github.com/repos/caskdata/cdap/contents/{+path}","compare_url":"https://api.github.com/repos/caskdata/cdap/compare/{base}...{head}","merges_url":"https://api.github.com/repos/caskdata/cdap/merges","archive_url":"https://api.github.com/repos/caskdata/cdap/{archive_format}{/ref}","downloads_url":"https://api.github.com/repos/caskdata/cdap/downloads","issues_url":"https://api.github.com/repos/caskdata/cdap/issues{/number}","pulls_url":"https://api.github.com/repos/caskdata/cdap/pulls{/number}","milestones_url":"https://api.github.com/repos/caskdata/cdap/milestones{/number}","notifications_url":"https://api.github.com/repos/caskdata/cdap/notifications{?since,all,participating}","labels_url":"https://api.github.com/repos/caskdata/cdap/labels{/name}","releases_url":"https://api.github.com/repos/caskdata/cdap/releases{/id}","created_at":"2014-08-02T09:00:10Z","updated_at":"2015-09-04T03:31:50Z","pushed_at":"2015-09-09T07:54:28Z","git_url":"git://github.com/caskdata/cdap.git","ssh_url":"git@github.com:caskdata/cdap.git","clone_url":"https://github.com/caskdata/cdap.git","svn_url":"https://github.com/caskdata/cdap","homepage":"http://cdap.io","size":857269,"stargazers_count":64,"watchers_count":64,"language":"Java","has_issues":false,"has_downloads":true,"has_wiki":false,"has_pages":true,"forks_count":19,"mirror_url":null,"open_issues_count":14,"forks":19,"open_issues":14,"watchers":64,"default_branch":"develop"}},"_links":{"self":{"href":"https://api.github.com/repos/caskdata/cdap/pulls/3855"},"html":{"href":"https://github.com/caskdata/cdap/pull/3855"},"issue":{"href":"https://api.github.com/repos/caskdata/cdap/issues/3855"},"comments":{"href":"https://api.github.com/repos/caskdata/cdap/issues/3855/comments"},"review_comments":{"href":"https://api.github.com/repos/caskdata/cdap/pulls/3855/comments"},"review_comment":{"href":"https://api.github.com/repos/caskdata/cdap/pulls/comments{/number}"},"commits":{"href":"https://api.github.com/repos/caskdata/cdap/pulls/3855/commits"},"statuses":{"href":"https://api.github.com/repos/caskdata/cdap/statuses/f73d979939d1e2cbc75c8b2e6bcbf8184833e421"}}}}