from __future__ import unicode_literals

import balanced

from . import utils


class TestClient(utils.TestCase):

    def setUp(self):
        super(TestClient, self).setUp()

    def test_configure(self):
        balanced.configure('XXX')
        expected_headers = {
            'content-type': 'application/vnd.api+json;revision=1.1',
            'accept': 'application/json;revision=1.1',
            'User-Agent': u'balanced-python/1.0dev1'
        }
        self.assertDictContainsSubset(
            expected_headers, balanced.config.client.config.headers
        )

from __future__ import unicode_literals

import balanced

from . import utils


class TestClient(utils.TestCase):

    def setUp(self):
        super(TestClient, self).setUp()

    def test_configure(self):
        expected_headers = {
            'content-type': 'application/vnd.api+json;revision=1.1',
            'accept': 'application/json;revision=1.1',
            'User-Agent': u'balanced-python/1.0dev1'
        }
        self.assertDictContainsSubset(
            expected_headers, balanced.config.client.config.headers
        )

Fix 401 in test suite
Fix 401 in test suite
  from __future__ import unicode_literals
  
  import balanced
  
  from . import utils
  
  
  class TestClient(utils.TestCase):
  
      def setUp(self):
          super(TestClient, self).setUp()
  
      def test_configure(self):
-         balanced.configure('XXX')
          expected_headers = {
              'content-type': 'application/vnd.api+json;revision=1.1',
              'accept': 'application/json;revision=1.1',
              'User-Agent': u'balanced-python/1.0dev1'
          }
          self.assertDictContainsSubset(
              expected_headers, balanced.config.client.config.headers
          )
  
Fix 401 in test suite
## Code Before:
from __future__ import unicode_literals

import balanced

from . import utils


class TestClient(utils.TestCase):

    def setUp(self):
        super(TestClient, self).setUp()

    def test_configure(self):
        balanced.configure('XXX')
        expected_headers = {
            'content-type': 'application/vnd.api+json;revision=1.1',
            'accept': 'application/json;revision=1.1',
            'User-Agent': u'balanced-python/1.0dev1'
        }
        self.assertDictContainsSubset(
            expected_headers, balanced.config.client.config.headers
        )

## Instruction:
Fix 401 in test suite
## Code After:
from __future__ import unicode_literals

import balanced

from . import utils


class TestClient(utils.TestCase):

    def setUp(self):
        super(TestClient, self).setUp()

    def test_configure(self):
        expected_headers = {
            'content-type': 'application/vnd.api+json;revision=1.1',
            'accept': 'application/json;revision=1.1',
            'User-Agent': u'balanced-python/1.0dev1'
        }
        self.assertDictContainsSubset(
            expected_headers, balanced.config.client.config.headers
        )
