test/storage/sql_test.rb in diaspora-vines-0.1.25 vs test/storage/sql_test.rb in diaspora-vines-0.1.26

@@ -1,10 +1,18 @@
 # encoding: UTF-8
 
 require 'test_helper'
 require 'storage/sql_schema'
 
+module Vines
+  class Config
+    def instance.max_offline_msgs
+      return 1
+    end
+  end
+end
+
 module Diaspora
   class Application < Rails::Application
     def config.database_configuration
       {
         "development" => {
@@ -71,9 +79,73 @@
   end
 
   after do
     db = Rails.application.config.database_configuration["development"]["database"]
     File.delete(db) if File.exist?(db)
+  end
+
+  def test_save_message
+    fibered do
+      db = storage
+
+      assert_nil db.save_message("", "", "")
+      assert_nil db.save_message("dude@valid@jid", "[e-mail]", "")
+      assert_nil db.save_message("dude@valid@jid", "", "test")
+      assert_nil db.save_message("", "[e-mail]", "test")
+
+      db.save_message(@test_user[:jid], "[e-mail]", "test")
+
+      msgs = Vines::Storage::Sql::ChatOfflineMessage.where(:to => "[e-mail]")
+      assert_equal 1, msgs.count
+      assert_equal "[e-mail]", msgs.first.to
+      assert_equal @test_user[:jid], msgs.first.from
+      assert_equal "test", msgs.first.message
+
+      db.save_message("[e-mail]", "[e-mail]", "test2")
+
+      msgs = Vines::Storage::Sql::ChatOfflineMessage.where(:to => "[e-mail]")
+      assert_equal 1, msgs.count # due max limit equals one (see max_offline_msgs)
+      assert_equal "[e-mail]", msgs.first.to
+      assert_equal "[e-mail]", msgs.first.from
+      assert_equal "test2", msgs.first.message # should be latest message
+    end
+  end
+
+  def test_find_messages
+    fibered do
+      db = storage
+
+      assert_nil db.find_messages("")
+      assert_equal 0, db.find_messages("[e-mail]").keys.count
+
+      Vines::Storage::Sql::ChatOfflineMessage.new(
+        :from => @test_user[:jid],
+        :to => "[e-mail]",
+        :message => "test"
+      ).save
+
+      msgs = db.find_messages("[e-mail]")
+      assert_equal 1, msgs.keys.count
+      assert_equal "[e-mail]", msgs[1][:to]
+      assert_equal @test_user[:jid], msgs[1][:from]
+      assert_equal "test", msgs[1][:message]
+    end
+  end
+
+  def test_destroy_message
+    fibered do
+      db = storage
+      com = Vines::Storage::Sql::ChatOfflineMessage
+      com.new(:from => @test_user[:jid],
+        :to => "[e-mail]",
+        :message => "test"
+      ).save
+
+      db.destroy_message(1)
+
+      count = Vines::Storage::Sql::ChatOfflineMessage.count(:id => 1)
+      assert_equal 0, count
+    end
   end
 
   def test_aspect_chat_enabled
     fibered do
       db = storage