Don't try to mock context manager; use a simple class instead

Don't try to mock context manager; use a simple class instead #146