# Copyright (C) 2007-2009 Michael Foord
# E-mail: fuzzyman AT voidspace DOT org DOT uk
# http://www.voidspace.org.uk/python/mock/

from __future__ import with_statement

import os
import sys
this_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
if not this_dir in sys.path:
    sys.path.insert(0, this_dir)

from testcase import TestCase
from testutils import RunTests

if __name__ == '__main__':
    sys.modules['testwith'] = sys.modules['__main__']

if 'testwith' in sys.modules:
    # Fix for running tests under Wing
    import tests
    import testwith
    tests.testwith = testwith

from mock import Mock, patch, patch_object, sentinel

something  = sentinel.Something
something_else  = sentinel.SomethingElse


class WithTest(TestCase):
    def testWithStatement(self):
        with patch('tests.testwith.something', sentinel.Something2):
            self.assertEquals(something, sentinel.Something2, "unpatched")        
        self.assertEquals(something, sentinel.Something)
        
    def testWithStatementException(self):
        try:
            with patch('tests.testwith.something', sentinel.Something2):
                self.assertEquals(something, sentinel.Something2, "unpatched")   
                raise Exception('pow')
        except Exception:
            pass
        else:
            self.fail("patch swallowed exception")
        self.assertEquals(something, sentinel.Something)


    def testWithStatementAs(self):
        with patch('tests.testwith.something') as mock_something:
            self.assertEquals(something, mock_something, "unpatched")        
            self.assertTrue(isinstance(mock_something, Mock), "patching wrong type")
        self.assertEquals(something, sentinel.Something)


    def testPatchObjectWithStatementAs(self):
        mock = Mock()
        original = mock.something
        with patch_object(mock, 'something') as mock_something:
            self.assertNotEquals(mock.something, original, "unpatched")        
        self.assertEquals(mock.something, original)


    def testWithStatementNested(self):
        from contextlib import nested
        with nested(patch('tests.testwith.something'), 
                    patch('tests.testwith.something_else')) as (mock_something, mock_something_else):
            self.assertEquals(something, mock_something, "unpatched")
            self.assertEquals(something_else, mock_something_else, "unpatched")
        self.assertEquals(something, sentinel.Something)
        self.assertEquals(something_else, sentinel.SomethingElse)


    def testWithStatementSpecified(self):
        with patch('tests.testwith.something', sentinel.Patched) as mock_something:
            self.assertEquals(something, mock_something, "unpatched")        
            self.assertEquals(mock_something, sentinel.Patched, "wrong patch")        
        self.assertEquals(something, sentinel.Something)



if __name__ == '__main__':
    RunTests(WithTest)