#1437 Fix edge name for entity origin

TheHive-Project · Jul 16, 2020 · fec2a32 · fec2a32
1 parent 1909572
commit fec2a32
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/thehive/app/org/thp/thehive/services/ObservableSrv.scala b/thehive/app/org/thp/thehive/services/ObservableSrv.scala
@@ -232,7 +232,7 @@ class ObservableSteps(raw: GremlinScala[Vertex])(implicit @Named("with-thehive-s
 
   def organisations = new OrganisationSteps(raw.inTo[ShareObservable].inTo[OrganisationShare])
 
-  def origin: OrganisationSteps = new OrganisationSteps(raw.inTo[ShareCase].has(Key("owner") of true).inTo[OrganisationShare])
+  def origin: OrganisationSteps = new OrganisationSteps(raw.inTo[ShareObservable].has(Key("owner") of true).inTo[OrganisationShare])
 
   override def newInstance(): ObservableSteps = new ObservableSteps(raw.clone())
 

diff --git a/thehive/app/org/thp/thehive/services/TaskSrv.scala b/thehive/app/org/thp/thehive/services/TaskSrv.scala
@@ -159,7 +159,7 @@ class TaskSteps(raw: GremlinScala[Vertex])(implicit db: Database, graph: Graph)
   def organisations(permission: Permission) =
     new OrganisationSteps(raw.inTo[ShareTask].filter(_.outTo[ShareProfile].has(Key("permissions") of permission)).inTo[OrganisationShare])
 
-  def origin: OrganisationSteps = new OrganisationSteps(raw.inTo[ShareCase].has(Key("owner") of true).inTo[OrganisationShare])
+  def origin: OrganisationSteps = new OrganisationSteps(raw.inTo[ShareTask].has(Key("owner") of true).inTo[OrganisationShare])
 
   def assignableUsers(implicit authContext: AuthContext): UserSteps =
     organisations(Permissions.manageTask)