Skip to content

Instantly share code, notes, and snippets.

@OniDaito
Created July 5, 2023 07:56
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save OniDaito/051ade0aba716ae2fea858ef70e16423 to your computer and use it in GitHub Desktop.
Save OniDaito/051ade0aba716ae2fea858ef70e16423 to your computer and use it in GitHub Desktop.
platform.io script to alter the the Unity Tests to use Serial1 instead of Serial for the ATMega32u4
from platformio.public import UnityTestRunner
class CustomTestRunner(UnityTestRunner):
UNITY_FRAMEWORK_CONFIG = dict(
native=dict(
code="""
#include <stdio.h>
void unityOutputStart(unsigned long baudrate) { (void) baudrate; }
void unityOutputChar(unsigned int c) { putchar(c); }
void unityOutputFlush(void) { fflush(stdout); }
void unityOutputComplete(void) { }
""",
language="c",
),
arduino=dict(
code="""
#include <Arduino.h>
void unityOutputStart(unsigned long baudrate) { Serial1.begin(baudrate); }
void unityOutputChar(unsigned int c) { Serial1.write(c); }
void unityOutputFlush(void) { Serial1.flush(); }
void unityOutputComplete(void) { Serial1.end(); }
""",
language="cpp",
),
mbed=dict(
code="""
#include <mbed.h>
#if MBED_MAJOR_VERSION == 6
UnbufferedSerial pc(USBTX, USBRX);
#else
RawSerial pc(USBTX, USBRX);
#endif
void unityOutputStart(unsigned long baudrate) { pc.baud(baudrate); }
void unityOutputChar(unsigned int c) {
#if MBED_MAJOR_VERSION == 6
pc.write(&c, 1);
#else
pc.putc(c);
#endif
}
void unityOutputFlush(void) { }
void unityOutputComplete(void) { }
""",
language="cpp",
),
espidf=dict(
code="""
#include <stdio.h>
void unityOutputStart(unsigned long baudrate) { (void) baudrate; }
void unityOutputChar(unsigned int c) { putchar(c); }
void unityOutputFlush(void) { fflush(stdout); }
void unityOutputComplete(void) { }
""",
language="c",
),
zephyr=dict(
code="""
#include <sys/printk.h>
void unityOutputStart(unsigned long baudrate) { (void) baudrate; }
void unityOutputChar(unsigned int c) { printk("%c", c); }
void unityOutputFlush(void) { }
void unityOutputComplete(void) { }
""",
language="c",
),
legacy_custom_transport=dict(
code="""
#include <unittest_transport.h>
void unityOutputStart(unsigned long baudrate) { unittest_uart_begin(); }
void unityOutputChar(unsigned int c) { unittest_uart_putchar(c); }
void unityOutputFlush(void) { unittest_uart_flush(); }
void unityOutputComplete(void) { unittest_uart_end(); }
""",
language="cpp",
),
)
@OniDaito
Copy link
Author

OniDaito commented Jul 5, 2023

Place this file inside your platform.io test directory. To setup the tests place the following YAML inside your platformio.ini file:

[env:testlocal]
platform_packages = platformio/tool-simavr
test_speed = 38400
test_framework = custom
test_testing_command = 
     ${platformio.packages_dir}/tool-simavr/bin/simavr
    -m
    atmega32u4
    -f
    8000000L
    ${platformio.build_dir}/${this.__env__}/firmware.elf
 build_flags = 
    -DTESTLOCAL
    -DSENSOR_TYPE=5
    -Werror

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment