praetorian-cli 2.1.3__tar.gz → 2.1.5__tar.gz

{praetorian_cli-2.1.3/praetorian_cli.egg-info → praetorian_cli-2.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praetorian-cli
-Version: 2.1.3
+Version: 2.1.5
 Summary: For interacting with the Chariot API
 Home-page: https://github.com/praetorian-inc/praetorian-cli
 Author: Praetorian

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/handlers/add.py

@@ -6,7 +6,7 @@ import click
 from praetorian_cli.handlers.chariot import chariot
 from praetorian_cli.handlers.cli_decorators import cli_handler, praetorian_only
 from praetorian_cli.handlers.utils import error
-from praetorian_cli.sdk.model.globals import AddRisk, Asset, Seed
+from praetorian_cli.sdk.model.globals import AddRisk, Asset, Seed, Kind
 
 
 @chariot.group()
@@ -18,11 +18,12 @@ def add():
 @add.command()
 @cli_handler
 @click.option('-d', '--dns', required=True, help='The DNS of the asset')
-@click.option('-n', '--name', required=False, help='The name of the asset, e.g, IP address, GitHub repo URL')
+@click.option('-n', '--name', required=False, help='The name of the asset, e.g, IP address')
+@click.option('-t', '--type', 'asset_type', required=False, help='The type of the asset (asset, repository, etc.)', default=Kind.ASSET.value)
 @click.option('-s', '--status', type=click.Choice([s.value for s in Asset]), required=False,
               default=Asset.ACTIVE.value, help=f'Status of the asset', show_default=True)
 @click.option('-f', '--surface', required=False, default='', help=f'Attack surface of the asset', show_default=False)
-def asset(sdk, name, dns, status, surface):
+def asset(sdk, name, dns, asset_type, status, surface):
     """ Add an asset
 
     Add an asset to the Chariot database. This command requires a DNS name for the asset.
@@ -43,7 +44,7 @@ def asset(sdk, name, dns, status, surface):
     """
     if not name:
         name = dns
-    sdk.assets.add(dns, name, status, surface)
+    sdk.assets.add(dns, name, asset_type, status, surface)
 
 
 @add.command()

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/handlers/chariot.py

@@ -10,6 +10,8 @@ def chariot(click_context):
     """ Command group for interacting with the Chariot product """
     # Replace the click context (previously a Keychain instance) with a Chariot
     # instance, after creating it using the Keychain instance.
-    keychain = click_context.obj
-    chariot = Chariot(keychain=keychain)
+    keychain = click_context.obj['keychain']
+    proxy = click_context.obj['proxy']
+
+    chariot = Chariot(keychain=keychain, proxy=proxy)
     click_context.obj = chariot

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/handlers/cli_decorators.py

@@ -52,13 +52,16 @@ def cli_handler(func):
     return func
 
 
-def list_params(filter_by, has_details=True):
+def list_params(filter_by, has_details=True, has_filter=True, has_type=False):
     def decorator(func):
-        func = click.option('-f', '--filter', default='', help=f'Filter by {filter_by}')(func)
         func = pagination(func)
         func = cli_handler(func)
+        if has_filter:
+            func = click.option('-f', '--filter', default='', help=f'Filter by {filter_by}')(func)
         if has_details:
             func = click.option('-d', '--details', is_flag=True, default=False, help='Show detailed information')(func)
+        if has_type:
+            func = click.option('-t', '--type', "model_type", default='', help='Select only a subset by type')(func)
         return func
 
     return decorator

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/handlers/list.py

@@ -12,8 +12,8 @@ def list():
 
 
 @list.command()
-@list_params('DNS')
-def assets(chariot, filter, details, offset, page):
+@list_params('DNS', has_type=True)
+def assets(chariot, filter, model_type, details, offset, page):
     """ List assets
 
    	Retrieve and display a list of assets.
@@ -24,8 +24,9 @@ def assets(chariot, filter, details, offset, page):
         - praetorian chariot list assets --filter api.example.com
         - praetorian chariot list assets --details
         - praetorian chariot list assets --page all
+        - praetorian chariot list assets --type repository
     """
-    render_list_results(chariot.assets.list(filter, offset, pagination_size(page)), details)
+    render_list_results(chariot.assets.list(filter, model_type, offset, pagination_size(page)), details)
 
 
 @list.command()
@@ -309,7 +310,7 @@ def keys(chariot, details, offset, page):
     render_list_results(chariot.keys.list(offset, pagination_size(page)), details)
 
 @list.command()
-@list_params('credential ID')
+@list_params('credential ID', has_details=False, has_filter=False)
 def credentials(chariot, offset, page):
     """ List credentials
 

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/main.py

@@ -22,13 +22,14 @@ from praetorian_cli.sdk.keychain import Keychain
 @click.option('--profile', default='United States', help='The profile to use in the keychain file', show_default=True)
 @click.option('--account', default=None, help='Assume role into this account')
 @click.option('--debug', is_flag=True, default=False, help='Run the CLI in debug mode')
+@click.option('--proxy', default='', help='The proxy to use in the CLI')
 @click.pass_context
 @click.version_option()
-def main(click_context, profile, account, debug):
+def main(click_context, profile, account, debug, proxy):
     if debug:
         click.echo('Running in debug mode.')
     chariot.is_debug = debug
-    click_context.obj = Keychain(profile, account)
+    click_context.obj = {'keychain': Keychain(profile, account), 'proxy': proxy}
     praetorian_cli.handlers.script.load_dynamic_commands()
 
 

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/sdk/chariot.py

@@ -28,7 +28,7 @@ from praetorian_cli.sdk.model.query import Query, my_params_to_query, DEFAULT_PA
 
 class Chariot:
 
-    def __init__(self, keychain: Keychain):
+    def __init__(self, keychain: Keychain, proxy: str=''):
         self.keychain = keychain
         self.assets = Assets(self)
         self.seeds = Seeds(self)
@@ -49,6 +49,23 @@ class Chariot:
         self.keys = Keys(self)
         self.capabilities = Capabilities(self)
         self.credentials = Credentials(self)
+        self.proxy = proxy
+
+        if self.proxy:
+            import urllib3
+            urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+    def _make_request(self, method: str, url: str, **kwargs) -> requests.Response:
+        """
+        Centralized method to make HTTP requests with proxy configuration.
+        Automatically applies proxy settings and verify=False if proxy is defined.
+        """
+
+        if self.proxy:
+            kwargs['proxies'] = {'http': self.proxy, 'https': self.proxy}
+            kwargs['verify'] = False
+
+        return requests.request(method, url, headers=self.keychain.headers(), **kwargs)
 
     def my(self, params: dict, pages=1) -> dict:
         final_resp = dict()
@@ -60,7 +77,7 @@ class Chariot:
 
         # The search is on data in DynamoDB, which uses DynamoDB's native offset format.
         for _ in range(pages):
-            resp = requests.get(self.url('/my'), params=params, headers=self.keychain.headers())
+            resp = self._make_request('GET', self.url('/my'), params=params)
             process_failure(resp)
             resp = resp.json()
             extend(final_resp, resp)
@@ -87,7 +104,7 @@ class Chariot:
         final_resp = dict()
 
         while pages > 0:
-            resp = requests.post(self.url('/my'), json=raw_query, params=params, headers=self.keychain.headers())
+            resp = self._make_request('POST', self.url('/my'), json=raw_query, params=params)
             if is_query_limit_failure(resp):
                 # In this block, the data size is too large for the number of records requested in raw_query['limit'].
                 # We need to halve the page size: LIMIT = LIMIT / 2
@@ -114,22 +131,22 @@ class Chariot:
         return final_resp
 
     def post(self, type: str, body: dict, params: dict = {}) -> dict:
-        resp = requests.post(self.url(f'/{type}'), json=body, params=params, headers=self.keychain.headers())
+        resp = self._make_request('POST', self.url(f'/{type}'), json=body, params=params)
         process_failure(resp)
         return resp.json()
 
     def put(self, type: str, body: dict, params: dict = {}) -> dict:
-        resp = requests.put(self.url(f'/{type}'), json=body, params=params, headers=self.keychain.headers())
+        resp = self._make_request('PUT', self.url(f'/{type}'), json=body, params=params)
         process_failure(resp)
         return resp.json()
 
     def get(self, type: str, params: dict = {}) -> dict:
-        resp = requests.get(self.url(f'/{type}'), params=params, headers=self.keychain.headers())
+        resp = self._make_request('GET', self.url(f'/{type}'), params=params)
         process_failure(resp)
         return resp.json()
 
     def delete(self, type: str, body: dict, params: dict) -> dict:
-        resp = requests.delete(self.url(f'/{type}'), json=body, params=params, headers=self.keychain.headers())
+        resp = self._make_request('DELETE', self.url(f'/{type}'), json=body, params=params)
         process_failure(resp)
         return resp.json()
 
@@ -149,14 +166,12 @@ class Chariot:
         return self.put(type, body, params)
 
     def link_account(self, username: str, value: str = '', config: dict = {}) -> dict:
-        resp = requests.post(self.url(f'/account/{username}'), json=dict(config=config, value=value),
-                             headers=self.keychain.headers())
+        resp = self._make_request('POST', self.url(f'/account/{username}'), json=dict(config=config, value=value))
         process_failure(resp)
         return resp.json()
 
     def unlink(self, username: str, value: str = '', config: dict = {}) -> dict:
-        resp = requests.delete(self.url(f'/account/{username}'), headers=self.keychain.headers(),
-                               json=dict(value=value, config=config))
+        resp = self._make_request('DELETE', self.url(f'/account/{username}'), json=dict(value=value, config=config))
         process_failure(resp)
         return resp.json()
 
@@ -170,8 +185,7 @@ class Chariot:
     def _upload(self, chariot_filepath: str, content: str) -> dict:
         # It is a two-step upload. The PUT request to the /file endpoint is to get a presigned URL for S3.
         # There is no data transfer.
-        presigned_url = requests.put(self.url('/file'), params=dict(name=chariot_filepath),
-                                     headers=self.keychain.headers())
+        presigned_url = self._make_request('PUT', self.url('/file'), params=dict(name=chariot_filepath))
         process_failure(presigned_url)
         resp = requests.put(presigned_url.json()['url'], data=content)
         process_failure(resp)
@@ -182,12 +196,12 @@ class Chariot:
         if global_:
             params |= GLOBAL_FLAG
 
-        resp = requests.get(self.url('/file'), params=params, allow_redirects=True, headers=self.keychain.headers())
+        resp = self._make_request('GET', self.url('/file'), params=params, allow_redirects=True)
         process_failure(resp)
         return resp.content
 
     def count(self, params: dict) -> dict:
-        resp = requests.get(self.url('/my/count'), params=params, headers=self.keychain.headers())
+        resp = self._make_request('GET', self.url('/my/count'), params=params)
         process_failure(resp)
         return resp.json()
 
@@ -196,11 +210,11 @@ class Chariot:
         return json.loads(self.download(f'enrichments/{type}/{filename}', True).decode('utf-8'))
 
     def purge(self):
-        requests.delete(self.url('/account/purge'), headers=self.keychain.headers())
+        self._make_request('DELETE', self.url('/account/purge'))
 
     def agent(self, agent: str, body: dict) -> dict:
         body = body | dict(agent=agent)
-        resp = requests.put(self.url('/agent'), json=body, headers=self.keychain.headers())
+        resp = self._make_request('PUT', self.url('/agent'), json=body)
         process_failure(resp)
         return resp.json()
 

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/sdk/entities/assets.py

@@ -9,20 +9,20 @@ class Assets:
     def __init__(self, api):
         self.api = api
 
-    def add(self, dns, name, status=Asset.ACTIVE.value, surface=''):
+    def add(self, group, identifier, type=Kind.ASSET.value, status=Asset.ACTIVE.value, surface=''):
         """ Add an asset
 
         Arguments:
-        dns: str
-            The DNS name of the asset
-        name: str
-            The name of the asset
+        group: str
+            The group (DNS, repository URL, etc.) of the asset
+        identifier: str
+            The identifier (IP address, repository name, etc.) of the asset
         status: str
             The status of the asset
         surface: str
             The attack surface of the asset
         """
-        return self.api.upsert('asset', dict(dns=dns, name=name, status=status, surface=[surface]))[0]
+        return self.api.upsert('asset', dict(group=group, identifier=identifier, status=status, surface=[surface], type=type))[0]
 
     def get(self, key, details=False):
         """ Get details of an asset
@@ -65,7 +65,7 @@ class Assets:
         """
         return self.api.delete_by_key('asset', key)
 
-    def list(self, prefix_filter='', offset=None, pages=100000) -> tuple:
+    def list(self, prefix_filter='', asset_type='', offset=None, pages=100000) -> tuple:
         """ List assets
 
         Arguments:
@@ -78,7 +78,12 @@ class Assets:
         pages: int
             The number of pages of results to retrieve.
         """
-        return self.api.search.by_key_prefix(f'#asset#{prefix_filter}', offset, pages)
+        dns_prefix = ''
+        if prefix_filter:
+            dns_prefix = f'group:{prefix_filter}'
+        if asset_type == '':
+            asset_type = Kind.ASSET.value
+        return self.api.search.by_term(dns_prefix, asset_type, offset, pages)
 
     def attributes(self, key):
         """ list associated attributes """

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/sdk/model/globals.py

@@ -100,6 +100,9 @@ class AgentType(Enum):
 
 class Kind(Enum):
     ASSET = 'asset'
+    REPOSITORY = 'repository'
+    INTEGRATION = 'integration'
+    ADDOMAIN = 'addomain'
     RISK = 'risk'
     ATTRIBUTE = 'attribute'
     SEED = 'seed'

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/sdk/model/query.py

@@ -21,6 +21,8 @@ class Filter:
 
     class Field(Enum):
         KEY = 'key'
+        GROUP = 'group'
+        IDENTIFIER = 'identifier'
         DNS = 'dns'
         NAME = 'name'
         VALUE = 'value'
@@ -87,6 +89,9 @@ class Relationship:
 class Node:
     class Label(Enum):
         ASSET = 'Asset'
+        REPOSITORY = 'Repository'
+        INTEGRATION = 'Integration'
+        ADDOMAIN = 'Addomain'
         ATTRIBUTE = 'Attribute'
         RISK = 'Risk'
         PRESEED = 'Preseed'
@@ -149,6 +154,9 @@ KIND_TO_LABEL = {
     Kind.ATTRIBUTE.value: Node.Label.ATTRIBUTE,
     Kind.SEED.value: Node.Label.SEED,
     Kind.PRESEED.value: Node.Label.PRESEED,
+    Kind.REPOSITORY.value: Node.Label.REPOSITORY,
+    Kind.INTEGRATION.value: Node.Label.INTEGRATION,
+    Kind.ADDOMAIN.value: Node.Label.ADDOMAIN,
 }
 
 
@@ -196,8 +204,8 @@ def my_params_to_query(params: dict):
             return None
 
         field = Filter.Field(key.split(':')[0])
-        value = key.split(':')[1]
-        operator = Filter.Operator.STARTS_WITH
+        value = key.split(':', 1)[1]
+        operator = Filter.Operator.CONTAINS
 
     filter = Filter(field, operator, value)
 

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/sdk/model/utils.py

@@ -3,6 +3,14 @@
 def asset_key(dns, name):
     return f'#asset#{dns}#{name}'
 
+def ad_domain_key(dns, name):
+    return f'#addomain#{dns}#{name}'
+
+def repository_key(dns, name):
+    return f'#repository#{dns}#{name}'
+
+def integration_key(dns, name):
+    return f'#integration#{dns}#{name}'
 
 def risk_key(dns, name):
     return f'#risk#{dns}#{name}'

praetorian_cli-2.1.5/praetorian_cli/sdk/test/test_asset.py

@@ -0,0 +1,83 @@
+import pytest
+
+from praetorian_cli.sdk.model.globals import Asset, Kind
+from praetorian_cli.sdk.test.utils import make_test_values, clean_test_entities, setup_chariot
+
+
+@pytest.mark.coherence
+class TestAsset:
+
+    def setup_class(self):
+        self.sdk = setup_chariot()
+        make_test_values(self)
+
+    def test_add_asset(self):
+        asset = self.sdk.assets.add(self.asset_dns, self.asset_name, status=Asset.ACTIVE.value, surface='test-surface')
+        assert asset['key'] == self.asset_key
+        assert len(asset['surface']) == 1
+        assert asset['surface'][0] == 'test-surface'
+        assert asset['status'] == Asset.ACTIVE.value
+
+    def test_get_asset(self):
+        a = self.get_asset()
+        assert a['group'] == self.asset_dns
+        assert a['identifier'] == self.asset_name
+        assert a['status'] == Asset.ACTIVE.value
+
+    def test_list_asset(self):
+        results, _ = self.sdk.assets.list()
+        assert len(results) > 0
+        assert any([a['group'] == self.asset_dns for a in results])
+
+    def test_update_asset(self):
+        self.sdk.assets.update(self.asset_key, status=Asset.FROZEN.value)
+        assert self.get_asset()['status'] == Asset.FROZEN.value
+        self.sdk.assets.update(self.asset_key, surface='abc')
+        attributes = self.sdk.assets.attributes(self.asset_key)
+        assert any([a['name'] == 'surface' and a['value'] == 'abc' for a in attributes])
+
+    def test_delete_asset(self):
+        self.sdk.assets.delete(self.asset_key)
+        assert self.get_asset()['status'] == Asset.DELETED.value
+        deleted_assets, _ = self.sdk.search.by_status(Asset.DELETED.value, Kind.ASSET.value)
+        assert any([a['group'] == self.asset_dns for a in deleted_assets])
+    
+    def test_add_ad_domain(self):
+        asset = self.sdk.assets.add(self.ad_domain_name, self.ad_domain_name, status=Asset.ACTIVE.value, surface='test-surface', type=Kind.ADDOMAIN.value)
+        assert asset['key'] == self.ad_domain_key
+        assert len(asset['surface']) == 1
+        assert asset['surface'][0] == 'test-surface'
+        assert asset['status'] == Asset.ACTIVE.value
+    
+    def test_get_ad_domain(self):
+        asset = self.sdk.assets.get(self.ad_domain_key)
+        assert asset['key'] == self.ad_domain_key
+        assert asset['name'] == self.ad_domain_name
+        assert asset['status'] == Asset.ACTIVE.value
+    
+    def test_list_ad_domain(self):
+        results, _ = self.sdk.assets.list(asset_type=Kind.ADDOMAIN.value)
+        assert len(results) > 0
+        assert any([a['group'] == self.ad_domain_name for a in results])
+    
+    def test_update_ad_domain(self):
+        self.sdk.assets.update(self.ad_domain_key, status=Asset.FROZEN.value)
+        assert self.get_ad_domain()['status'] == Asset.FROZEN.value
+        self.sdk.assets.update(self.ad_domain_key, surface='abc')
+        attributes = self.sdk.assets.attributes(self.ad_domain_key)
+        assert any([a['name'] == 'surface' and a['value'] == 'abc' for a in attributes])
+    
+    def test_delete_ad_domain(self):
+        self.sdk.assets.delete(self.ad_domain_key)
+        assert self.get_ad_domain()['status'] == Asset.DELETED.value
+        deleted_assets, _ = self.sdk.search.by_status(Asset.DELETED.value, Kind.ADDOMAIN.value)
+        assert any([a['key'] == self.ad_domain_key for a in deleted_assets])
+
+    def get_asset(self):
+        return self.sdk.assets.get(self.asset_key)
+    
+    def get_ad_domain(self):
+        return self.sdk.assets.get(self.ad_domain_key)
+
+    def teardown_class(self):
+        clean_test_entities(self.sdk, self)

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/sdk/test/test_search.py

@@ -35,17 +35,17 @@ class TestSearch:
     def test_search_by_status(self):
         hits, _ = self.sdk.search.by_status(Asset.ACTIVE.value, Kind.ASSET.value)
         assert len(hits) >= 1
-        assert any([h['dns'] == self.asset_dns for h in hits])
+        assert any([h['group'] == self.asset_dns for h in hits])
 
     def test_search_by_dns(self):
         hits, _ = self.sdk.search.by_dns(self.asset_dns, Kind.ASSET.value)
         assert len(hits) == 1
-        assert hits[0]['dns'] == self.asset_dns
+        assert hits[0]['group'] == self.asset_dns
 
     def test_search_by_name(self):
         hits, _ = self.sdk.search.by_name(self.asset_name, Kind.ASSET.value)
         assert len(hits) == 1
-        assert hits[0]['dns'] == self.asset_dns
+        assert hits[0]['group'] == self.asset_dns
 
     def teardown_class(self):
         clean_test_entities(self.sdk, self)

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/sdk/test/test_z_cli.py

@@ -260,23 +260,6 @@ class TestZCli:
 
         self.verify(f'delete configuration "{o.configuration_key}"', ignore_stdout=True)
 
-    def test_job_cli(self):
-        o = make_test_values(lambda: None)
-        self.verify(f'add asset -n {o.asset_name} -d {o.asset_dns}')
-
-        config = {"test_config_key": "test_config_value"}
-        config_json = json.dumps(config)
-        self.verify(f'add job -k "{o.asset_key}" --config \'{config_json}\'')
-
-        self.verify(f'list jobs -f {o.asset_dns}', [o.asset_dns])
-        
-        invalid_json = '{"invalid_json": "missing_closing_brace"'
-        self.verify(f'add job -k "{o.asset_key}" --config \'{invalid_json}\'', 
-                   expected_stderr=["Invalid JSON in configuration string"])
-        
-        clean_test_entities(self.sdk, o)
-
-
     def test_help_cli(self):
         self.verify('--help', ignore_stdout=True)
         self.verify('list --help', ignore_stdout=True)

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/praetorian_cli/sdk/test/utils.py

@@ -5,7 +5,7 @@ from random import randint
 from praetorian_cli.sdk.chariot import Chariot
 from praetorian_cli.sdk.keychain import Keychain
 from praetorian_cli.sdk.model.globals import Risk, Preseed
-from praetorian_cli.sdk.model.utils import risk_key, asset_key, attribute_key, seed_key, preseed_key, setting_key, configuration_key
+from praetorian_cli.sdk.model.utils import risk_key, asset_key, ad_domain_key, attribute_key, seed_key, preseed_key, setting_key, configuration_key
 
 
 def epoch_micro():
@@ -24,10 +24,15 @@ def random_dns():
     return f'test-{epoch_micro()}.com'
 
 
+def random_ad_domain():
+    return f'test-{epoch_micro()}.local'
+
 def make_test_values(o):
     o.asset_dns = random_dns()
     o.asset_name = random_ip()
     o.asset_key = asset_key(o.asset_dns, o.asset_name)
+    o.ad_domain_name = random_ad_domain()
+    o.ad_domain_key = ad_domain_key(o.ad_domain_name, o.ad_domain_name)
     o.seed_dns = random_dns()
     o.seed_key = seed_key('domain', o.seed_dns)
     o.risk_name = f'test-risk-name-{epoch_micro()}'

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5/praetorian_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praetorian-cli
-Version: 2.1.3
+Version: 2.1.5
 Summary: For interacting with the Chariot API
 Home-page: https://github.com/praetorian-inc/praetorian-cli
 Author: Praetorian

{praetorian_cli-2.1.3 → praetorian_cli-2.1.5}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = praetorian-cli
-version = 2.1.3
+version = 2.1.5
 author = Praetorian
 author_email = support@praetorian.com
 description = For interacting with the Chariot API

praetorian_cli-2.1.3/praetorian_cli/sdk/test/test_asset.py

@@ -1,50 +0,0 @@
-import pytest
-
-from praetorian_cli.sdk.model.globals import Asset, Kind
-from praetorian_cli.sdk.test.utils import make_test_values, clean_test_entities, setup_chariot
-
-
-@pytest.mark.coherence
-class TestAsset:
-
-    def setup_class(self):
-        self.sdk = setup_chariot()
-        make_test_values(self)
-
-    def test_add_asset(self):
-        asset = self.sdk.assets.add(self.asset_dns, self.asset_name, Asset.ACTIVE.value, 'test-surface')
-        print(asset)
-        assert asset['key'] == self.asset_key
-        assert len(asset['surface']) == 1
-        assert asset['surface'][0] == 'test-surface'
-        assert asset['status'] == Asset.ACTIVE.value
-
-    def test_get_asset(self):
-        a = self.get_asset()
-        assert a['dns'] == self.asset_dns
-        assert a['name'] == self.asset_name
-        assert a['status'] == Asset.ACTIVE.value
-
-    def test_list_asset(self):
-        results, _ = self.sdk.assets.list()
-        assert len(results) > 0
-        assert any([a['dns'] == self.asset_dns for a in results])
-
-    def test_update_asset(self):
-        self.sdk.assets.update(self.asset_key, status=Asset.FROZEN.value)
-        assert self.get_asset()['status'] == Asset.FROZEN.value
-        self.sdk.assets.update(self.asset_key, surface='abc')
-        attributes = self.sdk.assets.attributes(self.asset_key)
-        assert any([a['name'] == 'surface' and a['value'] == 'abc' for a in attributes])
-
-    def test_delete_asset(self):
-        self.sdk.assets.delete(self.asset_key)
-        assert self.get_asset()['status'] == Asset.DELETED.value
-        deleted_assets, _ = self.sdk.search.by_status(Asset.DELETED.value, Kind.ASSET.value)
-        assert any([a['dns'] == self.asset_dns for a in deleted_assets])
-
-    def get_asset(self):
-        return self.sdk.assets.get(self.asset_key)
-
-    def teardown_class(self):
-        clean_test_entities(self.sdk, self)