|
3 | 3 | import tableauserverclient as TSC |
4 | 4 | from ..assets import mock_data |
5 | 5 | from unittest.mock import * |
| 6 | +from unittest import mock |
6 | 7 | from tabcmd.commands.datasources_and_workbooks.publish_command import PublishCommand |
| 8 | +from tabcmd.commands.datasources_and_workbooks import publish_command |
7 | 9 |
|
8 | 10 | from ..assets.mock_data import set_up_mock_args, set_up_mock_file, set_up_mock_path, set_up_mock_server |
9 | 11 |
|
10 | 12 | mock_args = set_up_mock_args() |
| 13 | +mock_logger = mock.MagicMock() |
11 | 14 |
|
12 | 15 |
|
13 | 16 | # mock the module as it is imported *when used* |
@@ -83,39 +86,39 @@ def test_get_files_to_publish_folder(self, mock_path, mock_glob, mock_session): |
83 | 86 | actual = PublishCommand.get_files_to_publish(mock_args, logging) |
84 | 87 | assert actual == expected |
85 | 88 |
|
86 | | - def test_default_publish_mode(self, mock_session, mock_server): |
| 89 | + def test_default_publish_mode(self, mock_path, mock_glob, mock_session): |
87 | 90 | mock_args.replace = False |
88 | 91 | mock_args.append = False |
89 | 92 | mock_args.overwrite = False |
90 | 93 |
|
91 | 94 | publish_mode = publish_command.PublishCommand.get_publish_mode(mock_args, mock_logger) |
92 | 95 | self.assertEqual(publish_mode, TSC.Server.PublishMode.CreateNew) |
93 | 96 |
|
94 | | - def test_replace_publish_mode(self, mock_session, mock_server): |
| 97 | + def test_replace_publish_mode(self, mock_path, mock_glob, mock_session): |
95 | 98 | mock_args.replace = True |
96 | 99 | mock_args.append = False |
97 | 100 | mock_args.overwrite = False |
98 | 101 |
|
99 | 102 | publish_mode = publish_command.PublishCommand.get_publish_mode(mock_args, mock_logger) |
100 | 103 | self.assertEqual(publish_mode, TSC.Server.PublishMode.Replace) |
101 | 104 |
|
102 | | - def test_append_publish_mode(self, mock_session, mock_server): |
| 105 | + def test_append_publish_mode(self, mock_path, mock_glob, mock_session): |
103 | 106 | mock_args.replace = False |
104 | 107 | mock_args.append = True |
105 | 108 | mock_args.overwrite = False |
106 | 109 |
|
107 | 110 | publish_mode = publish_command.PublishCommand.get_publish_mode(mock_args, mock_logger) |
108 | 111 | self.assertEqual(publish_mode, TSC.Server.PublishMode.Append) |
109 | 112 |
|
110 | | - def test_overwrite_publish_mode(self, mock_session, mock_server): |
| 113 | + def test_overwrite_publish_mode(self, mock_path, mock_glob, mock_session): |
111 | 114 | mock_args.replace = False |
112 | 115 | mock_args.append = False |
113 | 116 | mock_args.overwrite = True |
114 | 117 |
|
115 | 118 | publish_mode = publish_command.PublishCommand.get_publish_mode(mock_args, mock_logger) |
116 | 119 | self.assertEqual(publish_mode, TSC.Server.PublishMode.Overwrite) |
117 | 120 |
|
118 | | - def test_invalid_combination_of_modes(self, mock_session, mock_server): |
| 121 | + def test_invalid_combination_of_modes(self, mock_path, mock_glob, mock_session): |
119 | 122 | mock_args.replace = True |
120 | 123 | mock_args.append = True |
121 | 124 | mock_args.overwrite = False |
|
0 commit comments