def test_delete_workload_policy(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        resp = self.dr.delete_workload_policy('workload_policy_id', **kwargs)

        self.assertTrue(resp == DONE)
    def test_recover(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        resp = self.dr.recover('container_name', **kwargs)

        self.assertTrue(resp == DONE)
    def test_delete_resource_action(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        tuple_id = 1
        resp = self.dr.delete_resource_action(tuple_id, **kwargs)

        self.assertTrue(resp == DONE)
    def test_update_resource_action(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}
        resp = self.dr.update_resource_action(
            'workload_policy_id', '2f3b9a41-7bcb-46fb-82dc-1a5929e5a026', 1, 2,
            **kwargs)

        self.assertTrue(resp == DONE)
    def test_create_resource(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        resp = self.dr.create_resource('2f3b9a41-7bcb-46fb-82dc-1a5929e5a026',
                                       "test1", 2, **kwargs)

        self.assertTrue(resp == DONE)
    def test_protect(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}
        kwargs['body'] = {}
        kwargs['body']['consistent'] = True

        resp = self.dr.protect('workload_policy_id', **kwargs)

        self.assertTrue(resp == DONE)
    def test_list_actions(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}
        body = self.dr.list_actions('1', **kwargs)

        fields = ['id', 'name', 'class_name', 'resource_type_id', 'created_at']
        key_list = body.keys()
        key_string = ''
        key_string = key_string.join(key_list)
        for r in fields:
            self.assertRegexpMatches(key_string, r)
    def test_get_policy_executions(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        body = self.dr.get_policy_executions('policy_execution_id', **kwargs)

        fields = ['id', 'status', 'workload_policy_id']
        key_list = body.keys()
        key_string = ''
        key_string = key_string.join(key_list)
        for r in fields:
            self.assertRegexpMatches(key_string, r)
    def test_get_workload_policy(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        body = self.dr.get_workload_policy('workload_policy_id', **kwargs)

        fields = ['id', 'name', 'tenant_id', 'created_at']
        key_list = body.keys()
        key_string = ''
        key_string = key_string.join(key_list)
        for r in fields:
            self.assertRegexpMatches(key_string, r)
    def test_get_resource(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        body = self.dr.get_resource('2f3b9a41-7bcb-46fb-82dc-1a5929e5a026',
                                    **kwargs)

        fields = ['id', 'name', 'resource_type_id', 'created_at']
        key_list = body.keys()
        key_string = ''
        key_string = key_string.join(key_list)
        for r in fields:
            self.assertRegexpMatches(key_string, r)
    def test_get_policy_resource_actions(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        body = self.dr.get_policy_resource_actions('workload_policy_id',
                                                   **kwargs)

        fields = [
            'id', 'resource_id', 'action_id', 'workload_policy_id',
            'created_at'
        ]
        key_list = body.keys()
        key_string = ''
        key_string = key_string.join(key_list)
        for r in fields:
            self.assertRegexpMatches(key_string, r)
    def test_create_policy(self):

        self.api = api()
        self.dr = dr.DRManager(self.api)
        kwargs = {}

        body =\
            self.dr.create_workload_policy('policy_name',
                                           'c9a54702410e4573bd45909d90ada12c',
                                           **kwargs)

        # self.assertTrue(body["id"] == 'c9a54702410e4573bd45909d90ada12c')
        fields = ['id', 'name', 'tenant_id', 'created_at']
        key_list = body.keys()
        key_string = ''
        key_string = key_string.join(key_list)
        for r in fields:
            self.assertRegexpMatches(key_string, r)
 def __init__(self, *args, **kwargs):
     """Initialize a new client for the Dragon v1 API."""
     super(Client, self).__init__(*args, **kwargs)
     self.dr = dr.DRManager(self)