redirect/test/redirect_project_template_test.py

# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import unittest

from google.appengine.ext import ndb
from google.appengine.ext import testbed
from redirect import redirect_project_template


class TestRedirectCustomValue(unittest.TestCase):

  def setUp(self):
    self.testbed = testbed.Testbed()
    self.testbed.activate()
    self.testbed.init_datastore_v3_stub()
    self.testbed.init_memcache_stub()
    ndb.get_context().clear_cache()

  def tearDown(self):
    self.testbed.deactivate()

  def testGetRedirectProjectTemplate(self):
    redirectProjectTemplate = redirect_project_template.RedirectProjectTemplate
    redirectProjectTemplate(
        ProjectName='a',
        MonorailTemplateName='default template',
        RedirectComponentID='123',
        RedirectTemplateID='456',
        id='a:default template').put()

    (t, v) = redirectProjectTemplate.Get('a', 'default template')
    self.assertEqual(t, '123')
    self.assertEqual(v, '456')

  def testGetRedirectProjectTemplateWithoutValue(self):
    redirectProjectTemplate = redirect_project_template.RedirectProjectTemplate

    (t, v) = redirectProjectTemplate.Get('a', 'default template')
    self.assertEqual(t, None)
    self.assertEqual(v, None)