Skip to content

Commit 755be22

Browse files
committed
Updated MockedEncoder with rev changes
1 parent b4438e8 commit 755be22

File tree

2 files changed

+46
-36
lines changed

2 files changed

+46
-36
lines changed

src/main/java/org/carlmontrobotics/lib199/sim/MockedEncoder.java

Lines changed: 31 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -128,38 +128,42 @@ public double getVelocity() {
128128
* velocityConversionFactor;
129129
}
130130

131-
// @Override
132-
// public REVLibError setPositionConversionFactor(double factor) {
133-
// positionConversionFactor = factor;
134-
// return REVLibError.kOk;
135-
// }
131+
//Let Mock to map these methods although they don't exist in the original class
132+
public interface RemovedMethods {
133+
REVLibError setPositionConversionFactor(double factor);
134+
double getPositionConversionFactor();
135+
REVLibError setVelocityConversionFactor(double factor);
136+
double getVelocityConversionFactor();
137+
REVLibError setInverted(boolean inverted);
138+
boolean getInverted();
139+
}
136140

137-
// @Override
138-
// public double getPositionConversionFactor() {
139-
// return positionConversionFactor;
140-
// }
141+
public REVLibError setPositionConversionFactor(double factor) {
142+
positionConversionFactor = factor;
143+
return REVLibError.kOk;
144+
}
141145

142-
// @Override
143-
// public REVLibError setVelocityConversionFactor(double factor) {
144-
// velocityConversionFactor = factor;
145-
// return REVLibError.kOk;
146-
// }
146+
public double getPositionConversionFactor() {
147+
return positionConversionFactor;
148+
}
147149

148-
// @Override
149-
// public double getVelocityConversionFactor() {
150-
// return velocityConversionFactor;
151-
// }
150+
public REVLibError setVelocityConversionFactor(double factor) {
151+
velocityConversionFactor = factor;
152+
return REVLibError.kOk;
153+
}
152154

153-
// @Override
154-
// public REVLibError setInverted(boolean inverted) {
155-
// this.inverted = inverted;
156-
// return REVLibError.kOk;
157-
// }
155+
public double getVelocityConversionFactor() {
156+
return velocityConversionFactor;
157+
}
158158

159-
// @Override
160-
// public boolean getInverted() {
161-
// return inverted;
162-
// }
159+
public REVLibError setInverted(boolean inverted) {
160+
this.inverted = inverted;
161+
return REVLibError.kOk;
162+
}
163+
164+
public boolean getInverted() {
165+
return inverted;
166+
}
163167

164168
// @Override
165169
// public REVLibError setAverageDepth(int depth) {

src/test/java/org/carlmontrobotics/lib199/sim/MockedEncoderTest.java

Lines changed: 15 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,12 @@
99

1010
import com.revrobotics.REVLibError;
1111
import com.revrobotics.RelativeEncoder;
12+
import com.revrobotics.spark.SparkBase.PersistMode;
13+
import com.revrobotics.spark.SparkBase.ResetMode;
14+
import com.revrobotics.spark.SparkFlex;
15+
import com.revrobotics.spark.SparkMax;
16+
import com.revrobotics.spark.config.SparkFlexConfig;
17+
import com.revrobotics.spark.config.SparkMaxConfig;
1218

1319
import org.carlmontrobotics.lib199.Mocks;
1420
import org.carlmontrobotics.lib199.REVLibErrorAnswer;
@@ -60,8 +66,8 @@ public void testFunctionality() {
6066
}
6167

6268
private void testFunctionalityWithPositionConversionFactor(double factor, RelativeEncoder enc, SimDouble positionSim) {
63-
assertEquals(REVLibError.kOk, enc.setPositionConversionFactor(factor)); //FIXME: make this in a config with positionConversionFactor
64-
assertEquals(factor, enc.getPositionConversionFactor(), 0.01);
69+
assertEquals(REVLibError.kOk, ((MockedEncoder) enc).setPositionConversionFactor(factor));
70+
assertEquals(factor, ((MockedEncoder) enc).getPositionConversionFactor(), 0.01);
6571
testPosition(10, enc, factor, positionSim);
6672
testPosition(0, enc, factor, positionSim);
6773
testPosition(-10, enc, factor, positionSim);
@@ -73,7 +79,7 @@ private void testPosition(double position, RelativeEncoder enc, double conversio
7379
assertEquals(position, enc.getPosition(), 0.02);
7480
assertEquals(REVLibError.kOk, enc.setPosition(0));
7581
assertEquals(0, enc.getPosition(), 0.02);
76-
positionSim.set(position / enc.getPositionConversionFactor() + positionSim.get());
82+
positionSim.set(position / ((MockedEncoder) enc).getPositionConversionFactor() + positionSim.get());
7783
assertEquals(position, enc.getPosition(), 0.02);
7884
}
7985

@@ -87,11 +93,11 @@ private boolean simDeviceExists(String deviceName) {
8793

8894
private SafelyClosable createEncoder(int deviceId) {
8995
SimDevice device = SimDevice.create("testDevice", deviceId);
90-
return (SafelyClosable)Mocks.createMock(
91-
RelativeEncoder.class,
92-
new MockedEncoder(device, 4096, false, false),
93-
new REVLibErrorAnswer(),
94-
SafelyClosable.class);
96+
return (SafelyClosable)Mocks.createMock(
97+
RelativeEncoder.class,
98+
new MockedEncoder(device, 4096, false, false),
99+
new REVLibErrorAnswer(),
100+
SafelyClosable.class, MockedEncoder.RemovedMethods.class);
95101
}
96102

97103
private void withEncoders(EncoderTest func) {
@@ -112,5 +118,5 @@ private void withEncoder(int id, EncoderTest func) {
112118
private interface EncoderTest {
113119
public void test(RelativeEncoder encoder, SimDeviceSim sim, SimDouble posSim);
114120
}
115-
121+
116122
}

0 commit comments

Comments
 (0)