aboutsummaryrefslogtreecommitdiffstats
path: root/github3/tests/handler_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'github3/tests/handler_test.py')
-rw-r--r--github3/tests/handler_test.py24
1 files changed, 23 insertions, 1 deletions
diff --git a/github3/tests/handler_test.py b/github3/tests/handler_test.py
index e691182..e95374b 100644
--- a/github3/tests/handler_test.py
+++ b/github3/tests/handler_test.py
@@ -4,7 +4,7 @@
from mock import Mock, patch
from unittest import TestCase
from github3 import api
-from github3.handlers.base import Handler
+from github3.handlers.base import Handler, MimeTypeMixin
from github3.exceptions import *
from github3.converters import *
from github3.models.user import User
@@ -13,6 +13,28 @@ import json
import requests
+class TestMimeTypeMixin(TestCase):
+
+ def setUp(self):
+ self.mixin = MimeTypeMixin()
+
+ def _parse_mime_type(self, type):
+ return 'application/vnd.github.%s.%s+json' % (
+ MimeTypeMixin.VERSION, type)
+
+ def test_add_mimetypes(self):
+ self.mixin.add_raw()
+ self.mixin.add_text()
+ self.mixin.add_html()
+ self.mixin.add_full()
+ self.assertEquals(sorted(self.mixin.mime_header()), sorted({
+ 'Accept': '%s, %s, %s, %s' % (
+ self._parse_mime_type('raw'),
+ self._parse_mime_type('text'),
+ self._parse_mime_type('html'),
+ self._parse_mime_type('full'))}))
+
+
class TestHandler(TestCase):
def setUp(self):