#!/usr/bin/env python

import unittest
from test import test_support
import poplib

class PopTest(unittest.TestCase):
    testServer = 'cyrus.andrew.cmu.edu'
    remotePort = 110
    remoteUser = 'anonymous'
    remotePass = 'pass'
    
    def setUp(self):
        self.server = poplib.POP3(self.testServer, self.remotePort)

    def tearDown(self):
        rs = self.server.quit()
        self.assertEqual (rs, '+OK')

    def test_havecapa(self):
        rs = self.server.capa()
        self.assertTrue ('IMPLEMENTATION' in rs)

class PopStartTlsTest(PopTest):
    def setUp(self):
        self.server = poplib.POP3(self.testServer, self.remotePort)
        rs = self.server.capa()
        self.assertTrue('STLS' in rs)
        rs = self.server.stls()
        self.assertEqual(rs[:3], '+OK')

    def test_anonlogin(self):
        rs = self.server.capa()
        self.assertTrue('USER' in rs)
        rs = self.server.user(self.remoteUser)
        self.assertEqual (rs[:3], '+OK')
        rs = self.server.pass_(self.remotePass)
        self.assertEqual(rs[:3], '+OK')
        
class PopSSLTest(PopStartTlsTest):
    remotePort = 995
    def setUp(self):
        self.server = poplib.POP3_SSL(self.testServer, self.remotePort)
      
def test_main():
    test_support.requires("network")
    test_support.run_unittest(PopTest, 
                              PopStartTlsTest, 
                              PopSSLTest
                              )

if __name__ == "__main__":
    test_main()