class RobotTests(object):
def setup_helper(self):
self.conn = MagicMock()
self.sensor_state = defaultdict(int)
self.sense = MorseSense(self.conn, self.sensor_state)
self.conn.subscribe = MagicMock(
side_effect=self.subscribe_side_effect()
)
self.conn.unscribe = MagicMock()
self.sense.start(.001)
def assert_bot_detection(self, robot):
self.assertEqual(self.sensor_state['robot'], robot)
def assert_sensor_data(self, decoded_sensor):
for key, value in decoded_sensor.iteritems():
self.assertEqual(self.sensor_state[key], value)
def assert_start(self):
self.assertEqual(self.conn.subscribe.call_count, 2)
def assert_stop(self):
self.assertEqual(self.conn.subscribe.call_count, 2)
self.conn.reset_mock()
self.sense.unsubscribe()
self.assertEqual(self.conn.unsubscribe.call_count, 2)
def setup_helper(self):
self.conn = MagicMock()
self.sensor_state = defaultdict(int)
self.sense = MorseSense(self.conn, self.sensor_state)
self.conn.subscribe = MagicMock(
side_effect=self.subscribe_side_effect())
self.conn.unscribe = MagicMock()
self.sense.start(.001)
def setup_helper(self):
self.conn = MagicMock()
self.sensor_state = defaultdict(int)
self.sense = MorseSense(self.conn, self.sensor_state)
self.conn.subscribe = MagicMock(
side_effect=self.subscribe_side_effect()
)
self.conn.unscribe = MagicMock()
self.sense.start(.001)
