Add some more unit tests for the Trace model

diff --git a/test/traces/3.gpx b/test/traces/3.gpx
new file mode 100644 (file)
index 0000000..cdfa941
Binary files /dev/null and b/test/traces/3.gpx differ

diff --git a/test/unit/trace_test.rb b/test/unit/trace_test.rb
index f840d6c5c545306e522f1bd960296f1319e9dcbf..d2720e993d900b1ac96783a6c639cc5261f291aa 100644 (file)
--- a/test/unit/trace_test.rb
+++ b/test/unit/trace_test.rb
@@ -3,22 +3,38 @@ require File.dirname(__FILE__) + '/../test_helper'
 class TraceTest < ActiveSupport::TestCase
   api_fixtures
   
 class TraceTest < ActiveSupport::TestCase
   api_fixtures
   

+  def setup
+    @gpx_trace_dir = Object.send("remove_const", "GPX_TRACE_DIR")
+    Object.const_set("GPX_TRACE_DIR", File.dirname(__FILE__) + "/../traces")
+  end
+
+  def teardown
+    Object.send("remove_const", "GPX_TRACE_DIR")
+    Object.const_set("GPX_TRACE_DIR", @gpx_trace_dir)
+  end
+

   def test_trace_count
     assert_equal 5, Trace.count
   end
 
   def test_visible
   def test_trace_count
     assert_equal 5, Trace.count
   end
 
   def test_visible

-    check_query(Trace.visible, :public_trace_file, :anon_trace_file, :trackable_trace_file, :identifiable_trace_file)
+    check_query(Trace.visible, [:public_trace_file, :anon_trace_file, :trackable_trace_file, :identifiable_trace_file])

   end
 
   def test_visible_to
   end
 
   def test_visible_to

-    check_query(Trace.visible_to(1), :public_trace_file, :identifiable_trace_file)
-    check_query(Trace.visible_to(2), :public_trace_file, :anon_trace_file, :trackable_trace_file, :identifiable_trace_file)
-    check_query(Trace.visible_to(3), :public_trace_file, :identifiable_trace_file)
+    check_query(Trace.visible_to(1), [:public_trace_file, :identifiable_trace_file])
+    check_query(Trace.visible_to(2), [:public_trace_file, :anon_trace_file, :trackable_trace_file, :identifiable_trace_file])
+    check_query(Trace.visible_to(3), [:public_trace_file, :identifiable_trace_file])

   end
 
   def test_public
   end
 
   def test_public

-    check_query(Trace.public, :public_trace_file, :identifiable_trace_file, :deleted_trace_file)
+    check_query(Trace.public, [:public_trace_file, :identifiable_trace_file, :deleted_trace_file])
+  end
+
+  def test_tagged
+    check_query(Trace.tagged("London"), [:public_trace_file, :anon_trace_file])
+    check_query(Trace.tagged("Birmingham"), [:anon_trace_file, :identifiable_trace_file])
+    check_query(Trace.tagged("Unknown"), [])

   end
 
   def test_validations
   end
 
   def test_validations


@@ -76,9 +92,23 @@ class TraceTest < ActiveSupport::TestCase
     assert_equal false, gpx_files(:deleted_trace_file).identifiable?
   end
 
     assert_equal false, gpx_files(:deleted_trace_file).identifiable?
   end
 

+  def test_mime_type
+    assert_equal "application/gpx+xml", gpx_files(:public_trace_file).mime_type
+    assert_equal "application/gpx+xml", gpx_files(:anon_trace_file).mime_type
+    assert_equal "application/x-bzip2", gpx_files(:trackable_trace_file).mime_type
+    assert_equal "application/x-gzip", gpx_files(:identifiable_trace_file).mime_type
+  end
+
+  def test_extension_name
+    assert_equal ".gpx", gpx_files(:public_trace_file).extension_name
+    assert_equal ".gpx", gpx_files(:anon_trace_file).extension_name
+    assert_equal ".gpx.bz2", gpx_files(:trackable_trace_file).extension_name
+    assert_equal ".gpx.gz", gpx_files(:identifiable_trace_file).extension_name
+  end
+

 private
 
 private
 

-  def check_query(query, *traces)
+  def check_query(query, traces)

     traces = traces.map { |t| gpx_files(t) }.sort
     assert_equal traces, query.order(:id)
   end
     traces = traces.map { |t| gpx_files(t) }.sort
     assert_equal traces, query.order(:id)
   end