One click categories do not have animations

cms/djangoapps/contentstore/features/component_settings_editor_helpers.py

@@ -12,16 +12,17 @@ def create_component_instance(step, component_button_css, category,
                               has_multiple_templates=True):
 
     click_new_component_button(step, component_button_css)
-
-    def animation_done(_driver):
-        return world.browser.evaluate_script("$('div.new-component').css('display')") == 'none'
-    world.wait_for(animation_done)
+    if category == 'problem' or category == 'html':
+        def animation_done(_driver):
+            return world.browser.evaluate_script("$('div.new-component').css('display')") == 'none'
+        world.wait_for(animation_done)
 
     if has_multiple_templates:
         click_component_from_menu(category, boilerplate, expected_css)
 
     assert_equal(1, len(world.css_find(expected_css)))
 
+
 @world.absorb
 def click_new_component_button(step, component_button_css):
     step.given('I have clicked the new unit button')

cms/djangoapps/contentstore/features/problem-editor.py

@@ -155,8 +155,12 @@ def cancel_does_not_save_changes(step):
 @step('I have created a LaTeX Problem')
 def create_latex_problem(step):
     world.click_new_component_button(step, '.large-problem-icon')
+
+    def animation_done(_driver):
+            return world.browser.evaluate_script("$('div.new-component').css('display')") == 'none'
+    world.wait_for(animation_done)
     # Go to advanced tab.
-    world.css_click('#ui-id-2', success_condition=lambda: world.css_has_class('div.ui-tabs li', 'ui-state-active', index=1))
+    world.css_click('#ui-id-2')
     world.click_component_from_menu("problem", "latex_problem.yaml", '.xmodule_CapaModule')