import unittest
import os
from tempfile import NamedTemporaryFile

from .. import archivebot


class TestRename(unittest.TestCase):
    def test_valid_rename(self):
        try:
            tmp = NamedTemporaryFile()
            new_tmp = archivebot._rename(tmp.name)
            self.assertFalse(os.path.isfile(tmp.name))
            self.assertTrue(os.path.isfile(new_tmp))
            self.assertTrue(new_tmp.find(".compress") > 1)
        finally:
            if os.path.isfile(tmp.name):
                os.remove(tmp.name)
            if os.path.isfile(new_tmp):
                os.remove(new_tmp)


class TestCompress(unittest.TestCase):
    def test_dotcompress(self):
        with NamedTemporaryFile(prefix="roomname.compress@example") as tmp:
            self.assertRaises(ValueError, archivebot.compress, tmp.name)

    def test_valid_compress(self):
        try:
            tmp = NamedTemporaryFile(suffix=".compress")
            tmp.write("test")
            gz_file = archivebot.compress(tmp.name)
            self.assertEqual(gz_file, tmp.name.replace(".compress", ".gz"))
        finally:
            if os.path.isfile(tmp.name):
                os.remove(tmp.name)

import unittest
import os
from tempfile import NamedTemporaryFile

from .. import archivebot


class TestRename(unittest.TestCase):
    def test_valid_rename(self):
        try:
            tmp = NamedTemporaryFile()
            new_tmp = archivebot._rename(tmp.name)
            self.assertFalse(os.path.isfile(tmp.name))
            self.assertTrue(os.path.isfile(new_tmp))
            self.assertTrue(new_tmp.find(".compress") > 1)
        finally:
            for file in [tmp.name, new_tmp]:
                if os.path.isfile(file):
                    os.remove(file)


class TestCompress(unittest.TestCase):
    def test_dotcompress(self):
        with NamedTemporaryFile(prefix="roomname.compress@example") as tmp:
            self.assertRaises(ValueError, archivebot.compress, tmp.name)

    def test_valid_compress(self):
        try:
            tmp = NamedTemporaryFile(suffix=".compress")
            tmp.write("test")
            gz_file = archivebot.compress(tmp.name)
            self.assertEqual(gz_file, tmp.name.replace(".compress", ".gz"))
        finally:
            for file in [tmp.name, gz_file]:
                if os.path.isfile(file):
                    os.remove(file)

Remove all temporary files from disk
Remove all temporary files from disk

  import unittest
  import os
  from tempfile import NamedTemporaryFile
  
  from .. import archivebot
  
  
  class TestRename(unittest.TestCase):
      def test_valid_rename(self):
          try:
              tmp = NamedTemporaryFile()
              new_tmp = archivebot._rename(tmp.name)
              self.assertFalse(os.path.isfile(tmp.name))
              self.assertTrue(os.path.isfile(new_tmp))
              self.assertTrue(new_tmp.find(".compress") > 1)
          finally:
+             for file in [tmp.name, new_tmp]:
-             if os.path.isfile(tmp.name):
-                 os.remove(tmp.name)
-             if os.path.isfile(new_tmp):
+                 if os.path.isfile(file):
-                 os.remove(new_tmp)
+                     os.remove(file)
  
  
  class TestCompress(unittest.TestCase):
      def test_dotcompress(self):
          with NamedTemporaryFile(prefix="roomname.compress@example") as tmp:
              self.assertRaises(ValueError, archivebot.compress, tmp.name)
  
      def test_valid_compress(self):
          try:
              tmp = NamedTemporaryFile(suffix=".compress")
              tmp.write("test")
              gz_file = archivebot.compress(tmp.name)
              self.assertEqual(gz_file, tmp.name.replace(".compress", ".gz"))
          finally:
+             for file in [tmp.name, gz_file]:
-             if os.path.isfile(tmp.name):
+                 if os.path.isfile(file):
-                 os.remove(tmp.name)
+                     os.remove(file)
  
Remove all temporary files from disk
## Code Before:
import unittest
import os
from tempfile import NamedTemporaryFile

from .. import archivebot


class TestRename(unittest.TestCase):
    def test_valid_rename(self):
        try:
            tmp = NamedTemporaryFile()
            new_tmp = archivebot._rename(tmp.name)
            self.assertFalse(os.path.isfile(tmp.name))
            self.assertTrue(os.path.isfile(new_tmp))
            self.assertTrue(new_tmp.find(".compress") > 1)
        finally:
            if os.path.isfile(tmp.name):
                os.remove(tmp.name)
            if os.path.isfile(new_tmp):
                os.remove(new_tmp)


class TestCompress(unittest.TestCase):
    def test_dotcompress(self):
        with NamedTemporaryFile(prefix="roomname.compress@example") as tmp:
            self.assertRaises(ValueError, archivebot.compress, tmp.name)

    def test_valid_compress(self):
        try:
            tmp = NamedTemporaryFile(suffix=".compress")
            tmp.write("test")
            gz_file = archivebot.compress(tmp.name)
            self.assertEqual(gz_file, tmp.name.replace(".compress", ".gz"))
        finally:
            if os.path.isfile(tmp.name):
                os.remove(tmp.name)

## Instruction:
Remove all temporary files from disk
## Code After:
import unittest
import os
from tempfile import NamedTemporaryFile

from .. import archivebot


class TestRename(unittest.TestCase):
    def test_valid_rename(self):
        try:
            tmp = NamedTemporaryFile()
            new_tmp = archivebot._rename(tmp.name)
            self.assertFalse(os.path.isfile(tmp.name))
            self.assertTrue(os.path.isfile(new_tmp))
            self.assertTrue(new_tmp.find(".compress") > 1)
        finally:
            for file in [tmp.name, new_tmp]:
                if os.path.isfile(file):
                    os.remove(file)


class TestCompress(unittest.TestCase):
    def test_dotcompress(self):
        with NamedTemporaryFile(prefix="roomname.compress@example") as tmp:
            self.assertRaises(ValueError, archivebot.compress, tmp.name)

    def test_valid_compress(self):
        try:
            tmp = NamedTemporaryFile(suffix=".compress")
            tmp.write("test")
            gz_file = archivebot.compress(tmp.name)
            self.assertEqual(gz_file, tmp.name.replace(".compress", ".gz"))
        finally:
            for file in [tmp.name, gz_file]:
                if os.path.isfile(file):
                    os.remove(file)
