Add unit tests for overriddenBy
This commit is contained in:
SiboVG
parent
03431b9c41
commit
6d80378a3f
@ -1,32 +1,21 @@
|
||||
package net.sf.openrocket.rocketcomponent;
|
||||
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import org.junit.Test;
|
||||
|
||||
import net.sf.openrocket.aerodynamics.AerodynamicForces;
|
||||
import net.sf.openrocket.aerodynamics.BarrowmanCalculator;
|
||||
import net.sf.openrocket.aerodynamics.FlightConditions;
|
||||
import net.sf.openrocket.aerodynamics.WarningSet;
|
||||
|
||||
import net.sf.openrocket.rocketcomponent.AxialStage;
|
||||
import net.sf.openrocket.rocketcomponent.BodyTube;
|
||||
import net.sf.openrocket.rocketcomponent.FinSet;
|
||||
import net.sf.openrocket.rocketcomponent.FlightConfiguration;
|
||||
import net.sf.openrocket.rocketcomponent.NoseCone;
|
||||
import net.sf.openrocket.rocketcomponent.Rocket;
|
||||
import net.sf.openrocket.rocketcomponent.RocketComponent;
|
||||
|
||||
import net.sf.openrocket.util.Coordinate;
|
||||
import net.sf.openrocket.util.BaseTestCase.BaseTestCase;
|
||||
import net.sf.openrocket.util.MathUtil;
|
||||
import net.sf.openrocket.util.TestRockets;
|
||||
import net.sf.openrocket.util.BaseTestCase.BaseTestCase;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.Map;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
public class OverrideTest extends BaseTestCase {
|
||||
|
||||
@ -109,4 +98,268 @@ public class OverrideTest extends BaseTestCase {
|
||||
Map<RocketComponent, AerodynamicForces> forceMap = calc.getForceAnalysis(configuration, conditions, warnings);
|
||||
assertEquals(sustainer.getOverrideCD() + bodytube.getOverrideCD() + forceMap.get(nosecone).getCD(), forceMap.get(rocket).getCD(), MathUtil.EPSILON);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test whether children components of a parent that has subcomponents overridden for mass, CG, or CD have the correct
|
||||
* overriddenBy object.
|
||||
*/
|
||||
@Test
|
||||
public void testOverriddenBy() {
|
||||
// Create test rocket
|
||||
Rocket rocket = TestRockets.makeEstesAlphaIII();
|
||||
|
||||
// Obtain the necessary components
|
||||
AxialStage sustainer = rocket.getStage(0);
|
||||
NoseCone noseCone = (NoseCone) sustainer.getChild(0);
|
||||
BodyTube bodyTube = (BodyTube) sustainer.getChild(1);
|
||||
FinSet finSet = (FinSet) bodyTube.getChild(0);
|
||||
LaunchLug launchLug = (LaunchLug) bodyTube.getChild(1);
|
||||
InnerTube innerTube = (InnerTube) bodyTube.getChild(2);
|
||||
EngineBlock engineBlock = (EngineBlock) innerTube.getChild(0);
|
||||
Parachute parachute = (Parachute) bodyTube.getChild(3);
|
||||
CenteringRing bulkhead = (CenteringRing) bodyTube.getChild(4);
|
||||
|
||||
// Check initial override by components
|
||||
assertNull(rocket.getMassOverriddenBy());
|
||||
assertNull(rocket.getCGOverriddenBy());
|
||||
assertNull(rocket.getCDOverriddenBy());
|
||||
assertNull(sustainer.getMassOverriddenBy());
|
||||
assertNull(sustainer.getCGOverriddenBy());
|
||||
assertNull(sustainer.getCDOverriddenBy());
|
||||
assertNull(noseCone.getMassOverriddenBy());
|
||||
assertNull(noseCone.getCGOverriddenBy());
|
||||
assertNull(noseCone.getCDOverriddenBy());
|
||||
assertNull(bodyTube.getMassOverriddenBy());
|
||||
assertNull(bodyTube.getCGOverriddenBy());
|
||||
assertNull(bodyTube.getCDOverriddenBy());
|
||||
assertNull(finSet.getMassOverriddenBy());
|
||||
assertNull(finSet.getCGOverriddenBy());
|
||||
assertNull(finSet.getCDOverriddenBy());
|
||||
assertNull(launchLug.getMassOverriddenBy());
|
||||
assertNull(launchLug.getCGOverriddenBy());
|
||||
assertNull(launchLug.getCDOverriddenBy());
|
||||
assertNull(innerTube.getMassOverriddenBy());
|
||||
assertNull(innerTube.getCGOverriddenBy());
|
||||
assertNull(innerTube.getCDOverriddenBy());
|
||||
assertNull(engineBlock.getMassOverriddenBy());
|
||||
assertNull(engineBlock.getCGOverriddenBy());
|
||||
assertNull(engineBlock.getCDOverriddenBy());
|
||||
assertNull(parachute.getMassOverriddenBy());
|
||||
assertNull(parachute.getCGOverriddenBy());
|
||||
assertNull(parachute.getCDOverriddenBy());
|
||||
assertNull(bulkhead.getMassOverriddenBy());
|
||||
assertNull(bulkhead.getCGOverriddenBy());
|
||||
assertNull(bulkhead.getCDOverriddenBy());
|
||||
|
||||
// Override body tube mass, CG, and CD without for subcomponents
|
||||
bodyTube.setMassOverridden(true);
|
||||
bodyTube.setCGOverridden(true);
|
||||
bodyTube.setCDOverridden(true);
|
||||
|
||||
assertNull(sustainer.getMassOverriddenBy());
|
||||
assertNull(sustainer.getCGOverriddenBy());
|
||||
assertNull(sustainer.getCDOverriddenBy());
|
||||
assertNull(noseCone.getMassOverriddenBy());
|
||||
assertNull(noseCone.getCGOverriddenBy());
|
||||
assertNull(noseCone.getCDOverriddenBy());
|
||||
assertNull(bodyTube.getMassOverriddenBy());
|
||||
assertNull(bodyTube.getCGOverriddenBy());
|
||||
assertNull(bodyTube.getCDOverriddenBy());
|
||||
assertNull(finSet.getMassOverriddenBy());
|
||||
assertNull(finSet.getCGOverriddenBy());
|
||||
assertNull(finSet.getCDOverriddenBy());
|
||||
assertNull(launchLug.getMassOverriddenBy());
|
||||
assertNull(launchLug.getCGOverriddenBy());
|
||||
assertNull(launchLug.getCDOverriddenBy());
|
||||
assertNull(innerTube.getMassOverriddenBy());
|
||||
assertNull(innerTube.getCGOverriddenBy());
|
||||
assertNull(innerTube.getCDOverriddenBy());
|
||||
assertNull(engineBlock.getMassOverriddenBy());
|
||||
assertNull(engineBlock.getCGOverriddenBy());
|
||||
assertNull(engineBlock.getCDOverriddenBy());
|
||||
assertNull(parachute.getMassOverriddenBy());
|
||||
assertNull(parachute.getCGOverriddenBy());
|
||||
assertNull(parachute.getCDOverriddenBy());
|
||||
assertNull(bulkhead.getMassOverriddenBy());
|
||||
assertNull(bulkhead.getCGOverriddenBy());
|
||||
assertNull(bulkhead.getCDOverriddenBy());
|
||||
|
||||
// Override body tube mass for subcomponents
|
||||
bodyTube.setSubcomponentsOverriddenMass(true);
|
||||
|
||||
assertNull(sustainer.getMassOverriddenBy());
|
||||
assertNull(sustainer.getCGOverriddenBy());
|
||||
assertNull(sustainer.getCDOverriddenBy());
|
||||
assertNull(noseCone.getMassOverriddenBy());
|
||||
assertNull(noseCone.getCGOverriddenBy());
|
||||
assertNull(noseCone.getCDOverriddenBy());
|
||||
assertNull(bodyTube.getMassOverriddenBy());
|
||||
assertNull(bodyTube.getCGOverriddenBy());
|
||||
assertNull(bodyTube.getCDOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getMassOverriddenBy());
|
||||
assertNull(finSet.getCGOverriddenBy());
|
||||
assertNull(finSet.getCDOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getMassOverriddenBy());
|
||||
assertNull(launchLug.getCGOverriddenBy());
|
||||
assertNull(launchLug.getCDOverriddenBy());
|
||||
assertEquals(bodyTube, innerTube.getMassOverriddenBy());
|
||||
assertNull(innerTube.getCGOverriddenBy());
|
||||
assertNull(innerTube.getCDOverriddenBy());
|
||||
assertEquals(bodyTube, engineBlock.getMassOverriddenBy());
|
||||
assertNull(engineBlock.getCGOverriddenBy());
|
||||
assertNull(engineBlock.getCDOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getMassOverriddenBy());
|
||||
assertNull(parachute.getCGOverriddenBy());
|
||||
assertNull(parachute.getCDOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getMassOverriddenBy());
|
||||
assertNull(bulkhead.getCGOverriddenBy());
|
||||
assertNull(bulkhead.getCDOverriddenBy());
|
||||
|
||||
// Undo override body tube mass for subcomponents, do override of CG and CD for subcomponents
|
||||
bodyTube.setSubcomponentsOverriddenMass(false);
|
||||
bodyTube.setSubcomponentsOverriddenCG(true);
|
||||
bodyTube.setSubcomponentsOverriddenCD(true);
|
||||
|
||||
assertNull(noseCone.getMassOverriddenBy());
|
||||
assertNull(noseCone.getCGOverriddenBy());
|
||||
assertNull(noseCone.getCDOverriddenBy());
|
||||
assertNull(bodyTube.getMassOverriddenBy());
|
||||
assertNull(bodyTube.getCGOverriddenBy());
|
||||
assertNull(bodyTube.getCDOverriddenBy());
|
||||
assertNull(finSet.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCDOverriddenBy());
|
||||
assertNull(launchLug.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCDOverriddenBy());
|
||||
assertNull(innerTube.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, innerTube.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, innerTube.getCDOverriddenBy());
|
||||
assertNull(engineBlock.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, engineBlock.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, engineBlock.getCDOverriddenBy());
|
||||
assertNull(parachute.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCDOverriddenBy());
|
||||
assertNull(bulkhead.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCDOverriddenBy());
|
||||
|
||||
// Move the inner tube from the body tube to the nose cone
|
||||
bodyTube.removeChild(innerTube);
|
||||
noseCone.addChild(innerTube);
|
||||
|
||||
assertNull(noseCone.getMassOverriddenBy());
|
||||
assertNull(noseCone.getCGOverriddenBy());
|
||||
assertNull(noseCone.getCDOverriddenBy());
|
||||
assertNull(innerTube.getMassOverriddenBy());
|
||||
assertNull(innerTube.getCGOverriddenBy());
|
||||
assertNull(innerTube.getCDOverriddenBy());
|
||||
assertNull(engineBlock.getMassOverriddenBy());
|
||||
assertNull(engineBlock.getCGOverriddenBy());
|
||||
assertNull(engineBlock.getCDOverriddenBy());
|
||||
assertNull(bodyTube.getMassOverriddenBy());
|
||||
assertNull(bodyTube.getCGOverriddenBy());
|
||||
assertNull(bodyTube.getCDOverriddenBy());
|
||||
assertNull(finSet.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCDOverriddenBy());
|
||||
assertNull(launchLug.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCDOverriddenBy());
|
||||
assertNull(parachute.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCDOverriddenBy());
|
||||
assertNull(bulkhead.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCDOverriddenBy());
|
||||
|
||||
// Override mass of nose cone
|
||||
noseCone.setMassOverridden(true);
|
||||
|
||||
assertNull(noseCone.getMassOverriddenBy());
|
||||
assertNull(noseCone.getCGOverriddenBy());
|
||||
assertNull(noseCone.getCDOverriddenBy());
|
||||
assertNull(innerTube.getMassOverriddenBy());
|
||||
assertNull(innerTube.getCGOverriddenBy());
|
||||
assertNull(innerTube.getCDOverriddenBy());
|
||||
assertNull(engineBlock.getMassOverriddenBy());
|
||||
assertNull(engineBlock.getCGOverriddenBy());
|
||||
assertNull(engineBlock.getCDOverriddenBy());
|
||||
assertNull(bodyTube.getMassOverriddenBy());
|
||||
assertNull(bodyTube.getCGOverriddenBy());
|
||||
assertNull(bodyTube.getCDOverriddenBy());
|
||||
assertNull(finSet.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCDOverriddenBy());
|
||||
assertNull(launchLug.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCDOverriddenBy());
|
||||
assertNull(parachute.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCDOverriddenBy());
|
||||
assertNull(bulkhead.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCDOverriddenBy());
|
||||
|
||||
// Override nose cone mass for all subcomponents
|
||||
noseCone.setSubcomponentsOverriddenMass(true);
|
||||
|
||||
assertNull(sustainer.getMassOverriddenBy());
|
||||
assertNull(sustainer.getCGOverriddenBy());
|
||||
assertNull(sustainer.getCDOverriddenBy());
|
||||
assertNull(noseCone.getMassOverriddenBy());
|
||||
assertNull(noseCone.getCGOverriddenBy());
|
||||
assertNull(noseCone.getCDOverriddenBy());
|
||||
assertEquals(noseCone, innerTube.getMassOverriddenBy());
|
||||
assertNull(innerTube.getCGOverriddenBy());
|
||||
assertNull(innerTube.getCDOverriddenBy());
|
||||
assertEquals(noseCone, engineBlock.getMassOverriddenBy());
|
||||
assertNull(engineBlock.getCGOverriddenBy());
|
||||
assertNull(engineBlock.getCDOverriddenBy());
|
||||
assertNull(bodyTube.getMassOverriddenBy());
|
||||
assertNull(bodyTube.getCGOverriddenBy());
|
||||
assertNull(bodyTube.getCDOverriddenBy());
|
||||
assertNull(finSet.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCDOverriddenBy());
|
||||
assertNull(launchLug.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCDOverriddenBy());
|
||||
assertNull(parachute.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCDOverriddenBy());
|
||||
assertNull(bulkhead.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCDOverriddenBy());
|
||||
|
||||
// Override inner tube CG for all subcomponents
|
||||
innerTube.setCGOverridden(true);
|
||||
innerTube.setSubcomponentsOverriddenCG(true);
|
||||
|
||||
assertNull(noseCone.getMassOverriddenBy());
|
||||
assertNull(noseCone.getCGOverriddenBy());
|
||||
assertNull(noseCone.getCDOverriddenBy());
|
||||
assertEquals(noseCone, innerTube.getMassOverriddenBy());
|
||||
assertNull(innerTube.getCGOverriddenBy());
|
||||
assertNull(innerTube.getCDOverriddenBy());
|
||||
assertEquals(noseCone, engineBlock.getMassOverriddenBy());
|
||||
assertEquals(innerTube, engineBlock.getCGOverriddenBy());
|
||||
assertNull(engineBlock.getCDOverriddenBy());
|
||||
assertNull(bodyTube.getMassOverriddenBy());
|
||||
assertNull(bodyTube.getCGOverriddenBy());
|
||||
assertNull(bodyTube.getCDOverriddenBy());
|
||||
assertNull(finSet.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, finSet.getCDOverriddenBy());
|
||||
assertNull(launchLug.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, launchLug.getCDOverriddenBy());
|
||||
assertNull(parachute.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, parachute.getCDOverriddenBy());
|
||||
assertNull(bulkhead.getMassOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCGOverriddenBy());
|
||||
assertEquals(bodyTube, bulkhead.getCDOverriddenBy());
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user