// Copyright (c) Facebook, Inc. and its affiliates. (http://www.facebook.com)
#include "strarray-builtins.h"

#include "gtest/gtest.h"

#include "builtins.h"
#include "test-utils.h"

namespace py {
namespace testing {

using StrArrayBuiltinsTest = RuntimeFixture;

TEST_F(StrArrayBuiltinsTest, DunderNewAndDunderInitCreateStrArray) {
  HandleScope scope(thread_);
  ASSERT_FALSE(runFromCStr(runtime_, "obj = _str_array('hello')").isError());
  StrArray self(&scope, mainModuleAt(runtime_, "obj"));
  EXPECT_TRUE(isStrEqualsCStr(runtime_->strFromStrArray(self), "hello"));
}

TEST_F(StrArrayBuiltinsTest, DunderInitWithWrongTypeRaisesTypeError) {
  EXPECT_TRUE(raisedWithStr(runFromCStr(runtime_, "obj = _str_array(b'hello')"),
                            LayoutId::kTypeError,
                            "_str_array can only be initialized with str"));
}

TEST_F(StrArrayBuiltinsTest, DunderReprWithNonStrArrayRaisesTypeError) {
  EXPECT_TRUE(raisedWithStr(
      runFromCStr(runtime_, "_str_array.__repr__(b'')"), LayoutId::kTypeError,
      "'__repr__' for '_str_array' objects doesn't apply to a 'bytes' object"));
}

TEST_F(StrArrayBuiltinsTest, DunderReprWithSimpleStrArrayReturnsStr) {
  HandleScope scope(thread_);
  ASSERT_FALSE(
      runFromCStr(runtime_, "obj = _str_array('hello').__repr__()").isError());
  Str self(&scope, mainModuleAt(runtime_, "obj"));
  EXPECT_TRUE(isStrEqualsCStr(*self, "_str_array('hello')"));
}

TEST_F(StrArrayBuiltinsTest, DunderStrWithNonStrArrayRaisesTypeError) {
  EXPECT_TRUE(raisedWithStr(
      runFromCStr(runtime_, "_str_array.__str__(b'')"), LayoutId::kTypeError,
      "'__str__' for '_str_array' objects doesn't apply to a 'bytes' object"));
}

TEST_F(StrArrayBuiltinsTest, DunderStrWithEmptyStrArrayReturnsEmptyStr) {
  HandleScope scope(thread_);
  StrArray self(&scope, runtime_->newStrArray());
  Object repr(&scope, runBuiltin(METH(_str_array, __str__), self));
  EXPECT_TRUE(isStrEqualsCStr(*repr, ""));
}

TEST_F(StrArrayBuiltinsTest, DunderStrWithSimpleStrArrayReturnsStr) {
  HandleScope scope(thread_);
  StrArray self(&scope, runtime_->newStrArray());
  const char* test_str = "foo";
  Str foo(&scope, runtime_->newStrFromCStr(test_str));
  runtime_->strArrayAddStr(thread_, self, foo);
  Object repr(&scope, runBuiltin(METH(_str_array, __str__), self));
  EXPECT_TRUE(isStrEqualsCStr(*repr, test_str));
}

}  // namespace testing
}  // namespace py