improve tests

Author: alexphelps

ntk/command.py

@@ -20,6 +20,7 @@
     level=logging.INFO,
     datefmt='%Y-%m-%d %H:%M:%S'
 )
+logging.getLogger('watchfiles').setLevel(logging.WARNING)
 
 
 class Command:
@@ -45,10 +46,10 @@ def _handle_files_change(self, changes):
         for event_type, pathfile in changes:
             template_name = get_template_name(pathfile)
             if event_type in [Change.added, Change.modified]:
-                logging.info(f'[{self.config.env}] {str(event_type)} {template_name}')
+                logging.info(f'[{self.config.env}] {event_type.name.title()} {template_name}')
                 self._push_templates([template_name], compile_sass=True)
             elif event_type == Change.deleted:
-                logging.info(f'[{self.config.env}] {str(event_type)} {template_name}')
+                logging.info(f'[{self.config.env}] {event_type.name.title()} {template_name}')
                 self._delete_templates([template_name])
 
     def _push_templates(self, template_names, compile_sass=False):

tests/test_command.py

@@ -310,6 +310,62 @@ def test_pull_command_with_configs_and_filenames_should_be_download_only_file_in
 
         mock_write_config.assert_not_called()
 
+    #####
+    # push
+    #####
+    def test_push_command_without_config_file_should_be_required_api_key_store_and_theme_id(self):
+        with self.assertRaises(TypeError) as error:
+            self.parser.apikey = None
+            self.parser.store = None
+            self.parser.theme_id = None
+            self.command.push(self.parser)
+        self.assertEqual(
+            str(error.exception), '[development] argument -a/--apikey, -s/--store, -t/--theme_id are required.')
+
+    @patch("ntk.command.Command._get_accept_files", autospec=True)
+    def test_push_command_with_configs_and_without_filenames_should_upload_all_files(
+        self, mock_get_accept_files
+    ):
+        mock_get_accept_files.return_value = [
+            f'{os.getcwd()}/layout/base.html',
+        ]
+        self.mock_gateway.return_value.create_or_update_template.return_value.ok = True
+        self.mock_gateway.return_value.create_or_update_template.return_value.headers = {
+            'content-type': 'application/json; charset=utf-8'}
+        self.command.config.parser_config(self.parser)
+        self.parser.filenames = None
+        with patch("builtins.open", self.mock_file):
+            self.command.push(self.parser)
+        expected_call = call().create_or_update_template(
+            theme_id=1234,
+            template_name='layout/base.html',
+            content='{% load i18n %}\n\n<div class="mt-2">My home page</div>',
+            files={}
+        )
+        self.assertIn(expected_call, self.mock_gateway.mock_calls)
+
+    @patch("ntk.command.Command._get_accept_files", autospec=True)
+    def test_push_command_with_filenames_should_upload_only_specified_files(
+        self, mock_get_accept_files
+    ):
+        mock_get_accept_files.return_value = [
+            f'{os.getcwd()}/layout/base.html',
+        ]
+        self.mock_gateway.return_value.create_or_update_template.return_value.ok = True
+        self.mock_gateway.return_value.create_or_update_template.return_value.headers = {
+            'content-type': 'application/json; charset=utf-8'}
+        self.command.config.parser_config(self.parser)
+        self.parser.filenames = ['layout/base.html']
+        with patch("builtins.open", self.mock_file):
+            self.command.push(self.parser)
+        expected_call = call().create_or_update_template(
+            theme_id=1234,
+            template_name='layout/base.html',
+            content='{% load i18n %}\n\n<div class="mt-2">My home page</div>',
+            files={}
+        )
+        self.assertIn(expected_call, self.mock_gateway.mock_calls)
+
     #####
     # watch (_handle_files_change)
     #####