@@ -31,8 +31,8 @@ def test_create_actor(self):
3131 """
3232 result = self .runner .invoke (process_tracker .cli .main , 'create -t actor -n "Test Test"' )
3333
34- instance = self .session .query (Actor ).filter (Actor .actor_name == 'Test Test' )
35- given_name = instance [ 0 ] .actor_name
34+ instance = self .session .query (Actor ).filter (Actor .actor_name == 'Test Test' ). first ()
35+ given_name = instance .actor_name
3636
3737 self .runner .invoke (process_tracker .cli .main , 'delete -t actor -n "Test Test"' )
3838
@@ -46,8 +46,8 @@ def test_create_extract_status(self):
4646 """
4747 result = self .runner .invoke (process_tracker .cli .main , 'create -t "extract status" -n "New Status"' )
4848
49- instance = self .session .query (ExtractStatus ).filter (ExtractStatus .extract_status_name == 'New Status' )
50- given_name = instance [ 0 ] .extract_status_name
49+ instance = self .session .query (ExtractStatus ).filter (ExtractStatus .extract_status_name == 'New Status' ). first ()
50+ given_name = instance .extract_status_name
5151 self .runner .invoke (process_tracker .cli .main , 'delete -t "extract status" -n "New Status"' )
5252
5353 self .assertEqual ('New Status' , given_name )
@@ -61,8 +61,8 @@ def test_create_error_type(self):
6161
6262 result = self .runner .invoke (process_tracker .cli .main , 'create -t "error type" -n "New Error Type"' )
6363
64- instance = self .session .query (ErrorType ).filter (ErrorType .error_type_name == 'New Error Type' )
65- given_name = instance [ 0 ] .error_type_name
64+ instance = self .session .query (ErrorType ).filter (ErrorType .error_type_name == 'New Error Type' ). first ()
65+ given_name = instance .error_type_name
6666
6767 self .runner .invoke (process_tracker .cli .main , 'delete -t "error type" -n "New Error Type"' )
6868
@@ -76,8 +76,8 @@ def test_create_process_type(self):
7676 """
7777 result = self .runner .invoke (process_tracker .cli .main , 'create -t "process type" -n "New Process Type"' )
7878
79- instance = self .session .query (ProcessType ).filter (ProcessType .process_type_name == 'New Process Type' )
80- given_name = instance [ 0 ] .process_type_name
79+ instance = self .session .query (ProcessType ).filter (ProcessType .process_type_name == 'New Process Type' ). first ()
80+ given_name = instance .process_type_name
8181
8282 self .runner .invoke (process_tracker .cli .main , 'delete -t "process type" -n "New Process Type"' )
8383
@@ -91,8 +91,8 @@ def test_create_process_status(self):
9191 """
9292 result = self .runner .invoke (process_tracker .cli .main , 'create -t "process status" -n "New Status Type"' )
9393
94- instance = self .session .query (ProcessStatus ).filter (ProcessStatus .process_status_name == 'New Status Type' )
95- given_name = instance [ 0 ] .process_status_name
94+ instance = self .session .query (ProcessStatus ).filter (ProcessStatus .process_status_name == 'New Status Type' ). first ()
95+ given_name = instance .process_status_name
9696
9797 self .runner .invoke (process_tracker .cli .main , 'delete -t "process status" -n "New Status Type"' )
9898
@@ -106,8 +106,8 @@ def test_create_source(self):
106106 """
107107 result = self .runner .invoke (process_tracker .cli .main , 'create -t source -n "New Source"' )
108108
109- instance = self .session .query (Source ).filter (Source .source_name == 'New Source' )
110- given_name = instance [ 0 ] .source_name
109+ instance = self .session .query (Source ).filter (Source .source_name == 'New Source' ). first ()
110+ given_name = instance .source_name
111111
112112 self .runner .invoke (process_tracker .cli .main , 'delete -t "source" -n "New Source"' )
113113
@@ -121,8 +121,8 @@ def test_create_tool(self):
121121 """
122122 result = self .runner .invoke (process_tracker .cli .main , 'create -t tool -n "New Tool"' )
123123
124- instance = self .session .query (Tool ).filter (Tool .tool_name == 'New Tool' )
125- given_name = instance [ 0 ] .tool_name
124+ instance = self .session .query (Tool ).filter (Tool .tool_name == 'New Tool' ). first ()
125+ given_name = instance .tool_name
126126
127127 self .runner .invoke (process_tracker .cli .main , 'delete -t "tool" -n "New Tool"' )
128128
0 commit comments