class BaseTest(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(BaseTest, self).__init__(*args, **kwargs)
self.auto_power_on = True
def setUp(self):
obc_com = config['OBC_COM']
sys_bus_com = config['SYS_BUS_COM']
payload_bus_com = config['PAYLOAD_BUS_COM']
use_single_bus = config['SINGLE_BUS']
gpio_com = config['GPIO_COM']
self.gpio = Pins(gpio_com)
extensions.set_up(test_id=self.id())
self.system = System(obc_com, sys_bus_com, payload_bus_com, use_single_bus, self.gpio, self.auto_power_on)
def tearDown(self):
self.system.close()
self.gpio.close()
extensions.tear_down(test_id=self.id())
def power_on_obc(self):
self.system.obc.power_on()
def power_on_and_wait(self):
self.power_on_obc()
self.system.obc.wait_to_start()
def _prepare_environment():
import logging
from pins import Pins
from system import System
try:
from build_config import config
except ImportError as e:
raise ImportError(
"Error loading config: %s. Did you forget to add <build>/integration_tests to PYTHONPATH?"
% e.message)
logging.basicConfig(level=logging.DEBUG)
obc_com = config['OBC_COM']
mock_com = config['MOCK_COM']
gpio_com = config['GPIO_COM']
boot_handler = BootToUpper() if config['BOOT_UPPER'] else BootToIndex(
config['BOOT_INDEX'])
gpio = Pins(gpio_com)
s = System(obc_com, mock_com, gpio, boot_handler)
s.start()
return s
def loop():
mqtt = MQTTClient("umqtt_alarm",
MQTT_HOST,
port=1883,
user=MQTT_USER,
password=MQTT_PASSWORD)
mqtt.connect()
kp = Keypad(uart)
pins = Pins([{
"label": "hall",
"value": 4
}, {
"label": "stairs",
"value": 5
}, {
"label": "cellar",
"value": 18
}, {
"label": "unassigned19",
"value": 19
}], [{
"label": "unassigned21",
"value": 21
}, {
"label": "unassigned22",
"value": 22
}, {
"label": "unassigned23",
"value": 23
}])
alarm = Alarm(kp, pins, mqtt)
alarm.loop()
def setUp(self):
obc_com = config['OBC_COM']
sys_bus_com = config['SYS_BUS_COM']
payload_bus_com = config['PAYLOAD_BUS_COM']
use_single_bus = config['SINGLE_BUS']
gpio_com = config['GPIO_COM']
self.gpio = Pins(gpio_com)
extensions.set_up(test_id=self.id())
self.system = System(obc_com, sys_bus_com, payload_bus_com, use_single_bus, self.gpio, self.auto_power_on)
from Cheetah.Template import Template
from core import Core
from port import Ports
from irq import IRQTable
from timer import Timer
from pins import Pins
from spi import SPI
from usart import USART
converters=[
Core(),
IRQTable(),
Ports(),
Timer(),
Pins(),
SPI(),
USART()
]
from os import listdir
# for fn in listdir("atmel-xml/"):
# if (fn.find(".xml")<0
# or fn.find("ATxmega")>=0
# or fn.find("AT89")>=0
# or fn.find("AT86")>=0
# or fn.find("ATtiny10")>=0
# or fn.find("ATtiny28")>=0 # <- has weird port logic. FIXME: support this!
# or fn.find("HV")>=0 # 'smart battery devices' also have some weird port logic.
import time
from pins import Pins
from hipchat import HipChat
if __name__ == '__main__':
pins = Pins()
hipchat = HipChat()
pins.setup()
try:
while True:
status = hipchat.get_user_status()
pins.set_color(status.get_color())
time.sleep(5)
except KeyboardInterrupt:
pins.teardown()
def __init__(self):
self.red = Red()
self.pins = Pins()
self.nrst = self.pins.pin3
self.rrout = self.red.rrout
def __init__(self):
self.blue = Blue()
self.pins = Pins()
self.nrst = self.pins.pin3
self.rrout = self.blue.rrout
def __init__(self):
self.pins = Pins()
self.clk = self.pins.pin5
self.nrst = self.pins.pin3
