Add diagram direct edition tests