test/unit/frontend_message_test.rb in vertica-0.12.0 vs test/unit/frontend_message_test.rb in vertica-1.0.0.rc1

@@ -1,19 +1,19 @@
 require 'test_helper'
 
 class FrontendMessageTest < Minitest::Test
 
   def test_copy_done_message
-    message = Vertica::Messages::CopyDone.new
+    message = Vertica::Protocol::CopyDone.new
     assert_equal message.to_bytes, ['c', 0, 0, 0, 4].pack('AC4')
   end
-  
+
   def test_copy_fail_message
-    message = Vertica::Messages::CopyFail.new('sad panda')
+    message = Vertica::Protocol::CopyFail.new('sad panda')
     assert_equal message.to_bytes, ['f', 0, 0, 0, 14, "sad panda"].pack('AC4Z*')
   end
-  
+
   def test_copy_data_message
-    message = Vertica::Messages::CopyData.new("foo|bar\n")
+    message = Vertica::Protocol::CopyData.new("foo|bar\n")
     assert_equal message.to_bytes, ['d', 0, 0, 0, 12, "foo|bar\n"].pack('AC4A*')
   end
 end