# Copyright 1999 by Jeffrey Chang.  All rights reserved.
# This code is part of the Biopython distribution and governed by its
# license.  Please see the LICENSE file that should have been included
# as part of this package.
 
from __future__ import print_function
 
import os.path
import unittest
import shutil
from Bio._py3k import StringIO
import tempfile
 
from Bio import File
 
 
data = """This
is
a multi-line
file"""
 
 
### UndoHandle
 
h = File.UndoHandle(StringIO(data))
 
print(h.readline())   # 'This'
print(h.peekline())   # 'is'
print(h.readline())   # 'is'
h.saveline("saved")
print(h.peekline())   # 'saved'
h.saveline("another")
print(h.readline())   # 'another'
print(h.readline())   # 'saved'
 
# Test readlines after saveline
h.saveline("saved again")
lines = h.readlines()
print(repr(lines[0]))   # 'saved again'
print(repr(lines[1]))   # 'a multi-line'
print(repr(lines[2]))   # 'file'
 
# should be empty now
print(repr(h.readline()))       # ''
 
h.saveline("save after empty")
print(h.readline())             # 'save after empty'
print(repr(h.readline()))       # ''
 
# test read method
h = File.UndoHandle(StringIO("some text"))
h.saveline("more text")
print(h.read())                 # 'more textsome text'
 
 
class AsHandleTestCase(unittest.TestCase):
 
    def setUp(self):
        # Create a directory to work in
        self.temp_dir = tempfile.mkdtemp(prefix='biopython-test')
 
    def tearDown(self):
        shutil.rmtree(self.temp_dir)
 
    def _path(self, *args):
        return os.path.join(self.temp_dir, *args)
 
    def test_handle(self):
        "Test as_handle with a file-like object argument"
        p = self._path('test_file.fasta')
        with open(p, 'wb') as fp:
            with File.as_handle(fp) as handle:
                self.assertEqual(fp, handle, "as_handle should "
                        "return argument when given a file-like object")
                self.assertFalse(handle.closed)
 
            self.assertFalse(handle.closed,
                    "Exiting as_handle given a file-like object should not "
                    "close the file")
 
    def test_path(self):
        "Test as_handle with a path argument"
        p = self._path('test_file.fasta')
        mode = 'wb'
        with File.as_handle(p, mode=mode) as handle:
            self.assertEqual(p, handle.name)
            self.assertEqual(mode, handle.mode)
            self.assertFalse(handle.closed)
        self.assertTrue(handle.closed)
 
    def test_stringio(self):
        s = StringIO()
        with File.as_handle(s) as handle:
            self.assertEqual(s, handle)