Don't try to mock context manager; use a simple class instead (#39)

Don't try to mock context manager; use a simple class instead (#39) #147