import unittest, curses
from test import test_support

def testCursesColorSet(stdscrn):
   curses.init_pair(1, curses.COLOR_RED, curses.COLOR_WHITE)
   curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_BLUE)
  
   stdscrn.addstr("Testing color_set function")
   curses.color_set(1)
   stdscrn.addstr("RED/BLACK\n")
   curses.color_set(2)
   stdscrn.addstr("WHITE/BLUE\n")
   curses.color_set(0)
   stdscrn.addstr("Default\n\nTesting wcolor_set function")
   stdscrn.color_set(1)
   stdscrn.addstr("RED/BLACK\n")
   stdscrn.color_set(2)
   stdscrn.addstr("WHITE/BLUE")
   stdscrn.color_set(0)
   stdscrn.addstr("Test Complete press any key to continue.")
   stdscrn.getch()

def test_main(stdscrn):
   curses.savetty()
   if curses.has_colors():
    testCursesColorSet(stdscrn)    
   else:
      stdscr.addstr( "Test Aborted: Color not supported on this terminal.")


if __name__ == '__main__':
    curses.wrapper(test_main)