add test for generate class exist

ftwbzhao · ftwbzhao · commit 70c066e02017 · 2015-05-20T18:28:44.000+08:00
diff --git a/tests/Command/GenerateTest.php b/tests/Command/GenerateTest.php
@@ -50,7 +50,20 @@ public function test_migration_generate()
         $migration_path = __DIR__ . '/../Fake/migrations/';
         $this->ci->config->set_item('migration_path', $migration_path);
         $status = $this->cmd->__invoke('migration', 'Test_of_generate_migration');
-        $this->assertEquals(0, $status);
+        $this->assertEquals(Status::SUCCESS, $status);
+
+        foreach (glob($migration_path . '*_Test_of_generate_migration.php') as $file) {
+            unlink($file);
+        }
+    }
+
+    public function test_migration_generate_class_exist()
+    {
+        $migration_path = __DIR__ . '/../Fake/migrations/';
+        $this->ci->config->set_item('migration_path', $migration_path);
+        $status = $this->cmd->__invoke('migration', 'Test_of_generate_migration');
+        $status = $this->cmd->__invoke('migration', 'Test_of_generate_migration');
+        $this->assertEquals(Status::FAILURE, $status);
 
         foreach (glob($migration_path . '*_Test_of_generate_migration.php') as $file) {
             unlink($file);
@@ -63,7 +76,7 @@ public function test_migration_cannot_write_to_file()
         $this->ci->config->set_item('migration_path', $migration_path);
         $status = $this->cmd->__invoke('migration', 'Test_of_generate_migration');
         $this->assertEquals(Status::FAILURE, $status);
-        
+
         $this->stderr->rewind();
         $actual = $this->stderr->fread(8192);
         $this->assertContains("Can't write to ", $actual);
