89 lines
2.7 KiB
Ruby
89 lines
2.7 KiB
Ruby
|
require_relative 'helper'
|
||
|
|
require 'net/ssh'
|
||
|
|
require 'net/scp'
|
||
|
|
|
||
|
|
module Mains
|
||
|
|
class TestArm < MiniTest::Test
|
||
|
|
|
||
|
|
# runnable_methods is called by minitest to determine which tests to run
|
||
|
|
def self.runnable_methods
|
||
|
|
all = Dir["test/mains/source/*.rb"]
|
||
|
|
tests =[]
|
||
|
|
all.each do |file_name|
|
||
|
|
fullname = file_name.split("/").last.split(".").first
|
||
|
|
name , stdout , exit_code = fullname.split("_")
|
||
|
|
method_name = "test_#{name}"
|
||
|
|
tests << method_name
|
||
|
|
input = File.read(file_name)
|
||
|
|
self.send(:define_method, method_name ) do
|
||
|
|
out , code = run_main(input , name)
|
||
|
|
assert_equal stdout , out , "Wrong stdout #{name}"
|
||
|
|
assert_equal exit_code , code.to_s , "Wrong exit code #{name}"
|
||
|
|
end
|
||
|
|
end
|
||
|
|
tests
|
||
|
|
end
|
||
|
|
|
||
|
|
DEBUG = true
|
||
|
|
|
||
|
|
def setup
|
||
|
|
Risc.machine.boot
|
||
|
|
end
|
||
|
|
|
||
|
|
def run_main(input , file)
|
||
|
|
Vool::VoolCompiler.ruby_to_binary( "class Space;def main(arg);#{input};end;end" )
|
||
|
|
writer = Elf::ObjectWriter.new(Risc.machine)
|
||
|
|
writer.save "test/#{file}.o"
|
||
|
|
run_ssh(file)
|
||
|
|
end
|
||
|
|
|
||
|
|
def run_ssh( file )
|
||
|
|
host = ENV["ARM_HOST"]
|
||
|
|
return unless host
|
||
|
|
port = (ENV["ARM_PORT"] || 2222)
|
||
|
|
user = (ENV["ARM_USER"] || "pi")
|
||
|
|
binary_file = "/tmp/#{file}"
|
||
|
|
object_file = binary_file + ".o"
|
||
|
|
Net::SCP.start(host, user , port: port ) do |scp|
|
||
|
|
puts "Copying test/#{file}.o to #{object_file}" if DEBUG
|
||
|
|
scp.upload! "test/#{file}.o", object_file
|
||
|
|
end
|
||
|
|
Net::SSH.start(host, user , port: port ) do |ssh|
|
||
|
|
puts "Linking #{object_file}" if DEBUG
|
||
|
|
stdout , exit_code = ssh_exec!(ssh , "ld -N -o #{binary_file} #{object_file}")
|
||
|
|
assert_equal 0 , exit_code , "Linking #{binary_file} failed"
|
||
|
|
puts "Running #{binary_file}" if DEBUG
|
||
|
|
stdout , exit_code = ssh_exec!(ssh , binary_file)
|
||
|
|
puts "Result #{stdout} #{exit_code}" if DEBUG
|
||
|
|
return stdout , exit_code
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def ssh_exec!(ssh, command)
|
||
|
|
stdout_data = ""
|
||
|
|
exit_code = nil
|
||
|
|
ssh.open_channel do |channel|
|
||
|
|
channel.exec(command) do |ch, success|
|
||
|
|
unless success
|
||
|
|
raise "FAILED: couldn't execute command (ssh.channel.exec)"
|
||
|
|
end
|
||
|
|
channel.on_data do |c,data|
|
||
|
|
stdout_data+=data
|
||
|
|
end
|
||
|
|
channel.on_extended_data do |c,type,data|
|
||
|
|
raise "#{ssh} received stderr #{data}"
|
||
|
|
end
|
||
|
|
channel.on_request("exit-status") do |c,data|
|
||
|
|
exit_code = data.read_long
|
||
|
|
end
|
||
|
|
channel.on_request("exit-signal") do |c, data|
|
||
|
|
raise "#{ssh} received signal #{data.read_long}"
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
ssh.loop
|
||
|
|
[stdout_data, exit_code]
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|