//Author: Charley Ruggiero

package com.example.test;

import com.example.model.*;
import junit.framework.*;
import java.lang.*;
import java.util.*;
import java.io.*;

public class ScheduleTest extends TestCase {

    public ScheduleTest(String name) {
	super(name);
    }

    public void testSchedule() {
	String freeTimes;
	String constraints = "MTWRF 8:00 AM-5:00 PM";
	String busyTimes = "MW 9:05 AM-12:15 PM, W 1:25 PM-2:15 PM, F 9:05 AM-11:00 AM, "  
                        + "T 12:30 PM-4:00 PM, R 12:30 PM-1:45 PM, R 3:30 PM-5:00 PM, " 
                        + "W 12:00 PM-3:00 PM";
	String expected = "M 8:00 AM-9:05 AM, 12:15 PM-5:00 PM" + '\n' +
			  "T 8:00 AM-12:30 PM, 4:00 PM-5:00 PM" + '\n' +
			  "W 8:00 AM-9:05 AM, 3:00 PM-5:00 PM" + '\n' +
			  "R 8:00 AM-12:30 PM, 1:45 PM-3:30 PM, " + '\n' +
			  "F 8:00 AM-9:05 AM, 11:00 AM-5:00 PM";
 
        freeTimes = setupScheduleTest(constraints, busyTimes);                  
	Assert.assertTrue(expected.equals(freeTimes));
    } 

    public void testFreeTimesNotSetException() {
	Schedule sched = new Schedule();
	String freeTimes = "";
	String busyTimes = "TR 9:30 AM-12:00 PM";
	String expected = "ERROR: Cannot add busy times.  " +
			  "Must set free time constraints first.\n";

	try { 
            sched.setBusyTimes(busyTimes); 
        }        
        catch (FreeTimesNotSetException e) { 
            freeTimes = e.getMessage(); 
	    Assert.assertTrue(expected.equals(freeTimes));
        } 
    }

    public void testFreeTimesAlreadySetException() {
	Schedule sched = new Schedule();
	String freeTimes = "";
	String constraints1 = "MTWRF 8:00 AM-5:00 PM";
	String constraints2 = "TR 9:30 AM-12:00 PM";
	String expected = "ERROR: Free time constraints already set.  " +
			  "Add busy times.\n";

	try { 
            sched.setFreeTimeConstraints(constraints1); 
        }        
        catch (FreeTimesAlreadySetException e) { 
            freeTimes = e.getMessage(); 
	    Assert.assertTrue(expected.equals(freeTimes));
        } 

	try { 
            sched.setFreeTimeConstraints(constraints2); 
        }        
        catch (FreeTimesAlreadySetException e) { 
            freeTimes = e.getMessage(); 
	    Assert.assertTrue(expected.equals(freeTimes));
        } 
    }

    public void testSetMinFreeTimeSpan() {
	Schedule sched = new Schedule();

	String freeTimeSpan = "60";
	int expected = 60;

	sched.setMinFreeTimeSpan(freeTimeSpan);

	Assert.assertTrue(expected == sched.getMinFreeTimeSpan());
    }

    private String setupScheduleTest(String constraints, String busyTimes) {
	Schedule sched = new Schedule();
	String freeTimes = ""; 
         
	try {         
            sched.setFreeTimeConstraints(constraints);
	}
	catch (FreeTimesAlreadySetException e) {
	    freeTimes = e.getMessage();
            return freeTimes;
	}
         
        try { 
            sched.setBusyTimes(busyTimes); 
        }        
        catch (FreeTimesNotSetException e) { 
            freeTimes = e.getMessage(); 
	    return freeTimes;
        } 

	return sched.getFreeTimes();
    }

}