rspec version type should be case sensitive
[sfa.git] / sfa / rspecs / rspec_converter.py
index 5301c2d..8a66108 100755 (executable)
@@ -2,6 +2,8 @@
 
 from sfa.rspecs.pg_rspec_converter import PGRSpecConverter
 from sfa.rspecs.sfa_rspec_converter import SfaRSpecConverter
+from sfa.rspecs.sfa_rspec import sfa_rspec_version
+from sfa.rspecs.pg_rspec import pg_rspec_version
 from sfa.rspecs.rspec_parser import parse_rspec
 
 
@@ -10,9 +12,9 @@ class RSpecConverter:
     @staticmethod
     def to_sfa_rspec(in_rspec):
         rspec = parse_rspec(in_rspec)
-        if rspec.version['type'].lower() == 'sfa'
+        if rspec.version['type'] == sfa_rspec_version['type']
           return in_rspec
-        elif rspec.version['type'].lower() == 'protogeni':
+        elif rspec.version['type'] == pg_rspec_version['type']:
             return PGRSpecConverter.to_sfa_rspec(in_rspec)
         else:
              return in_rspec 
@@ -20,9 +22,9 @@ class RSpecConverter:
     @staticmethod 
     def to_pg_rspec(in_rspec):
         rspec = parse_rspec(in_rspec)
-        if rspec.version['type'].lower() == 'protogeni': 
+        if rspec.version['type'] == pg_rspec_version['type']:
             return in_rspec
-        elif rspec.version['type'].lower() == 'sfa':
+        elif rspec.version['type'] == sfa_rspec_version['type']:
             return SfaRSpecConverter.to_pg_rspec(in_rspec)
         else:
             return in_rspec