Skip to content

Commit 26ec39d

Browse files
committed
Adding CLI Capabilities
🐛 Trying to fix failing tests on TravisCI. Working on #3
1 parent 5332897 commit 26ec39d

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

tests/test_cli.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)