1 files changed, 36 insertions, 0 deletions
diff --git a/spec/factories/clusters/clusters.rb b/spec/factories/clusters/clusters.rb
new file mode 100644
index 00000000000..9e73a19e856
--- /dev/null
+++ b/spec/factories/clusters/clusters.rb
@@ -0,0 +1,36 @@
+FactoryGirl.define do
+  factory :cluster, class: Clusters::Cluster do
+    user
+    name 'test-cluster'
+
+    trait :project do
+      after(:create) do |cluster, evaluator|
+        cluster.projects << create(:project)
+      end
+    end
+
+    trait :provided_by_user do
+      provider_type :user
+      platform_type :kubernetes
+
+      platform_kubernetes factory: [:cluster_platform_kubernetes, :configured]
+    end
+
+    trait :provided_by_gcp do
+      provider_type :gcp
+      platform_type :kubernetes
+
+      provider_gcp factory: [:cluster_provider_gcp, :created]
+      platform_kubernetes factory: [:cluster_platform_kubernetes, :configured]
+    end
+
+    trait :providing_by_gcp do
+      provider_type :gcp
+      provider_gcp factory: [:cluster_provider_gcp, :creating]
+    end
+
+    trait :disabled do
+      enabled false
+    end
+  end
+end