added tests

.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+ - 2.2
+

Rakefile

@@ -0,0 +1,5 @@
+task default: %w[test]
+
+task :test do
+  ruby "snapr_test.rb"
+end

snapr.rb

@@ -10,5 +10,6 @@
 	     "  Send a message:       $ ruby snapr.rb \"You're Message in Quotes\"\n" \
 	     "  Live message display: $ ruby snapr.rb -watch"
 else
+
 	SnaprRunner.send ARGV[0], ENV['USER']
 end

snapr_message.rb

@@ -0,0 +1,19 @@
+require "zlib"
+
+class SnaprMessage
+
+	attr_accessor :message
+
+	def initialize(message)
+		@message = message
+	end
+
+	def encode_to_ascii
+		Zlib::Deflate.deflate(@message)
+	end
+
+	def self.decode_from_ascii(encoded_string)
+		SnaprMessage.new Zlib::Inflate.inflate(encoded_string)
+	end
+
+end

snapr_receiver.rb

@@ -1,6 +1,7 @@
 require "socket"
 require "ipaddr"
 require "zlib"
+require_relative "snapr_message"
 
 class SnaprReceiver
 
@@ -21,7 +22,8 @@ def self.run
 		loop do
 		  message, _ = socket.recvfrom(65000)
     	  system('clear')
-		  puts Zlib::Inflate.inflate(message)
+    	  msg = SnaprMessage.decode_from_ascii(message)
+    	  puts msg.message
 		  # keep the image up for atleast 5 seconds
 		  sleep 5
 		end

snapr_runner.rb

@@ -1,6 +1,7 @@
 require 'lolcommits'
 require 'catpix'
 require 'io/console'
+require_relative "snapr_message"
 require_relative 'snapr_receiver'
 require_relative 'snapr_sender'
 
@@ -70,8 +71,9 @@ def self.send(msg, user = "")
 
     # Broadcast the image on the network
     puts "Sending Image '#{filename}'..."
-    SnaprSender.send(Catpix.raw_img)
-    puts "Done"
+    msg = SnaprMessage.new(Catpix.raw_img)
+    SnaprSender.send(msg)
+    puts "Done. Support https://youthoffthestreets.com.au"
   end
 end
 

snapr_sender.rb

@@ -1,16 +1,16 @@
 require "socket"
 require "zlib"
+require_relative "snapr_message"
 
 class SnaprSender
 
 	MULTICAST_ADDR = "224.0.0.1"
 	PORT = 3000
 
-	def self.send img
-		compressed = Zlib::Deflate.deflate(img)
+	def self.send msg
 		socket = UDPSocket.open
 		socket.setsockopt(:IPPROTO_IP, :IP_MULTICAST_TTL, 1)
-		socket.send(compressed, 0, MULTICAST_ADDR, PORT)
+		socket.send(msg.encode_to_ascii, 0, MULTICAST_ADDR, PORT)
 		socket.close
 	end
 

snapr_test.rb

@@ -0,0 +1,22 @@
+require 'test/unit'
+require_relative 'snapr_message'
+
+class SnaprMessageTest < Test::Unit::TestCase
+
+  def test_message
+    msg = SnaprMessage.new("Hello")
+  	assert_equal("Hello", msg.message)
+  end
+
+  def test_encoding
+    msg = SnaprMessage.new("Hello")
+    expected = "x\x9C\xF3H\xCD\xC9\xC9\a\x00\x05\x8C\x01\xF5".force_encoding('ASCII-8BIT') 
+    assert_equal(expected, msg.encode_to_ascii)
+  end
+
+  def test_decoding
+    data = "x\x9C\xF3H\xCD\xC9\xC9\a\x00\x05\x8C\x01\xF5".force_encoding('ASCII-8BIT')
+    msg = SnaprMessage.decode_from_ascii(data)
+    assert_equal("Hello", msg.message)
+  end
+end