test/test_cola_models_main.py

   1 import helper
   2 import os
   3
   4 from cola.models.main import MainModel
   5
   6 class MainModelTestCase(helper.TestCase):
   7     """Tests the cola.models.main.MainModel class."""
   8
   9     def setup_baseline_repo(self, commit=True):
  10         """Create a baseline repo for testing."""
  11         self.shell("""
  12             git init >/dev/null &&
  13             touch the-file &&
  14             git add the-file
  15         """)
  16         if commit:
  17             self.shell("git commit -s -m'Initial commit' >/dev/null")
  18
  19     def test_project(self):
  20         """Test the MainModel's 'project' attribute."""
  21         self.setup_baseline_repo()
  22         model = MainModel()
  23         model.use_worktree(os.getcwd())
  24         project = os.path.basename(self.get_dir())
  25         self.assertEqual(project, model.project)