
import unittest
import sys

from helpers.statuspage import StatusPage
from test_postgresql import MockConnect

if sys.hexversion >= 0x03000000:
    from io import BytesIO as IO
else:
    from StringIO import StringIO as IO


class TestStatusPage(unittest.TestCase):

    def test_do_GET(self):
        for mock_recovery in [True, False]:
            for page in [b'GET /pg_master', b'GET /pg_slave', b'GET /pg_status', b'GET /not_found']:
                self.http_server = MockServer(('0.0.0.0', 8888), StatusPage, page, mock_recovery)


class MockRequest(object):

    def __init__(self, path):
        self.path = path

    def makefile(self, *args, **kwargs):
        return IO(self.path)


class MockServer(object):

    def __init__(self, ip_port, Handler, path, mock_recovery=False):
        self.postgresql = MockConnect()
        self.postgresql.mock_values['mock_recovery'] = mock_recovery
        Handler(MockRequest(path), ip_port, self)


if __name__ == '__main__':
    unittest.main()


import unittest
import sys

from helpers.statuspage import StatusPage
from test_postgresql import MockConnect

if sys.hexversion >= 0x03000000:
    from io import BytesIO as IO
else:
    from StringIO import StringIO as IO


class TestStatusPage(unittest.TestCase):

    def test_do_GET(self):
        for mock_recovery in [True, False]:
            for page in [b'GET /pg_master', b'GET /pg_slave', b'GET /pg_status', b'GET /not_found']:
                self.http_server = MockServer(('0.0.0.0', 8888), StatusPage, page, mock_recovery)


class MockRequest(object):

    def __init__(self, path):
        self.path = path

    def makefile(self, *args, **kwargs):
        return IO(self.path)


class MockServer(object):

    def __init__(self, ip_port, Handler, path, mock_recovery=False):
        self.postgresql = MockConnect()
        self.postgresql.mock_values['mock_recovery'] = mock_recovery
        Handler(MockRequest(path), ip_port, self)

Remove some more unneccesary code.
Unittests: Remove some more unneccesary code.

  
  import unittest
  import sys
  
  from helpers.statuspage import StatusPage
  from test_postgresql import MockConnect
  
  if sys.hexversion >= 0x03000000:
      from io import BytesIO as IO
  else:
      from StringIO import StringIO as IO
  
  
  class TestStatusPage(unittest.TestCase):
  
      def test_do_GET(self):
          for mock_recovery in [True, False]:
              for page in [b'GET /pg_master', b'GET /pg_slave', b'GET /pg_status', b'GET /not_found']:
                  self.http_server = MockServer(('0.0.0.0', 8888), StatusPage, page, mock_recovery)
  
  
  class MockRequest(object):
  
      def __init__(self, path):
          self.path = path
  
      def makefile(self, *args, **kwargs):
          return IO(self.path)
  
  
  class MockServer(object):
  
      def __init__(self, ip_port, Handler, path, mock_recovery=False):
          self.postgresql = MockConnect()
          self.postgresql.mock_values['mock_recovery'] = mock_recovery
          Handler(MockRequest(path), ip_port, self)
  
- 
- if __name__ == '__main__':
-     unittest.main()
- 
Remove some more unneccesary code.
## Code Before:

import unittest
import sys

from helpers.statuspage import StatusPage
from test_postgresql import MockConnect

if sys.hexversion >= 0x03000000:
    from io import BytesIO as IO
else:
    from StringIO import StringIO as IO


class TestStatusPage(unittest.TestCase):

    def test_do_GET(self):
        for mock_recovery in [True, False]:
            for page in [b'GET /pg_master', b'GET /pg_slave', b'GET /pg_status', b'GET /not_found']:
                self.http_server = MockServer(('0.0.0.0', 8888), StatusPage, page, mock_recovery)


class MockRequest(object):

    def __init__(self, path):
        self.path = path

    def makefile(self, *args, **kwargs):
        return IO(self.path)


class MockServer(object):

    def __init__(self, ip_port, Handler, path, mock_recovery=False):
        self.postgresql = MockConnect()
        self.postgresql.mock_values['mock_recovery'] = mock_recovery
        Handler(MockRequest(path), ip_port, self)


if __name__ == '__main__':
    unittest.main()

## Instruction:
Remove some more unneccesary code.
## Code After:

import unittest
import sys

from helpers.statuspage import StatusPage
from test_postgresql import MockConnect

if sys.hexversion >= 0x03000000:
    from io import BytesIO as IO
else:
    from StringIO import StringIO as IO


class TestStatusPage(unittest.TestCase):

    def test_do_GET(self):
        for mock_recovery in [True, False]:
            for page in [b'GET /pg_master', b'GET /pg_slave', b'GET /pg_status', b'GET /not_found']:
                self.http_server = MockServer(('0.0.0.0', 8888), StatusPage, page, mock_recovery)


class MockRequest(object):

    def __init__(self, path):
        self.path = path

    def makefile(self, *args, **kwargs):
        return IO(self.path)


class MockServer(object):

    def __init__(self, ip_port, Handler, path, mock_recovery=False):
        self.postgresql = MockConnect()
        self.postgresql.mock_values['mock_recovery'] = mock_recovery
        Handler(MockRequest(path), ip_port, self)
