Ejemplo n.º 1
0
 def setUp(self):
     self.cases = Case(
         PropertyMock(url='test_url',
                      team='test_team',
                      token='old_value',
                      verify=True,
                      r=Mock(return_value=(200, {
                          'token': '123'
                      }))))
     self.data = {'a': 1, 'b': 2}
Ejemplo n.º 2
0
 def setUp(self):
     self.cases = Case(
         PropertyMock(url='test_url',
                      team='test_team',
                      signature='test_signature',
                      token='old_value',
                      verify=True,
                      r=Mock(return_value=(200, {
                          'token': '123'
                      }))))
Ejemplo n.º 3
0
class TestGetAllMethod(unittest.TestCase):
    def setUp(self):
        self.cases = Case(
            PropertyMock(url='test_url',
                         team='test_team',
                         signature='test_signature',
                         token='old_value',
                         verify=True,
                         r=Mock(return_value=(200, {
                             'token': '123'
                         }))))

    def test_get_all_request(self):
        self.cases.get_all()
        self.cases.root.r.assert_called_with(
            'GET',
            f'{self.cases.root.url}/api/v1.2/cases',
            body=None,
            headers=None,
            verify=self.cases.root.verify)
Ejemplo n.º 4
0
class TestUpdateMethod(unittest.TestCase):
    def setUp(self):
        self.cases = Case(
            PropertyMock(url='test_url',
                         team='test_team',
                         token='old_value',
                         verify=True,
                         r=Mock(return_value=(200, {
                             'token': '123'
                         }))))
        self.id = 'test_id'
        self.data = {'a': 1, 'b': 2}

    def test_update_request(self):
        self.cases.update(self.id, self.data)
        self.cases.root.r.assert_called_with(
            'PUT',
            f'{self.cases.root.url}/api/v1.2/cases/{self.id}',
            json.dumps(self.data),
            headers=None,
            verify=self.cases.root.verify)

    def test_update_with_incorrect_id_type(self):
        with self.assertRaisesRegex(AssertionError, 'The ID must be a string'):
            self.cases.update(123, self.data)

    def test_update_with_incorrect_data_type(self):
        with self.assertRaisesRegex(AssertionError,
                                    'The data type must be a dictionary'):
            self.cases.update(self.id, 123)

    def test_update_with_empty_id_string(self):
        with self.assertRaisesRegex(AssertionError,
                                    'The ID must not be an empty string'):
            self.cases.update('', self.data)

    def test_update_with_empty_data_dict(self):
        with self.assertRaisesRegex(
                AssertionError, 'Case data must not be an empty dictionary'):
            self.cases.update(self.id, {})
Ejemplo n.º 5
0
class TestDeleteMethod(unittest.TestCase):
    def setUp(self):
        self.cases = Case(
            PropertyMock(url='test_url',
                         team='test_team',
                         token='old_value',
                         verify=True,
                         r=Mock(return_value=(200, {
                             'token': '123'
                         }))))
        self.id = 'test_id'

    def test_delete_request(self):
        self.cases.delete(self.id)
        self.cases.root.r.assert_called_with(
            'DELETE',
            f'{self.cases.root.url}/api/v1.2/cases/{self.id}',
            body=None,
            headers=None,
            verify=self.cases.root.verify)

    def test_update_with_incorrect_id_type(self):
        with self.assertRaisesRegex(AssertionError, 'The ID must be a string'):
            self.cases.delete(123)

    def test_update_with_empty_id_string(self):
        with self.assertRaisesRegex(AssertionError,
                                    'The ID must not be an empty string'):
            self.cases.delete('')
Ejemplo n.º 6
0
    def __init__(self, **settings):
        self.log = logging.getLogger(__name__)
        self.log.debug('Client has been init with %s', settings)

        assert 'url' in settings, 'url is required parameter'
        assert isinstance(settings['url'], str), 'url value must be a string'
        assert 'team' in settings, 'team is required parameter'
        assert isinstance(settings['team'], str), 'team value must be a string'

        self.url = str(settings['url'])
        self.af_url = str(
            settings['custom_af_url']
        ) if 'custom_af_url' in settings else self.url + ':7499'
        self.team = str(settings['team'])
        self.api_user_id = str(
            settings['api_user_id']) if 'api_user_id' in settings else ''
        self.api_signature = str(
            settings['api_signature']) if 'api_signature' in settings else ''
        self.af_version = str(
            settings['af_version']) if 'af_version' in settings else 'v2.2'
        self.verify = bool(
            settings['verify']) if 'verify' in settings else True

        self.token = ''
        self.login_time = 0
        self.ttl = 840  # Session TTL == 14 mins

        self.auth = Auth(self)
        self.subchannels = Subchannel(self)
        self.lists = List(self)
        self.users = User(self)
        self.channels = Channel(self)
        self.policies = Policy(self)
        self.events = Event(self)
        self.roles = Role(self)
        self.abtests = ABTest(self)
        self.af = AFClient(self)
        self.cases = Case(self)
        self.settings = Settings(self)
        self.queues = Queue(self)
        self.sessions = Session(self)
        self.tech = Tech(self)
        self.bi = BusinessIntelligence(self)
        self.schema = Schema(self)
        self.currencies = Currency(self)
        self.geo = Geo(self)
        self.bins = Bin(self)
        self.reports = Report(self)