@@ -23,16 +23,14 @@ defmodule Livebook.TeamsIntegrationHelper do
2323 Map . merge ( context , % { conn: conn , code: code } )
2424 end
2525
26- @ doc false
27- def create_user_hub ( node ) do
26+ defp create_user_hub ( node ) do
2827 context = new_user_hub ( node )
2928 Hubs . save_hub ( context . team )
3029
3130 wait_until_client_start ( context )
3231 end
3332
34- @ doc false
35- def new_user_hub ( node ) do
33+ defp new_user_hub ( node ) do
3634 { teams_key , key_hash } = generate_key_hash ( )
3735
3836 org = TeamsRPC . create_org ( node )
@@ -66,16 +64,14 @@ defmodule Livebook.TeamsIntegrationHelper do
6664 }
6765 end
6866
69- @ doc false
70- def create_agent_hub ( node , opts \\ [ ] ) do
67+ defp create_agent_hub ( node , opts \\ [ ] ) do
7168 context = new_agent_hub ( node , opts )
7269 Hubs . save_hub ( context . team )
7370
7471 wait_until_agent_start ( context )
7572 end
7673
77- @ doc false
78- def new_agent_hub ( node , opts \\ [ ] ) do
74+ defp new_agent_hub ( node , opts \\ [ ] ) do
7975 { teams_key , key_hash } = generate_key_hash ( )
8076
8177 org = TeamsRPC . create_org ( node )
@@ -119,8 +115,7 @@ defmodule Livebook.TeamsIntegrationHelper do
119115 }
120116 end
121117
122- @ doc false
123- def authenticate_user_on_teams ( name , conn , node , team ) do
118+ defp authenticate_user_on_teams ( name , conn , node , team ) do
124119 # Create a fresh connection to avoid session contamination
125120 fresh_conn = Phoenix.ConnTest . build_conn ( )
126121
@@ -151,7 +146,6 @@ defmodule Livebook.TeamsIntegrationHelper do
151146 code }
152147 end
153148
154- @ doc false
155149 def change_to_agent_session ( % { node: node , teams_for: :user } = context ) do
156150 pid = Hubs.TeamClient . get_pid ( context . team . id )
157151 Hubs.TeamClient . stop ( context . team . id )
0 commit comments