package fr.umlv.fruits;

import static org.junit.Assert.*;

import org.junit.Test;
 
public class Basket2Test {
  @Test
  public void findFirstItemByCountry() {
    Fruit fruit1 = new Orange("US", 75);
    Fruit fruit2 = new Banana("Spain");
    Fruit fruit3 = new Banana("Spain");
    Basket basket = new Basket();
    basket.add(fruit1);
    basket.add(fruit2);
    basket.add(fruit3);
    assertSame(fruit1, basket.findFirstItemByCountry("US"));
    assertSame(fruit2, basket.findFirstItemByCountry("Spain"));
  }

  @Test
  public void getPrice() {
    Fruit fruit1 = new Orange("US", 75);
    Banana fruit2 = new Banana("Spain");
    Basket basket = new Basket();
    basket.add(fruit1);
    basket.add(fruit2);
    assertEquals(fruit1.getPrice() + fruit2.getPrice(), basket.getPrice());
  }
}
