xref: /arkcompiler/ets_frontend/ets2panda/test/parser/ets/enum.sts

/*
 * Copyright (c) 2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

enum Color { Red, Green, Blue }

class A {
  public c: Color;
  public constructor(c_: Color) {
    this.c = c_;
  }
}

function colorToInt(c: Color): int {
  return c as int;
}

function main(): void {
  let red: Color = Color.Red;
  assert red == Color.Red;
  assert colorToInt(red) == 0;

  let red2: Color = red;
  assert red == red2;
  assert Color.Red == red2;

  let red_int: int = red as int;
  assert red_int == 0;
  assert red_int == Color.Red as int;

  let a: A = new A(red);
  assert a.c == Color.Red;

  switch(red) {
    case Color.Green:
      assert false;
      break;
    case Color.Red:
      assert true;
      break;
    default:
      assert false;
  }

  switch(Color.Blue) {
    case Color.Green:
    case Color.Red:
      assert false;
      break;
    case Color.Blue:
      assert true;
      break;
    default:
      assert false;
  }
}