예제 #1
0
 def setUp(self):
     self.SECRET = 'FAKE_WILDDOG_SECRET'
     self.DSN = 'https://scm.wilddogio.com'
     self.EMAIL = '*****@*****.**'
     self.authentication = WilddogAuthentication(self.SECRET, self.EMAIL,
                                                 extra={'uid': '123'})
     self.wilddog = WilddogApplication(self.DSN, self.authentication)
예제 #2
0
class WilddogTestCase(unittest.TestCase):
    def setUp(self):
        self.SECRET = 'FAKE_WILDDOG_SECRET'
        self.DSN = 'https://scm.wilddogio.com'
        self.EMAIL = '*****@*****.**'
        self.authentication = WilddogAuthentication(self.SECRET, self.EMAIL,
                                                    extra={'uid': '123'})
        self.wilddog = WilddogApplication(self.DSN, self.authentication)
        # self.wilddog.set_token('xxxx')

    def test_build_endpoint_url(self):
        url1 = os.path.join(self.DSN, 'users', '1.json')
        self.assertEqual(self.wilddog._build_endpoint_url('/users', '1'), url1)
        url2 = os.path.join(self.DSN, 'users/1/.json')
        self.assertEqual(self.wilddog._build_endpoint_url('/users/1', None), url2)

    def test_make_get_request(self):
        response = MockResponse(403, json.dumps({'error': 'Permission required.'}))
        connection = MockConnection(response)
        result = self.wilddog.get('url', 'shapshot', params={}, headers={},
                                  connection=connection)
        self.assertEqual(result, json.loads(response.content))

    def test_make_post_request(self):
        response = MockResponse(403, json.dumps({'error': 'Permission required.'}))
        connection = MockConnection(response)
        result = self.wilddog.post('url', {}, params={}, headers={},
                                   connection=connection)
        self.assertEqual(result, json.loads(response.content))

    def test_make_put_request(self):
        response = MockResponse(403, json.dumps({'error': 'Permission required.'}))
        connection = MockConnection(response)
        result = self.wilddog.put('url', 'snapshot', {}, params={}, headers={},
                                  connection=connection)
        self.assertEqual(result, json.loads(response.content))

    def test_make_patch_request(self):
        response = MockResponse(403, json.dumps({'error': 'Permission required.'}))
        connection = MockConnection(response)
        result = self.wilddog.patch('url', {}, params={}, headers={},
                                    connection=connection)
        self.assertEqual(result, json.loads(response.content))

    def test_make_delete_request(self):
        response = MockResponse(403, json.dumps({'error': 'Permission required.'}))
        connection = MockConnection(response)
        result = self.wilddog.delete('url', 'snapshot', params={}, headers={},
                                     connection=connection)
        self.assertEqual(result, json.loads(response.content))