from __future__ import absolute_import, unicode_literals

import inspect
import logging
import os
import unittest
import vcr


logger = logging.getLogger(__name__)


class VCRTestCase(unittest.TestCase):
    vcr_enabled = True

    def setUp(self):
        super(VCRTestCase, self).setUp()
        if self.vcr_enabled:
            myvcr = vcr.VCR(**self._get_vcr_kwargs())
            name = self._get_cassette_name()
            cm = myvcr.use_cassette(name)
            self.cassette = cm.__enter__()
            self.addCleanup(cm.__exit__, None, None, None)

    def _get_vcr_kwargs(self):
        return dict(
            cassette_library_dir=self._get_cassette_library_dir(),
        )

    def _get_cassette_library_dir(self):
        testdir = os.path.dirname(inspect.getfile(self.__class__))
        return os.path.join(testdir, 'cassettes')

    def _get_cassette_name(self):
        return '{0}.{1}'.format(self.__class__.__name__,
                                self._testMethodName)

from __future__ import absolute_import, unicode_literals

import inspect
import logging
import os
import unittest
import vcr


logger = logging.getLogger(__name__)


class VCRTestCase(unittest.TestCase):
    vcr_enabled = True

    def setUp(self):
        super(VCRTestCase, self).setUp()
        if self.vcr_enabled:
            myvcr = vcr.VCR(**self._get_vcr_kwargs())
            name = self._get_cassette_name()
            cm = myvcr.use_cassette(name)
            self.cassette = cm.__enter__()
            self.addCleanup(cm.__exit__, None, None, None)

    def _get_vcr_kwargs(self):
        return dict(
            cassette_library_dir=self._get_cassette_library_dir(),
        )

    def _get_cassette_library_dir(self):
        testdir = os.path.dirname(inspect.getfile(self.__class__))
        return os.path.join(testdir, 'cassettes')

    def _get_cassette_name(self):
        return '{0}.{1}.yaml'.format(self.__class__.__name__,
                                     self._testMethodName)

Add .yaml extension on default cassette name.
Add .yaml extension on default cassette name.

  from __future__ import absolute_import, unicode_literals
  
  import inspect
  import logging
  import os
  import unittest
  import vcr
  
  
  logger = logging.getLogger(__name__)
  
  
  class VCRTestCase(unittest.TestCase):
      vcr_enabled = True
  
      def setUp(self):
          super(VCRTestCase, self).setUp()
          if self.vcr_enabled:
              myvcr = vcr.VCR(**self._get_vcr_kwargs())
              name = self._get_cassette_name()
              cm = myvcr.use_cassette(name)
              self.cassette = cm.__enter__()
              self.addCleanup(cm.__exit__, None, None, None)
  
      def _get_vcr_kwargs(self):
          return dict(
              cassette_library_dir=self._get_cassette_library_dir(),
          )
  
      def _get_cassette_library_dir(self):
          testdir = os.path.dirname(inspect.getfile(self.__class__))
          return os.path.join(testdir, 'cassettes')
  
      def _get_cassette_name(self):
-         return '{0}.{1}'.format(self.__class__.__name__,
+         return '{0}.{1}.yaml'.format(self.__class__.__name__,
-                                 self._testMethodName)
+                                      self._testMethodName)
  
Add .yaml extension on default cassette name.
## Code Before:
from __future__ import absolute_import, unicode_literals

import inspect
import logging
import os
import unittest
import vcr


logger = logging.getLogger(__name__)


class VCRTestCase(unittest.TestCase):
    vcr_enabled = True

    def setUp(self):
        super(VCRTestCase, self).setUp()
        if self.vcr_enabled:
            myvcr = vcr.VCR(**self._get_vcr_kwargs())
            name = self._get_cassette_name()
            cm = myvcr.use_cassette(name)
            self.cassette = cm.__enter__()
            self.addCleanup(cm.__exit__, None, None, None)

    def _get_vcr_kwargs(self):
        return dict(
            cassette_library_dir=self._get_cassette_library_dir(),
        )

    def _get_cassette_library_dir(self):
        testdir = os.path.dirname(inspect.getfile(self.__class__))
        return os.path.join(testdir, 'cassettes')

    def _get_cassette_name(self):
        return '{0}.{1}'.format(self.__class__.__name__,
                                self._testMethodName)

## Instruction:
Add .yaml extension on default cassette name.
## Code After:
from __future__ import absolute_import, unicode_literals

import inspect
import logging
import os
import unittest
import vcr


logger = logging.getLogger(__name__)


class VCRTestCase(unittest.TestCase):
    vcr_enabled = True

    def setUp(self):
        super(VCRTestCase, self).setUp()
        if self.vcr_enabled:
            myvcr = vcr.VCR(**self._get_vcr_kwargs())
            name = self._get_cassette_name()
            cm = myvcr.use_cassette(name)
            self.cassette = cm.__enter__()
            self.addCleanup(cm.__exit__, None, None, None)

    def _get_vcr_kwargs(self):
        return dict(
            cassette_library_dir=self._get_cassette_library_dir(),
        )

    def _get_cassette_library_dir(self):
        testdir = os.path.dirname(inspect.getfile(self.__class__))
        return os.path.join(testdir, 'cassettes')

    def _get_cassette_name(self):
        return '{0}.{1}.yaml'.format(self.__class__.__name__,
                                     self._testMethodName)
