Пример #1
0
    def test_get_bundle_uuid(self):
        tester = self

        worksheet_uuid = '0x12345'
        missing_name = 'missing_name'
        unique_name = 'unique_name'
        ambiguous_name = 'ambiguous_name'
        bundle_counts = {
            missing_name: 0,
            unique_name: 1,
            ambiguous_name: 2,
        }
        test_uuids = ['test_uuid_0', 'test_uuid_1', 'test_uuid_2']
        test_uuid = test_uuids[0]

        class MockBundleModel(object):
            def get_bundle_uuids(self, conditions, max_results, count=False):
                tester.assertEqual(set(conditions),
                                   set(['name', 'user_id', 'worksheet_uuid']))
                name = conditions['name'].replace('%', '')
                tester.assertIn(name, bundle_counts)
                count = bundle_counts[name]
                return [test_uuids[i] for i in range(count)]

        model = MockBundleModel()
        user_id = None

        # Test that get_bundle_uuid is idempotent on generated uuids.
        uuid = spec_util.generate_uuid()
        self.assertEqual(
            uuid,
            canonicalize.get_bundle_uuid(model, user_id, worksheet_uuid, uuid))
        # Test that get_bundle_uuid returns the uuid of a uniquely named bundle.
        self.assertEqual(
            test_uuid,
            canonicalize.get_bundle_uuid(model, user_id, worksheet_uuid,
                                         unique_name))
        # Test that get_bundle_uuid raises UsageErrors on missing names.
        self.assertRaises(
            UsageError,
            lambda: canonicalize.get_bundle_uuid(model, user_id,
                                                 worksheet_uuid, missing_name),
        )
        # Test that get_bundle_uuid raises UsageError on specs that can be neither a
        # name or a uuid.
        self.assertRaises(
            UsageError,
            lambda: canonicalize.get_bundle_uuid(
                model, user_id, worksheet_uuid, 'names have no exclamations!'),
        )
Пример #2
0
  def test_get_bundle_uuid(self):
    tester = self

    worksheet_uuid = '0x12345'
    missing_name = 'missing_name'
    unique_name = 'unique_name'
    ambiguous_name = 'ambiguous_name'
    bundle_counts = {
      missing_name: 0,
      unique_name: 1,
      ambiguous_name: 2,
    }
    test_uuids = ['test_uuid_0', 'test_uuid_1', 'test_uuid_2']
    test_uuid = test_uuids[0]

    class MockBundleModel(object):
      def get_bundle_uuids(self, conditions, max_results, count=False):
        tester.assertEqual(set(conditions), set(['name', 'user_id', 'worksheet_uuid']))
        name = conditions['name'].replace('%', '')
        tester.assertIn(name, bundle_counts)
        count = bundle_counts[name]
        return [
          test_uuids[i]
          for i in range(count)
        ]
    model = MockBundleModel()
    user_id = None

    # Test that get_bundle_uuid is idempotent on generated uuids.
    uuid = spec_util.generate_uuid()
    self.assertEqual(uuid, canonicalize.get_bundle_uuid(model, user_id, worksheet_uuid, uuid))
    # Test that get_bundle_uuid returns the uuid of a uniquely named bundle.
    self.assertEqual(test_uuid, canonicalize.get_bundle_uuid(model, user_id, worksheet_uuid, unique_name))
    # Test that get_bundle_uuid raises UsageErrors on missing names.
    self.assertRaises(
      UsageError,
      lambda: canonicalize.get_bundle_uuid(model, user_id, worksheet_uuid, missing_name),
    )
    # Test that get_bundle_uuid raises UsageError on specs that can be neither a
    # name or a uuid.
    self.assertRaises(
      UsageError,
      lambda: canonicalize.get_bundle_uuid(model, user_id, worksheet_uuid, 'names have no exclamations!'),
    )
Пример #3
0
 def get_bundle_uuid(self, worksheet_uuid, bundle_spec):
     return canonicalize.get_bundle_uuid(self.model, self._current_user_id(), worksheet_uuid, bundle_spec)