TERM 2 Assignment 2 Strand

/*

 * AP CS MOOC

 * Term 2 - Assignment 2, Part 2: Strand

 * A class which represents a strand of lights.

 */



import java.util.ArrayList;



public class Strand

{

 // An ArrayList that stores a strand of lights

 private ArrayList<Light> strand = new ArrayList<Light>();



 // Default constructor that sets strand to an ArrayList holding one

 // turned on white bulb, that is not burnt out.

 public Strand()

 {

  this (1);

 }



 // A constructor that sets strand to an ArrayList of size n, holding

 // n white bulbs, that are all turned on and not burnt out. If n <= 0,

 // then the strand should be set to size one, with a white bulb, on

 // and not burnt out.

 public Strand(int n)

 {

  if (n <= 0)

  {

    n = 1;

  }



  for (int i = 1; i <= n; i++)

  {

   strand.add (new Light (true, false, "white"));

  }



 }



 // This method returns a String representation of the

 // Light objects in the ArrayList, one per line. For example,

 // here is the String returned when toString is called on a

 // Strand with 5 lights:

 //

 // green    on not burnt out

 // red    off not burnt out

 // green    off burnt out

 // blue    on not burnt out

 // red    on not burnt out

 //

 // Note: there is a tab between the value for color and "off"/"on,"

 // and one space before the "burnt out" or "not burnt out".

 public String toString()

 {

  String t = "";



   for (Light l: strand)

   {

     t += l.toString() + "\n";

   }

  return t;

 }



 // This method sets the color of all the light bulbs in the entire Strand.

 public void setColor(String c)

 {

   for (Light l: strand)

   {

     l.setColor(c);

   }

 }



 // This method sets the light bulbs to the pattern "blue", "red", "green",

 // "blue", "red", "green", ... until the end of the strand.

 public void setMulti()

 {

  for (int i = 0; i < strand.size(); i++)

  {

    if (i % 3 == 0)

    {

      strand.get(i).setColor ("blue");

    }

    if (i % 3 == 1)

    {

      strand.get(i).setColor ("red");

    }

    if (i % 3 == 2)

    {

      strand.get(i).setColor ("green");

    }

  }

 }

 // This method turns on all the lights in the strand. Each individual bulb

 // can only be turned on if it's burntOut variable is false.

 public void turnOn()

 {

   for (Light l: strand)

   {

     if (!l.isOn())

     {

       l.flip();

     }

   }

 }

 // This method turns off all the lights in the strand.

 public void turnOff()

 {

  for (Light l: strand)

   {

     if (l.isOn())

     {

       l.flip();

     }

   }

 }



 // This method sets the Light at location i’s burntOut variable to true.

 public void burnOut(int i)

 {

  strand.get(i).burnOut();

 }



 public static void main(String[] args)

 {

  // *************************************************************************

  // 1. Test Strand()

  // *************************************************************************

  System.out.println("1. Test the default constructor Strand()");

  Strand strand1 = new Strand();

  if (strand1.strand.size() == 1)

   System.out.println("*** PASS: Strand() creates a list of size 1");

  else

   System.out.println("*** FAIL: Strand() creates a list of size "

     + strand1.strand.size()

     + ", when a list of size 1 is expected.");



  // ***********************************

  // 2. Test Strand(n)

  // ***********************************

  System.out.println("\n2. Test the constructor Strand(n)");

  // Try to create a strand of lights with 0 bulbs

  Strand emptyStrand = new Strand(0);

  if (emptyStrand.strand.size() == 1)

   System.out.println("*** PASS: Strand(0) creates a list of size 1");

  else

   System.out.println("*** FAIL: Strand(0)  creates a list of size "

     + emptyStrand.strand.size()

     + ", when a list of size 1 is expected.");

  // Try to create a strand of lights with a negative number

  Strand negativeStrand = new Strand(-7);

  if (negativeStrand.strand.size() == 1)

   System.out.println("*** PASS: Strand(-7) creates a list of size 1");

  else

   System.out.println("*** FAIL: Strand(-7) creates a list of size "

     + negativeStrand.strand.size()

     + ", when a list of size 1 is expected.");

  // Try to create a strand of lights with a positive number

  Strand strandWithFiveBulbs = new Strand(5);

  if (strandWithFiveBulbs.strand.size() == 5)

   System.out.println("*** PASS: Strand(5) creates a list of size 5");

  else

   System.out.println("*** FAIL: Strand(5) creates a list of size "

     + strandWithFiveBulbs.strand.size()

     + ", when a list of size 5 is expected.");

  // Verify that all the light bulbs are initialized properly

  boolean success = true;

  for (Light bulb : strandWithFiveBulbs.strand)

  {

   if (!(bulb.isOn() && bulb.getColor().equals("white")))

   {

    success = false;

   }

  }

  if (strandWithFiveBulbs.strand.size() > 0 && success)

  {

   System.out.println("*** PASS: Strand(5) initialized bulbs correctly");

  }

  else

  {

   System.out.println("*** FAIL: Strand(5) did not initialize bulb(s) correctly");

  }





  // ***********************************

  // 3. Test setColor(String)

  // ***********************************

  System.out.println("\n3. Test setColor(String)");

  // All of the bulbs in our strandWithFiveBulbs are white. Set them to

  // green.

  strandWithFiveBulbs.setColor("green");

  success = true;

  for (Light light : strandWithFiveBulbs.strand)

  {

   if (!light.getColor().equals("green"))

    success = false;

  }

  if (strandWithFiveBulbs.strand.size() > 0 && success)

   System.out.println("*** PASS: setColor worked as expected (green test)");

  else

   System.out.println("*** FAIL: setColor did not work as expected (green test)");

  // Now try to set them to a color that is not supported.  This should

  // cause all the bulbs to be set back to white.

  strandWithFiveBulbs.setColor("pink");

  success = true;

  for (Light light : strandWithFiveBulbs.strand)

  {

   if (!light.getColor().equals("white"))

    success = false;

  }

  if (strandWithFiveBulbs.strand.size() > 0 && success)

   System.out.println("*** PASS: setColor worked as expected (pink test)");

  else

   System.out.println("*** FAIL: setColor did not work as expected (pink test)");





  // ***********************************

  // 4. Test turnOff()

  // ***********************************

  System.out.println("\n4. Test turnOff()");

  strand1.turnOff();

  if (strand1.strand.size() > 0 && !strand1.strand.get(0).isOn())

  {

   System.out.println("*** PASS: turnOff() worked as expected");

  }

  else

  {

   System.out.println("*** FAIL: turnOff() did not work as expected");

  }



  // ***********************************

  // 5. Test turnOn()

  // ***********************************

  System.out.println("\n5. Test turnOn()");

  strand1.turnOn();

  if (strand1.strand.size() > 0 && strand1.strand.get(0).isOn())

  {

   System.out.println("*** PASS: turnOn() worked as expected");

  }

  else

  {

   System.out.println("*** FAIL: turnOn() did not work as expected");

  }



  // ***********************************

  // 6. Test burnOut(int)

  // ***********************************

  System.out.println("\n6. Test burnOut(n)");

  strand1.burnOut(0);

  if (strand1.toString().equals("white\toff burnt out\n"))

  {

   System.out.println("*** PASS: burnOut(1) works as expected.");

  }

  else

  {

   System.out.println("*** FAIL: burnOut(1) does not work as expected.");

  }

 }

}