from django.core.cache import cache
from django.test import TestCase

from opendebates.context_processors import global_vars
from opendebates.models import NUMBER_OF_VOTES_CACHE_ENTRY


class NumberOfVotesTest(TestCase):
    def test_number_of_votes(self):
        cache.set(NUMBER_OF_VOTES_CACHE_ENTRY, 2)
        context = global_vars(None)
        self.assertEqual(2, int(context['NUMBER_OF_VOTES']))

from django.test import TestCase
from mock import patch

from opendebates.context_processors import global_vars


class NumberOfVotesTest(TestCase):
    def test_number_of_votes(self):
        with patch('opendebates.utils.cache') as mock_cache:
            mock_cache.get.return_value = 2
            context = global_vars(None)
            self.assertEqual(2, int(context['NUMBER_OF_VOTES']))

Fix new test to work without cache
Fix new test to work without cache

- from django.core.cache import cache
  from django.test import TestCase
+ from mock import patch
  
  from opendebates.context_processors import global_vars
- from opendebates.models import NUMBER_OF_VOTES_CACHE_ENTRY
  
  
  class NumberOfVotesTest(TestCase):
      def test_number_of_votes(self):
-         cache.set(NUMBER_OF_VOTES_CACHE_ENTRY, 2)
+         with patch('opendebates.utils.cache') as mock_cache:
+             mock_cache.get.return_value = 2
-         context = global_vars(None)
+             context = global_vars(None)
-         self.assertEqual(2, int(context['NUMBER_OF_VOTES']))
+             self.assertEqual(2, int(context['NUMBER_OF_VOTES']))
  
Fix new test to work without cache
## Code Before:
from django.core.cache import cache
from django.test import TestCase

from opendebates.context_processors import global_vars
from opendebates.models import NUMBER_OF_VOTES_CACHE_ENTRY


class NumberOfVotesTest(TestCase):
    def test_number_of_votes(self):
        cache.set(NUMBER_OF_VOTES_CACHE_ENTRY, 2)
        context = global_vars(None)
        self.assertEqual(2, int(context['NUMBER_OF_VOTES']))

## Instruction:
Fix new test to work without cache
## Code After:
from django.test import TestCase
from mock import patch

from opendebates.context_processors import global_vars


class NumberOfVotesTest(TestCase):
    def test_number_of_votes(self):
        with patch('opendebates.utils.cache') as mock_cache:
            mock_cache.get.return_value = 2
            context = global_vars(None)
            self.assertEqual(2, int(context['NUMBER_OF_VOTES']))
