From b23a1fc613d9c7368ffd32e415ecd7c911eee1c2 Mon Sep 17 00:00:00 2001
From: Boone Gorges <boonebgorges@git.wordpress.org>
Date: Wed, 23 Sep 2015 14:04:55 +0000
Subject: [PATCH] Add a few simple tests for `get_comment_class()`.

Props walterebert, dipesh.kakadiya.
See #33947.

git-svn-id: https://develop.svn.wordpress.org/trunk@34453 602fd350-edb4-49c9-b593-d223f7449a82
---
 .../phpunit/tests/comment/getCommentClass.php | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 tests/phpunit/tests/comment/getCommentClass.php

diff --git a/tests/phpunit/tests/comment/getCommentClass.php b/tests/phpunit/tests/comment/getCommentClass.php
new file mode 100644
index 0000000000..c7ce64923b
--- /dev/null
+++ b/tests/phpunit/tests/comment/getCommentClass.php
@@ -0,0 +1,39 @@
+<?php
+
+/**
+ * @group comment
+ */
+class Tests_Comment_GetCommentClass extends WP_UnitTestCase {
+	public function test_should_accept_comment_id() {
+		$post_id    = $this->factory->post->create();
+		$comment_id = $this->factory->comment->create( array( 'comment_post_ID' => $post_id ) );
+
+		$classes = get_comment_class( '', $comment_id );
+		$this->assertContains( 'comment', $classes );
+	}
+
+	public function test_should_accept_comment_object() {
+		$post_id = $this->factory->post->create();
+		$comment = $this->factory->comment->create_and_get( array( 'comment_post_ID' => $post_id ) );
+
+		$classes = get_comment_class( '', $comment );
+		$this->assertContains( 'comment', $classes );
+	}
+
+	public function test_should_append_single_class() {
+		$post_id    = $this->factory->post->create();
+		$comment_id = $this->factory->comment->create( array( 'comment_post_ID' => $post_id ) );
+
+		$classes = get_comment_class( 'foo', $comment_id );
+		$this->assertContains( 'foo', $classes );
+	}
+
+	public function test_should_append_array_of_classes() {
+		$post_id    = $this->factory->post->create();
+		$comment_id = $this->factory->comment->create( array( 'comment_post_ID' => $post_id ) );
+
+		$classes = get_comment_class( array( 'foo', 'bar' ), $comment_id );
+		$this->assertContains( 'foo', $classes );
+		$this->assertContains( 'bar', $classes );
+	}
+}