updated unit tests

crawlab-team · Jun 5, 2022 · 94f5a02 · 94f5a02
1 parent 2718d6a
commit 94f5a02
Show file tree

Hide file tree

Showing 15 changed files with 11 additions and 31 deletions.
diff --git a/_ext/python/legacy/cli/__init__.py → _ext/python/_legacy/cli/__init__.py b/_ext/python/legacy/cli/__init__.py → _ext/python/_legacy/cli/__init__.py
diff --git a/_ext/python/legacy/core/__init__.py → _ext/python/_legacy/core/__init__.py b/_ext/python/legacy/core/__init__.py → _ext/python/_legacy/core/__init__.py
diff --git a/_ext/python/legacy/core/client.py → _ext/python/_legacy/core/client.py b/_ext/python/legacy/core/client.py → _ext/python/_legacy/core/client.py
diff --git a/_ext/python/legacy/core/config.py → _ext/python/_legacy/core/config.py b/_ext/python/legacy/core/config.py → _ext/python/_legacy/core/config.py
diff --git a/_ext/python/legacy/core/log.py → _ext/python/_legacy/core/log.py b/_ext/python/legacy/core/log.py → _ext/python/_legacy/core/log.py
diff --git a/_ext/python/legacy/core/request.py → _ext/python/_legacy/core/request.py b/_ext/python/legacy/core/request.py → _ext/python/_legacy/core/request.py
diff --git a/_ext/python/crawlab/models/__init__.py → _ext/python/_legacy/db/__init__.py b/_ext/python/crawlab/models/__init__.py → _ext/python/_legacy/db/__init__.py
diff --git a/_ext/python/legacy/db/es.py → _ext/python/_legacy/db/es.py b/_ext/python/legacy/db/es.py → _ext/python/_legacy/db/es.py
diff --git a/_ext/python/legacy/db/kafka.py → _ext/python/_legacy/db/kafka.py b/_ext/python/legacy/db/kafka.py → _ext/python/_legacy/db/kafka.py
diff --git a/_ext/python/legacy/db/mongo.py → _ext/python/_legacy/db/mongo.py b/_ext/python/legacy/db/mongo.py → _ext/python/_legacy/db/mongo.py
diff --git a/_ext/python/legacy/db/sql.py → _ext/python/_legacy/db/sql.py b/_ext/python/legacy/db/sql.py → _ext/python/_legacy/db/sql.py
diff --git a/_ext/python/legacy/pipelines/__init__.py → _ext/python/_legacy/pipelines/__init__.py b/_ext/python/legacy/pipelines/__init__.py → _ext/python/_legacy/pipelines/__init__.py
diff --git a/_ext/python/crawlab/models/spider.py b/_ext/python/crawlab/models/spider.py
diff --git a/_ext/python/legacy/db/__init__.py b/_ext/python/legacy/db/__init__.py
diff --git a/_ext/python/test/cli_action/upload_test.py b/_ext/python/test/cli_action/upload_test.py
@@ -66,10 +66,16 @@ def test_upload(self):
 
     def test_upload_with_crawlab_json(self):
         name, dir_path = self._setup()
+        description = 'test_description_' + f'_{int(datetime.now().timestamp())}'
+        mode = 'all-nodes'
+        priority = 1
+        cmd = 'echo hello'
+        param = 'test'
         with open(os.path.join(dir_path, 'main.py'), 'w') as f:
             f.write('print(\'hello world\')')
         with open(os.path.join(dir_path, 'crawlab.json'), 'w') as f:
-            f.write('{"name":"%s"}' % name)
+            f.write('{"name":"%s","description":"%s","mode":"%s","priority":%d,"cmd":"%s","param":"%s"}' % (
+                name, description, mode, priority, cmd, param))
         login(Namespace(
             username='admin',
             password='admin',
@@ -94,6 +100,10 @@ def test_upload_with_crawlab_json(self):
         assert len(data) == 10
         spider = data[0]
         assert spider.get('name') == name
+        assert spider.get('description') == description
+        assert spider.get('mode') == mode
+        assert spider.get('cmd') == cmd
+        assert spider.get('param') == param
         requests.delete(f'{self.endpoint}/spiders/{spider.get("_id")}',
                         headers={'Authorization': config.data.get("token")})