class BaseTestCase(TestCase): def setUp(self): self.db = init_db() self.user = User() self.user.username = '******' self.user.chat_id = '339847919' self.user.gitlab_user = '******' self.user.gitlab_user_id = '4047441' self.user.access_token = "123456" self.user.save() self.project = Project() self.project_name = 'ada-gitlab' self.project_id = '12532279' self.project.save_webhook_infos(self.user, self.project_name, self.project_id) self.user.save_gitlab_repo_data(self.project) self.GITLAB_API_TOKEN = "12345" self.mocked_404_response = Response() self.mocked_404_response.status_code = 404 self.response_unauthorized = Response() self.response_unauthorized.status_code = 401 def create_app(self): app = create_app() app.config.from_object("gitlab.config.TestingConfig") return app def tearDown(self): self.db.drop_database('api')
def test_save_gitlab_repo_data(self): User.drop_collection() user = User() username = "******" user.create_user(username) user.save() Project.drop_collection() project = Project() project.user_id = user.id project.description = "Test user add project" project.name = "Test user add project" project.web_url = "https://useraddProject.com" project.branches = ["branch1", "branch2"] project.save() user.save_gitlab_repo_data(project) project_user = User.objects(project=project).first() self.assertEqual(user, project_user)