diff --git a/src/test/java/seng202/group9/AirlineTest.java b/src/test/java/seng202/group9/AirlineTest.java
index d48d9e0..7024e53 100644
--- a/src/test/java/seng202/group9/AirlineTest.java
+++ b/src/test/java/seng202/group9/AirlineTest.java
@@ -6,6 +6,7 @@ import org.junit.Test;
 import org.junit.rules.ExpectedException;
 import seng202.group9.Controller.DataException;
 import seng202.group9.Core.Airline;
+import seng202.group9.Core.Country;
 import seng202.group9.Core.Route;
 
 import static org.junit.Assert.assertEquals;
@@ -176,6 +177,23 @@ public class AirlineTest{
 		allNipponAirways.hasDuplicate(allNipponAirway);
 	}
 
+	@Test
+	public void testCountry(){
+		Airline allNipponAirways = new Airline(324, "All Nippon Airways", "ANA All Nippon Airways",
+				"NH", "ANA", "ALL NIPPON", "Japan", "Y");
+		//,9,"U","Asia/Tokyo"
+		Country japan = new Country("U", "Japan");
+		allNipponAirways.setCountry(japan);
+		assertEquals(japan, allNipponAirways.getCountry());
+	}
+
+	@Test(expected = DataException.class)
+	public void getIDNullError() throws DataException{
+		Airline allNipponAirways = new Airline("All Nippon Airways", "ANA All Nippon Airways",
+				"NH", "ANA", "ALL NIPPON", "Japan", "Y");
+		allNipponAirways.getID();
+	}
+
 	@Test
     public void testToString(){
     	//ID, Name, Alias, IATA, ICAO, CallSign, Country, Active